From 1cccdce99a2bb6720ebe030a243a33dd793f46f3 Mon Sep 17 00:00:00 2001
From: Kakune55 <xiao_doubi9637@outlook.com>
Date: Sat, 4 Nov 2023 11:44:45 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E4=BA=86=E7=94=A8=E6=88=B7?=
 =?UTF-8?q?=E6=95=8F=E6=84=9F=E8=AF=8D=E5=AF=BC=E8=87=B4=E7=9A=84=E5=BC=82?=
 =?UTF-8?q?=E5=B8=B8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 server.py | 3 +++
 1 file changed, 3 insertions(+)

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']