use provider_name in auth error message

andrii-i · Dec 12, 2023 · dd25501 · dd25501
1 parent 2b060c9
commit dd25501
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/packages/jupyter-ai/jupyter_ai/chat_handlers/base.py b/packages/jupyter-ai/jupyter_ai/chat_handlers/base.py
@@ -129,8 +129,13 @@ def is_api_key_error(self, e: Exception):
         return isinstance(e, self.API_KEY_EXCEPTIONS)
 
     def handle_api_key_error(self, e: Exception, message: HumanChatMessage):
-        llm_id = f" {self.llm.id}" if self.llm and self.llm.id else ""
-        response = f"Oops! It seems there's an issue with your{llm_id} API key. Please update your{llm_id} API key in the chat Settings."
+        provider_name = ""
+        if hasattr(self.config_manager, "lm_provider") and hasattr(
+            self.config_manager.lm_provider, "name"
+        ):
+            name = getattr(self.config_manager.lm_provider, "name", "")
+            provider_name = f" {name}" if name else ""
+        response = f"Oops! It seems there's an issue with your{provider_name} API key. Please update your{provider_name} API key in the chat Settings."
         self.reply(response, message)
 
     async def _default_handle_exc(self, e: Exception, message: HumanChatMessage):