diff --git a/src/crewai/memory/long_term/long_term_memory.py b/src/crewai/memory/long_term/long_term_memory.py index d4b6f3bc8..5625531e8 100644 --- a/src/crewai/memory/long_term/long_term_memory.py +++ b/src/crewai/memory/long_term/long_term_memory.py @@ -66,7 +66,7 @@ class LongTermMemory(Memory): self, event=MemorySaveCompletedEvent( value=item.task, - metadata=item.metadata, + metadata=metadata, agent_role=item.agent, save_time_ms=(time.time() - start_time) * 1000, source_type="long_term_memory", diff --git a/tests/memory/test_long_term_memory.py b/tests/memory/test_long_term_memory.py index bf4c9285f..88fc4b791 100644 --- a/tests/memory/test_long_term_memory.py +++ b/tests/memory/test_long_term_memory.py @@ -1,15 +1,17 @@ -import pytest -from unittest.mock import ANY from collections import defaultdict +from unittest.mock import ANY + +import pytest + from crewai.events.event_bus import crewai_event_bus +from crewai.events.types.memory_events import ( + MemoryQueryCompletedEvent, + MemoryQueryStartedEvent, + MemorySaveCompletedEvent, + MemorySaveStartedEvent, +) from crewai.memory.long_term.long_term_memory import LongTermMemory from crewai.memory.long_term.long_term_memory_item import LongTermMemoryItem -from crewai.events.types.memory_events import ( - MemorySaveStartedEvent, - MemorySaveCompletedEvent, - MemoryQueryStartedEvent, - MemoryQueryCompletedEvent, -) @pytest.fixture @@ -98,7 +100,7 @@ def test_long_term_memory_search_events(long_term_memory): test_query = "test query" - long_term_memory.search(test_query, latest_n=5) + long_term_memory.search(test_query, limit=5) assert len(events["MemoryQueryStartedEvent"]) == 1 assert len(events["MemoryQueryCompletedEvent"]) == 1 @@ -151,7 +153,7 @@ def test_save_and_search(long_term_memory): metadata={"task": "test_task", "quality": 0.5}, ) long_term_memory.save(memory) - find = long_term_memory.search("test_task", latest_n=5)[0] + find = long_term_memory.search("test_task", limit=5)[0] assert find["score"] == 0.5 assert find["datetime"] == "test_datetime" assert find["metadata"]["agent"] == "test_agent"