修复客户端网络问题导致异常崩溃

This commit is contained in:
Kakune55 2023-11-04 11:54:42 +08:00
parent 1cccdce99a
commit 4c9a602244

View File

@ -28,7 +28,11 @@ data["userkey"] = config["userkey"]
while True: while True:
data["prompt"] = str(input(">> You:\n")) data["prompt"] = str(input(">> You:\n"))
try:
req = json.loads(requests.post(url=config['url'],json=data).text) req = json.loads(requests.post(url=config['url'],json=data).text)
except:
print("Internet Err : Can not connect Server")
break
if req["code"] == 200: if req["code"] == 200:
print(f"\n>> AI:\n{req['output']}\n\n<剩余tokens:{req['surplus']} >") print(f"\n>> AI:\n{req['output']}\n\n<剩余tokens:{req['surplus']} >")
elif req["code"] == 403: elif req["code"] == 403: