minor fix

neulab · Apr 15, 2024 · 85753e4 · 85753e4
1 parent 40255a1
commit 85753e4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/prompt2model/utils/api_tools.py b/prompt2model/utils/api_tools.py
@@ -169,14 +169,14 @@ async def _throttled_completion_acreate(
                         if isinstance(
                             e,
                             (
-                                openai.ServiceUnavailableError,
+                                openai.APIStatusError,
                                 openai.APIError,
                             ),
                         ):
                             logging.warning(
                                 ERROR_ERRORS_TO_MESSAGES[type(e)].format(e=e)
                             )
-                        elif isinstance(e, openai.InvalidRequestError):
+                        elif isinstance(e, openai.BadRequestError):
                             logging.warning(ERROR_ERRORS_TO_MESSAGES[type(e)])
                             return {
                                 "choices": [