还原被删除的auto-gpt

autogpt/commands/write_tests.py

@@ -0,0 +1,37 @@
+"""A module that contains a function to generate test cases for the submitted code."""
+from __future__ import annotations
+
+import json
+
+from autogpt.commands.command import command
+from autogpt.llm_utils import call_ai_function
+
+
+@command(
+    "write_tests",
+    "Write Tests",
+    '"code": "<full_code_string>", "focus": "<list_of_focus_areas>"',
+)
+def write_tests(code: str, focus: list[str]) -> str:
+    """
+    A function that takes in code and focus topics and returns a response from create
+      chat completion api call.
+
+    Parameters:
+        focus (list): A list of suggestions around what needs to be improved.
+        code (str): Code for test cases to be generated against.
+    Returns:
+        A result string from create chat completion. Test cases for the submitted code
+          in response.
+    """
+
+    function_string = (
+        "def create_test_cases(code: str, focus: Optional[str] = None) -> str:"
+    )
+    args = [code, json.dumps(focus)]
+    description_string = (
+        "Generates test cases for the existing code, focusing on"
+        " specific areas if required."
+    )
+
+    return call_ai_function(function_string, args, description_string)