fix(api): add error handling for OpenAI API failures

Byaidu · Feb 12, 2025 · 2e2e878 · 2e2e878
1 parent fece247
commit 2e2e878
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pdf2zh/translator.py b/pdf2zh/translator.py
@@ -400,6 +400,11 @@ def do_translate(self, text) -> str:
             **self.options,
             messages=self.prompt(text, self.prompttext),
         )
+        if not response.choices:
+            if hasattr(response, "error"):
+                raise ValueError("Empty response from OpenAI API", response.error)
+            else:
+                raise ValueError("Empty response from OpenAI API")
         return response.choices[0].message.content.strip()
 
     def get_formular_placeholder(self, id: int):