GPT评价错误提示中止

This commit is contained in:
Cp0204 2023-06-26 13:31:32 +08:00 committed by GitHub
parent a2f02f92df
commit 2e3727e9cf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions

View File

@ -184,7 +184,11 @@ def generation_ai(pname, _class=0, _type=1, opts=None):
} }
) )
response_text = response.json() response_text = response.json()
return 5, response_text["choices"][0]["message"]["content"].strip() if "error" in response_text:
print("\nOpenAI API 调用错误:\n", response_text["error"]["message"])
exit()
else:
return 5, response_text["choices"][0]["message"]["content"].strip()
# 查询全部评价 # 查询全部评价