Cleanup after the merge

jupyterlab · Dec 13, 2023 · 014576e · 014576e
1 parent 4ff22e1
commit 014576e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/packages/jupyter-ai/jupyter_ai/chat_handlers/generate.py b/packages/jupyter-ai/jupyter_ai/chat_handlers/generate.py
@@ -221,9 +221,7 @@ class GenerateChatHandler(BaseChatHandler):
     help = "Generates a Jupyter notebook, including name, outline, and section contents"
     routing_type = SlashCommandRoutingType(slash_id="generate")
 
-    def __init__(
-        self, preferred_dir: str, log_dir: Optional[str], *args, **kwargs
-    ):
+    def __init__(self, preferred_dir: str, log_dir: Optional[str], *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.log_dir = Path(log_dir) if log_dir else None
         self.preferred_dir = (

diff --git a/packages/jupyter-ai/jupyter_ai/extension.py b/packages/jupyter-ai/jupyter_ai/extension.py
@@ -178,14 +178,14 @@ def initialize_settings(self):
             "root_chat_handlers": self.settings["jai_root_chat_handlers"],
             "chat_history": self.settings["chat_history"],
             "root_dir": self.serverapp.root_dir,
-            "preferred_dir": self.serverapp.contents_manager.preferred_dir,
             "dask_client_future": dask_client_future,
             "model_parameters": self.settings["model_parameters"],
         }
         default_chat_handler = DefaultChatHandler(**chat_handler_kwargs)
         clear_chat_handler = ClearChatHandler(**chat_handler_kwargs)
         generate_chat_handler = GenerateChatHandler(
             **chat_handler_kwargs,
+            preferred_dir=self.serverapp.contents_manager.preferred_dir,
             log_dir=self.generation_log_dir,
         )
         learn_chat_handler = LearnChatHandler(**chat_handler_kwargs)