From fdf585d80132cb3272279ce04d08693bb1e28d4f Mon Sep 17 00:00:00 2001 From: Chris Mair Date: Wed, 14 Aug 2024 16:57:38 +0200 Subject: [PATCH] fix a bug that caused backend_query to exit if it got malformed json from backend --- rag/backend_query.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/rag/backend_query.py b/rag/backend_query.py index 21a925e..305d029 100644 --- a/rag/backend_query.py +++ b/rag/backend_query.py @@ -58,6 +58,8 @@ def log(msg: str): connection_timeout = 2.0 + claim = {} + while True: try: @@ -75,10 +77,14 @@ def log(msg: str): time.sleep(1.0) continue + try: + claim = json.loads(response.text) + except: + log("cannot parse JSON response from claim_job") + time.sleep(5.0) + continue break - claim = json.loads(response.text) - conversation = json.loads(claim.get("conversation")) conversation_llm = json.loads(claim.get("conversation_llm")) source = json.loads(claim.get("source"))