diff --git a/decode.py b/decode.py index c8d17eb..02e823b 100644 --- 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