refactor: update MemoryClient to use filters for entity parameters
This commit is contained in:
+7
-4
@@ -57,7 +57,10 @@ def test_add(memory_instance, version):
|
||||
memory_instance.config.version = version
|
||||
memory_instance._add_to_vector_store = Mock(return_value=[{"memory": "Test memory", "event": "ADD"}])
|
||||
|
||||
result = memory_instance.add(messages=[{"role": "user", "content": "Test message"}], user_id="test_user")
|
||||
result = memory_instance.add(
|
||||
messages=[{"role": "user", "content": "Test message"}],
|
||||
filters={"user_id": "test_user"},
|
||||
)
|
||||
|
||||
assert "results" in result
|
||||
assert result["results"] == [{"memory": "Test memory", "event": "ADD"}]
|
||||
@@ -105,7 +108,7 @@ def test_search(memory_instance, version):
|
||||
|
||||
with patch("mem0.memory.main.lemmatize_for_bm25", return_value="test query"), \
|
||||
patch("mem0.memory.main.extract_entities", return_value=[]):
|
||||
result = memory_instance.search("test query", user_id="test_user")
|
||||
result = memory_instance.search("test query", filters={"user_id": "test_user"})
|
||||
|
||||
assert "results" in result
|
||||
assert len(result["results"]) == 2
|
||||
@@ -185,7 +188,7 @@ def test_delete_all(memory_instance, version):
|
||||
memory_instance.vector_store.reset = Mock()
|
||||
memory_instance._delete_memory = Mock()
|
||||
|
||||
result = memory_instance.delete_all(user_id="test_user")
|
||||
result = memory_instance.delete_all(filters={"user_id": "test_user"})
|
||||
|
||||
assert memory_instance._delete_memory.call_count == 2
|
||||
# Ensure the collection is NOT dropped — only matched memories should be removed
|
||||
@@ -206,7 +209,7 @@ def test_get_all(memory_instance, version, expected_result):
|
||||
mock_memories = [Mock(id="1", payload={"data": "Memory 1", "user_id": "test_user"})]
|
||||
memory_instance.vector_store.list = Mock(return_value=(mock_memories, None))
|
||||
|
||||
result = memory_instance.get_all(user_id="test_user")
|
||||
result = memory_instance.get_all(filters={"user_id": "test_user"})
|
||||
|
||||
assert isinstance(result, dict)
|
||||
assert "results" in result
|
||||
|
||||
Reference in New Issue
Block a user