Implements code execution command!

This allows the AI to execute code inside it's workspace folder.
This commit is contained in:
Torantulino
2023-04-01 16:01:36 +01:00
parent 29d8fa5d38
commit de975d3bf9
6 changed files with 37 additions and 43 deletions

View File

@@ -39,9 +39,9 @@ def improve_code(suggestions: List[str], code: str) -> str:
### Writing tests
def write_tests(code: str, focus: Optional[str] = None) -> str:
def write_tests(code: str, focus: List[str]) -> str:
function_string = "def create_test_cases(code: str, focus: Optional[str] = None) -> str:"
args = [focus] if focus else []
args = [code, json.dumps(focus)]
description_string = """Generates test cases for the existing code, focusing on specific areas if required."""
result_string = call_ai_function(function_string, args, description_string)