From bbd3238d098ab8fd5b4e7037d53bae7078a8f80c Mon Sep 17 00:00:00 2001 From: "David L. Qiu" Date: Tue, 6 Aug 2024 15:20:39 -0700 Subject: [PATCH] fix unit tests --- packages/jupyter-ai/jupyter_ai/tests/test_handlers.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/packages/jupyter-ai/jupyter_ai/tests/test_handlers.py b/packages/jupyter-ai/jupyter_ai/tests/test_handlers.py index d2e73ce6c..2ed6c7755 100644 --- 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={} )