aljaz/Auto-GPT
1
0
Fork 0
mirror of https://github.com/aljazceru/Auto-GPT.git synced 2025-12-17 22:14:28 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update ai_functions.py

Browse Source 
Just import dumps from JSON library.
This commit is contained in:
Andres Caicedo
2023-04-04 10:49:28 +02:00
parent 27bf7c47e9
commit 1b4a5301d0
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/ai_functions.py
View File

@@ -1,5 +1,5 @@
from typing import List, Optional
import json
from json import dumps
from config import Config
from call_ai_function import call_ai_function
from json_parser import fix_and_parse_json
@@ -23,7 +23,7 @@ def improve_code(suggestions: List[str], code: str) -> str:
function_string = (
"def generate_improved_code(suggestions: List[str], code: str) -> str:"
)
args = [json.dumps(suggestions), code]
args = [dumps(suggestions), code]
description_string = """Improves the provided code based on the suggestions provided, making no other changes."""
result_string = call_ai_function(function_string, args, description_string)
@@ -36,7 +36,7 @@ def write_tests(code: str, focus: List[str]) -> str:
function_string = (
"def create_test_cases(code: str, focus: Optional[str] = None) -> str:"
)
args = [code, json.dumps(focus)]
args = [code, 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)