x

langchain-ai · Jan 6, 2024 · 607a171 · 607a171
1 parent c2da719
commit 607a171
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/libs/core/tests/unit_tests/runnables/test_history.py b/libs/core/tests/unit_tests/runnables/test_history.py
@@ -390,6 +390,7 @@ def factory(session_id):
         )
 
     session = factory("abc")
+    session.clear()
 
     chat_with_history = RunnableWithMessageHistory(
         chat,
@@ -400,7 +401,10 @@ def factory(session_id):
     config = {"configurable": {"session_id": "abc"}}
 
     chat_with_history.invoke(HumanMessage(content="Hi! I'm Bob"), config=config)
-    assert session.messages[:2] == [
+    chat_with_history.invoke(HumanMessage(content="Hi! I'm Alice"), config=config)
+    assert session.messages == [
         HumanMessage(content="Hi! I'm Bob"),
         AIMessage(content="you said: Hi! I'm Bob"),
+        HumanMessage(content="Hi! I'm Alice"),
+        AIMessage(content="you said: Hi! I'm Bob\nHi! I'm Alice"),
     ]