diff --git a/packages/jupyter-ai/jupyter_ai/chat_handlers/default.py b/packages/jupyter-ai/jupyter_ai/chat_handlers/default.py index 1c20fad9a..c674a383b 100644 --- a/packages/jupyter-ai/jupyter_ai/chat_handlers/default.py +++ b/packages/jupyter-ai/jupyter_ai/chat_handlers/default.py @@ -3,7 +3,6 @@ from jupyter_ai.models import ChatMessage, ClearMessage, HumanChatMessage from jupyter_ai_magics.providers import BaseProvider from langchain.chains import ConversationChain -from langchain.chat_models.base import BaseChatModel from langchain.memory import ConversationBufferWindowMemory from langchain.prompts import ( ChatPromptTemplate,