klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Add] Error handling for update method in OSS & platform code. (#1939)

Browse Source
This commit is contained in:
Parshva Daftari
2024-10-08 15:04:59 +05:30
committed by GitHub
parent ab862d0d40
commit c689f94c52
3 changed files with 15 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_main.py
View File

@@ -127,11 +127,15 @@ def test_search(memory_instance, version, enable_graph):
def test_update(memory_instance):
memory_instance.embedding_model = Mock()
memory_instance.embedding_model.embed = Mock(return_value=[0.1, 0.2, 0.3])
memory_instance._update_memory = Mock()
result = memory_instance.update("test_id", "Updated memory")
memory_instance._update_memory.assert_called_once_with("test_id", "Updated memory")
memory_instance._update_memory.assert_called_once_with("test_id", "Updated memory", {"Updated memory": [0.1, 0.2, 0.3]})
assert result["message"] == "Memory updated successfully!"