Moved the repair output behind the debug flag and removed the "clean"…

… one
letta-ai · Nov 7, 2023 · 4c5d239 · 4c5d239
1 parent 0a1343c
commit 4c5d239
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/memgpt/local_llm/json_parser.py b/memgpt/local_llm/json_parser.py
@@ -122,27 +122,29 @@ def repair_even_worse_json(json_string):
 
 
 def clean_json(raw_llm_output, messages=None, functions=None):
+    from memgpt.utils import printd
+
     """Try a bunch of hacks to parse the data coming out of the LLM"""
     try:
-        print("clean json runs:", raw_llm_output)
+        #printd("clean json runs:", raw_llm_output)
         data = json.loads(raw_llm_output)
     except json.JSONDecodeError:
         try:
-            print("trying adding }")
+            printd("trying adding }")
             data = json.loads(raw_llm_output + "}")
         except json.JSONDecodeError:
             try:
                 repaired = repair_json_string(raw_llm_output)
-                print("trying my repair json:", repaired)
+                printd("trying my repair json:", repaired)
                 data = json.loads(repaired)
             except json.JSONDecodeError:
                 try:
                     repaired = repair_json_string(raw_llm_output)
-                    print("trying my repair json:", repaired)
+                    printd("trying my repair json:", repaired)
                     data = json.loads(repaired)
                 except json.JSONDecodeError:
                     try:
-                        print("trying first_json")
+                        printd("trying first_json")
                         data = extract_first_json(raw_llm_output + "}")
                     except:
                         raise