mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-17 14:04:27 +01:00
1073954fb7
* Pi's message. * Fix most everything. * Blacked * Add Typing, Docstrings everywhere, organize the code a bit. * Black * fix import * Update message, dedupe. * Increase backoff time. * bump up retries
29 lines
881 B
Python
29 lines
881 B
Python
import json
|
|
from typing import List
|
|
|
|
from autogpt.llm_utils import call_ai_function
|
|
|
|
|
|
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)
|