Memory Reset (#2558)

This commit is contained in:
Dev Khant
2025-04-16 16:36:45 +05:30
committed by GitHub
parent 3613e2f14a
commit b8e4d0980a
3 changed files with 36 additions and 25 deletions

View File

@@ -195,21 +195,6 @@ def test_delete_all(memory_instance, version, enable_graph):
assert result["message"] == "Memories deleted successfully!"
def test_reset(memory_instance):
memory_instance.vector_store.delete_col = Mock()
# persisting vector store to make sure previous collection is deleted
initial_vector_store = memory_instance.vector_store
memory_instance.db.reset = Mock()
with patch.object(VectorStoreFactory, "create", return_value=Mock()) as mock_create:
memory_instance.reset()
initial_vector_store.delete_col.assert_called_once()
memory_instance.db.reset.assert_called_once()
mock_create.assert_called_once_with(
memory_instance.config.vector_store.provider, memory_instance.config.vector_store.config
)
@pytest.mark.parametrize(
"version, enable_graph, expected_result",