zcdsj
/
cert_recog
5
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
main.py
View File

@ -330,7 +330,12 @@ async def extract_info(
conf.CHAT_API, conf.CHAT_API,
json={"model": conf.CHAT_MODEL, "prompt": prompt, "stream": False}, 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) return JSONResponse(content=info)