Revised to support debug mode from command line

This commit is contained in:
kinance
2023-04-08 12:39:57 +09:00
parent c0aacac72e
commit 91fe21e64d
5 changed files with 14 additions and 5 deletions

View File

@@ -61,7 +61,7 @@ def fix_json(json_str: str, schema: str, debug=False) -> str:
result_string = call_ai_function(
function_string, args, description_string, model=cfg.fast_llm_model
)
if debug:
if cfg.debug_mode:
print("------------ JSON FIX ATTEMPT ---------------")
print(f"Original JSON: {json_str}")
print("-----------")