Significant-Gravitas
/
AutoGPT
mirror of https://github.com/Significant-Gravitas/AutoGPT.git
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #685 from STRML/fix/json-parsing 

Fix various JSON input bugs
pull/865/head^2
Richard Beales 2023-04-12 18:03:34 +01:00 committed by GitHub
parent a7176cd229 434f27fb51
commit c98471e333
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/json_parser.py
View File

@ -26,7 +26,7 @@ JSON_SCHEMA = """
""" """
def fix_and_parse_json( def fix_and_parse_json(
json_str: str, json_str: str,
try_to_fix_with_gpt: bool = True try_to_fix_with_gpt: bool = True
) -> Union[str, Dict[Any, Any]]: ) -> Union[str, Dict[Any, Any]]:
@ -35,8 +35,8 @@ def fix_and_parse_json(
json_str = json_str.replace('\t', '') json_str = json_str.replace('\t', '')
return json.loads(json_str) return json.loads(json_str)
except json.JSONDecodeError as _: # noqa: F841 except json.JSONDecodeError as _: # noqa: F841
json_str = correct_json(json_str)
try: try:
json_str = correct_json(json_str)
return json.loads(json_str) return json.loads(json_str)
except json.JSONDecodeError as _: # noqa: F841 except json.JSONDecodeError as _: # noqa: F841
pass pass
@ -53,6 +53,7 @@ def fix_and_parse_json(
last_brace_index = json_str.rindex("}") last_brace_index = json_str.rindex("}")
json_str = json_str[:last_brace_index+1] json_str = json_str[:last_brace_index+1]
return json.loads(json_str) return json.loads(json_str)
# Can throw a ValueError if there is no "{" or "}" in the json_str
except (json.JSONDecodeError, ValueError) as e: # noqa: F841 except (json.JSONDecodeError, ValueError) as e: # noqa: F841
if try_to_fix_with_gpt: if try_to_fix_with_gpt:
print("Warning: Failed to parse AI output, attempting to fix." print("Warning: Failed to parse AI output, attempting to fix."