| """Code evaluation module.""" | |
| from __future__ import annotations | |
| from autogpt.llm_utils import call_ai_function | |
| def analyze_code(code: str) -> list[str]: | |
| """ | |
| A function that takes in a string and returns a response from create chat | |
| completion api call. | |
| Parameters: | |
| code (str): Code to be evaluated. | |
| Returns: | |
| A result string from create chat completion. A list of suggestions to | |
| improve the code. | |
| """ | |
| function_string = "def analyze_code(code: str) -> List[str]:" | |
| args = [code] | |
| description_string = ( | |
| "Analyzes the given code and returns a list of suggestions" " for improvements." | |
| ) | |
| return call_ai_function(function_string, args, description_string) | |