improving error catching for openai result

RobokopU24 · Oct 28, 2024 · 09aff44 · 09aff44
1 parent aaf6b2c
commit 09aff44
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/parsers/LitCoin/src/bagel/bagel_gpt.py b/parsers/LitCoin/src/bagel/bagel_gpt.py
@@ -110,10 +110,10 @@ def query(prompt, requests_session):
         print(f'openai json did not contain expected key {k}: {response.json()}')
         raise k
 
-    chunk = content[content.index("["):(content.rindex("]") + 1)]
     try:
+        chunk = content[content.index("["):(content.rindex("]") + 1)]
         output = json.loads(chunk)
-    except json.JSONDecodeError as e:
-        print(f'openai results not contain valid json chunk, chunk: {chunk}')
-        output = []
+    except (json.JSONDecodeError, ValueError) as e:
+        print(f'openai results did not contain valid json chunk: {content}')
+        raise e
     return output