diff --git a/server.py b/server.py index 88ae93b..4d2ffe6 100644 --- a/server.py +++ b/server.py @@ -21,6 +21,9 @@ def request_TY(userkey,prompt,history = ""): # 发送POST请求 response = json.loads(requests.post(url, json=data ,headers=header).text) print(response) + if 'code' in response: + if response['code'] == 'DataInspectionFailed': + return response['message'] reduce_value(userkey, response["usage"]["total_tokens"]) return response['output']['text']