kaku/PyGetGPT
1
0
Fork 0
mirror of https://github.com/Kakune55/PyGetGPT.git synced 2025-06-28 15:58:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

修复了用户敏感词导致的异常

Browse Source
This commit is contained in:
Kakune55 2023-11-04 11:44:45 +08:00
parent d3f2d5f6d4
commit 1cccdce99a
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
server.py
View File

@ -21,6 +21,9 @@ def request_TY(userkey,prompt,history = ""):
# 发送POST请求 # 发送POST请求
response = json.loads(requests.post(url, json=data ,headers=header).text) response = json.loads(requests.post(url, json=data ,headers=header).text)
print(response) print(response)
if 'code' in response:
if response['code'] == 'DataInspectionFailed':
return response['message']
reduce_value(userkey, response["usage"]["total_tokens"]) reduce_value(userkey, response["usage"]["total_tokens"])
return response['output']['text'] return response['output']['text']