bugfix: decode.py

SCys · May 17, 2023 · 058783a · 058783a
1 parent 29039bc
commit 058783a
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/decode.py b/decode.py
@@ -4,10 +4,13 @@
 with open("main.log") as f:
     i = 1
     for line in f.readlines():
-        l = json.loads(line)
-        result = l['choices'][0].get('text') if l['choices'][0].get('text') else l['choices'][0]['message']['content']
-        print(f"{l['user']['name']} --- {l['user']['id']} --- {l['prompt']} --- {result}")
-        print()
-        print(i, "---------------------------------------------------------------------------------------------------")
-        print()
-        i+=1
+        try:
+            l = json.loads(line)
+            result = l['choices'][0].get('text') if l['choices'][0].get('text') else l['choices'][0]['message']['content']
+            print(f"{l['user']['name']} --- {l['user']['id']} --- {l['prompt']} --- {result}")
+            print()
+            print(i, "---------------------------------------------------------------------------------------------------")
+            print()
+            i+=1
+        except (json.decoder.JSONDecodeError, KeyError):
+            continue