feat: 修改 mian 文件

This commit is contained in:
zty 2024-10-29 17:15:23 +08:00
parent 80fef129a9
commit 694e52fef2
1 changed files with 6 additions and 1 deletions

View File

@ -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)