diff --git a/main.py b/main.py index 0f4d14a..5c3c5eb 100644 --- a/main.py +++ b/main.py @@ -330,7 +330,12 @@ async def extract_info( conf.CHAT_API, json={"model": conf.CHAT_MODEL, "prompt": prompt, "stream": False}, ) - info = json.loads((r.json()["response"])) + # info = json.loads((r.json()["response"])) + try: + # info = json.loads(r.json()["response"]) + info = r.json().get("response", {}) + except json.JSONDecodeError as e: + print(f"JSONDecodeError: {e}") return JSONResponse(content=info)