Migration per openai migrate

jupyterlab · Dec 27, 2023 · f564b1c · f564b1c
1 parent deaedda
commit f564b1c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
@@ -534,7 +534,7 @@ def is_api_key_exc(cls, e: Exception):
         """
         import openai
 
-        if isinstance(e, openai.error.AuthenticationError):
+        if isinstance(e, openai.AuthenticationError):
             error_details = e.json_body.get("error", {})
             return error_details.get("code") == "invalid_api_key"
         return False
@@ -574,7 +574,7 @@ def is_api_key_exc(cls, e: Exception):
         """
         import openai
 
-        if isinstance(e, openai.error.AuthenticationError):
+        if isinstance(e, openai.AuthenticationError):
             error_details = e.json_body.get("error", {})
             return error_details.get("code") == "invalid_api_key"
         return False
@@ -620,7 +620,7 @@ def is_api_key_exc(cls, e: Exception):
         """
         import openai
 
-        if isinstance(e, openai.error.AuthenticationError):
+        if isinstance(e, openai.AuthenticationError):
             error_details = e.json_body.get("error", {})
             return error_details.get("code") == "invalid_api_key"
         return False