diff --git a/mem0/vector_stores/chroma.py b/mem0/vector_stores/chroma.py index 12f43cb4b..b37872643 100644 --- a/mem0/vector_stores/chroma.py +++ b/mem0/vector_stores/chroma.py @@ -169,7 +169,7 @@ class ChromaDB(VectorStoreBase): Args: vector_id (str): ID of the vector to delete. """ - self.collection.delete(ids=vector_id) + self.collection.delete(ids=[vector_id]) def update( self, diff --git a/tests/vector_stores/test_chroma.py b/tests/vector_stores/test_chroma.py index dead40ba4..f13d3d8d2 100644 --- a/tests/vector_stores/test_chroma.py +++ b/tests/vector_stores/test_chroma.py @@ -122,7 +122,7 @@ def test_delete_vector(chromadb_instance): chromadb_instance.delete(vector_id=vector_id) - chromadb_instance.collection.delete.assert_called_once_with(ids=vector_id) + chromadb_instance.collection.delete.assert_called_once_with(ids=[vector_id]) def test_update_vector(chromadb_instance):