Fix RunnableWithMessageHistory import

jupyterlab · Sep 7, 2024 · 601cc0e · 601cc0e
1 parent 4b45ec2
commit 601cc0e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/packages/jupyter-ai/jupyter_ai/chat_handlers/default.py b/packages/jupyter-ai/jupyter_ai/chat_handlers/default.py
@@ -9,7 +9,8 @@
 )
 from jupyter_ai_magics.providers import BaseProvider
 from langchain_core.messages import AIMessageChunk
-from langchain_core.runnables import ConfigurableFieldSpec, RunnableWithMessageHistory
+from langchain_core.runnables import ConfigurableFieldSpec
+from langchain_core.runnables.history import RunnableWithMessageHistory
 
 from ..models import HumanChatMessage
 from .base import BaseChatHandler, SlashCommandRoutingType