Moves config to proper location, improves error logging

jupyterlab · Oct 19, 2023 · 1707f30 · 1707f30
1 parent 12b8e02
commit 1707f30
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 10 deletions.
diff --git a/packages/jupyter-ai-magics/pyproject.toml b/packages/jupyter-ai-magics/pyproject.toml
@@ -54,14 +54,6 @@ all = [
     "boto3"
 ]
 
-[project.entry-points."jupyter_ai.chat_handlers"]
-ask = "jupyter_ai:AskChatHandler"
-clear = "jupyter_ai:ClearChatHandler"
-default = "jupyter_ai:DefaultChatHandler"
-generate = "jupyter_ai:GenerateChatHandler"
-help = "jupyter_ai:HelpChatHandler"
-learn = "jupyter_ai:LearnChatHandler"
-
 [project.entry-points."jupyter_ai.model_providers"]
 ai21 = "jupyter_ai_magics:AI21Provider"
 anthropic = "jupyter_ai_magics:AnthropicProvider"

diff --git a/packages/jupyter-ai/jupyter_ai/extension.py b/packages/jupyter-ai/jupyter_ai/extension.py
@@ -101,9 +101,10 @@ def initialize_settings(self):
         for chat_handler_ep in chat_handler_eps:
             try:
                 chat_handler = chat_handler_ep.load()
-            except:
+            except Exception as err:
                 self.log.error(
-                    f"Unable to load chat handler class from entry point `{chat_handler_ep.name}`."
+                    f"Unable to load chat handler class from entry point `{chat_handler_ep.name}`: "
+                    + f"Unexpected {err=}, {type(err)=}"
                 )
                 continue
 

diff --git a/packages/jupyter-ai/pyproject.toml b/packages/jupyter-ai/pyproject.toml
@@ -40,6 +40,14 @@ dependencies = [
 
 dynamic = ["version", "description", "authors", "urls", "keywords"]
 
+[project.entry-points."jupyter_ai.chat_handlers"]
+ask = "jupyter_ai:chat_handlers:AskChatHandler"
+clear = "jupyter_ai:chat_handlers:ClearChatHandler"
+default = "jupyter_ai:chat_handlers:DefaultChatHandler"
+generate = "jupyter_ai:chat_handlers:GenerateChatHandler"
+help = "jupyter_ai:chat_handlers:HelpChatHandler"
+learn = "jupyter_ai:chat_handlers:LearnChatHandler"
+
 [project.entry-points."jupyter_ai.default_tasks"]
 core_default_tasks = "jupyter_ai:tasks"