fix unit tests

jupyterlab · Aug 6, 2024 · bbd3238 · bbd3238
1 parent e5b3e10
commit bbd3238
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/packages/jupyter-ai/jupyter_ai/tests/test_handlers.py b/packages/jupyter-ai/jupyter_ai/tests/test_handlers.py
@@ -5,12 +5,11 @@
 from unittest import mock
 
 import pytest
+from jupyter_ai.extension import DEFAULT_HELP_MESSAGE_TEMPLATE
 from jupyter_ai.chat_handlers import DefaultChatHandler, learn
 from jupyter_ai.config_manager import ConfigManager
 from jupyter_ai.handlers import RootChatHandler
 from jupyter_ai.models import (
-    AgentStreamChunkMessage,
-    AgentStreamMessage,
     ChatClient,
     ClosePendingMessage,
     HumanChatMessage,
@@ -73,6 +72,8 @@ def broadcast_message(message: Message) -> None:
             root_dir="",
             preferred_dir="",
             dask_client_future=None,
+            help_message_template=DEFAULT_HELP_MESSAGE_TEMPLATE,
+            chat_handlers={}
         )