diff --git a/lib/crewai/tests/memory/test_short_term_memory.py b/lib/crewai/tests/memory/test_short_term_memory.py index a619c51ba..075b1499c 100644 --- a/lib/crewai/tests/memory/test_short_term_memory.py +++ b/lib/crewai/tests/memory/test_short_term_memory.py @@ -81,6 +81,9 @@ def test_short_term_memory_search_events(short_term_memory): "from_agent": None, "agent_role": None, "agent_id": None, + "event_id": ANY, + "parent_event_id": None, + "emission_sequence": ANY, "query": "test value", "limit": 3, "score_threshold": 0.35, @@ -98,6 +101,9 @@ def test_short_term_memory_search_events(short_term_memory): "from_agent": None, "agent_role": None, "agent_id": None, + "event_id": ANY, + "parent_event_id": None, + "emission_sequence": ANY, "query": "test value", "results": [], "limit": 3, @@ -150,6 +156,9 @@ def test_short_term_memory_save_events(short_term_memory): "from_agent": None, "agent_role": None, "agent_id": None, + "event_id": ANY, + "parent_event_id": None, + "emission_sequence": ANY, "value": "test value", "metadata": {"task": "test_task"}, } @@ -166,6 +175,9 @@ def test_short_term_memory_save_events(short_term_memory): "from_agent": None, "agent_role": None, "agent_id": None, + "event_id": ANY, + "parent_event_id": None, + "emission_sequence": ANY, "value": "test value", "metadata": {"task": "test_task"}, "save_time_ms": ANY,