AutoGPT/autogpt/commands/evaluate_code.py

"""Code evaluation module."""
from __future__ import annotations

from autogpt.commands.command import command
from autogpt.llm_utils import call_ai_function


@command("evaluate_code", "Evaluate Code", '"code": "<full_code_string>"')
def evaluate_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)