GPT评价错误提示中止

2023-06-26 13:31:32 +08:00 · 2023-06-26 13:31:32 +08:00 · 2e3727e9cf
parent a2f02f92df
commit 2e3727e9cf
1 changed files with 5 additions and 1 deletions
--- a/jd_comment.py
+++ b/jd_comment.py
@ -184,7 +184,11 @@ def generation_ai(pname, _class=0, _type=1, opts=None):
        }
    )
    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()


 # 查询全部评价