From 601cc0e18f118953c8ed819e698475a6b8c35bb3 Mon Sep 17 00:00:00 2001
From: krassowski <5832902+krassowski@users.noreply.github.com>
Date: Sat, 7 Sep 2024 11:41:36 +0100
Subject: [PATCH] Fix `RunnableWithMessageHistory` import

---
 packages/jupyter-ai/jupyter_ai/chat_handlers/default.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/packages/jupyter-ai/jupyter_ai/chat_handlers/default.py b/packages/jupyter-ai/jupyter_ai/chat_handlers/default.py
index 02256f4eb..c8d0418ff 100644
--- 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