From 32aa87d4b2e1d6a01c2dd2ec5f2ce26b29aac6db Mon Sep 17 00:00:00 2001 From: jacoblee93 Date: Thu, 6 Feb 2025 11:52:13 -0800 Subject: [PATCH] Add test --- libs/langgraph/tests/test_messages_state.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/libs/langgraph/tests/test_messages_state.py b/libs/langgraph/tests/test_messages_state.py index 20e829f67..97f38051c 100644 --- a/libs/langgraph/tests/test_messages_state.py +++ b/libs/langgraph/tests/test_messages_state.py @@ -78,6 +78,19 @@ def test_duplicates_in_input(): assert result[0].content == "Hi there again!" +def test_duplicates_in_input_with_remove(): + left = [AIMessage(id="1", content="Hello!")] + right = [ + RemoveMessage(id="1"), + AIMessage(id="1", content="Hi there!"), + AIMessage(id="1", content="Hi there again!"), + ] + result = add_messages(left, right) + assert len(result) == 1 + assert result[0].id == "1" + assert result[0].content == "Hi there again!" + + def test_remove_message(): left = [ HumanMessage(content="Hello", id="1"),