Merge pull request #41 from Cp0204/patch-1

GPT评价错误提示中止
2023-06-27 21:59:05 +08:00 · 2023-06-27 21:59:05 +08:00 · 808476ee92
parent a2f02f92df 2e3727e9cf
commit 808476ee92
1 changed files with 5 additions and 1 deletions
--- a/jd_comment.py
+++ b/jd_comment.py
@ -184,6 +184,10 @@ def generation_ai(pname, _class=0, _type=1, opts=None):
        }
    )
    response_text = response.json()
+    if "error" in response_text:
+        print("\nOpenAI API 调用错误：\n", response_text["error"]["message"])
+        exit()
+    else:
        return 5, response_text["choices"][0]["message"]["content"].strip()