put debug setting to cfg and use it in when calling chat.chat_with_at and fix_json

This commit is contained in:
Alexander Nikulin
2023-04-09 14:33:30 +04:00
parent d8a7a811c8
commit 546d8783e7
3 changed files with 10 additions and 2 deletions

View File

@@ -40,7 +40,7 @@ def fix_and_parse_json(json_str: str, try_to_fix_with_gpt: bool = True):
if try_to_fix_with_gpt:
print(f"Warning: Failed to parse AI output, attempting to fix.\n If you see this warning frequently, it's likely that your prompt is confusing the AI. Try changing it up slightly.")
# Now try to fix this up using the ai_functions
ai_fixed_json = fix_json(json_str, json_schema, False)
ai_fixed_json = fix_json(json_str, json_schema, cfg.debug)
if ai_fixed_json != "failed":
return json.loads(ai_fixed_json)
else: