Catch openai RatelimitError

zyddnys · Nov 7, 2023 · 72b31cb · 72b31cb
1 parent c03cfb7
commit 72b31cb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/manga_translator/translators/chatgpt.py b/manga_translator/translators/chatgpt.py
@@ -1,5 +1,6 @@
 import re
 import requests
+import openai.error
 import litellm
 import litellm.exceptions
 import asyncio
@@ -147,7 +148,7 @@ async def _translate(self, from_lang: str, to_lang: str, queries: List[str]) ->
                 try:
                     response = await request_task
                     break
-                except litellm.exceptions.RateLimitError: # Server returned ratelimit response
+                except litellm.exceptions.RateLimitError or openai.error.RateLimitError: # Server returned ratelimit response
                     ratelimit_attempt += 1
                     if ratelimit_attempt >= self._RATELIMIT_RETRY_ATTEMPTS:
                         raise