mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-19 06:54:22 +01:00
bfbe613960
Additional changes: * Improve typing * Modularize message history memory & fix/refactor lots of things * Fix summarization * Move memory relevance calculation to MemoryItem & improve test * Fix import warnings in web_selenium.py * Remove `memory_add` ghost command * Implement overlap in `split_text` * Move memory tests into subdirectory * Remove deprecated `get_ada_embedding()` and helpers * Fix used token calculation in `chat_with_ai` * Replace Message TypedDict by dataclass * Fix AgentManager singleton issues in tests --------- Co-authored-by: Auto-GPT-Bot <github-bot@agpt.co>
36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
|
|
from autogpt.commands.command import command
|
|
from autogpt.llm.utils import call_ai_function
|
|
|
|
|
|
@command(
|
|
"improve_code",
|
|
"Get Improved Code",
|
|
'"suggestions": "<list_of_suggestions>", "code": "<full_code_string>"',
|
|
)
|
|
def improve_code(suggestions: list[str], code: str) -> str:
|
|
"""
|
|
A function that takes in code and suggestions and returns a response from create
|
|
chat completion api call.
|
|
|
|
Parameters:
|
|
suggestions (list): A list of suggestions around what needs to be improved.
|
|
code (str): Code to be improved.
|
|
Returns:
|
|
A result string from create chat completion. Improved code in response.
|
|
"""
|
|
|
|
function_string = (
|
|
"def generate_improved_code(suggestions: list[str], code: str) -> str:"
|
|
)
|
|
args = [json.dumps(suggestions), code]
|
|
description_string = (
|
|
"Improves the provided code based on the suggestions"
|
|
" provided, making no other changes."
|
|
)
|
|
|
|
return call_ai_function(function_string, args, description_string)
|