JDSS/mem0
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(chroma): wrap scalar vector_id in list for delete() (#5703)

Browse Source
This commit is contained in:
Hrushikesh Yadav
2026-06-24 10:54:36 +05:30
committed by GitHub
parent b611f69381
commit fee344db85
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mem0/vector_stores/chroma.py
+1 -1
View File
@@ -169,7 +169,7 @@ class ChromaDB(VectorStoreBase):
Args: Args:
vector_id (str): ID of the vector to delete. vector_id (str): ID of the vector to delete.
""" """
self.collection.delete(ids=vector_id) self.collection.delete(ids=[vector_id])
def update( def update(
self, self,
tests/vector_stores/test_chroma.py
+1 -1
View File
@@ -122,7 +122,7 @@ def test_delete_vector(chromadb_instance):
chromadb_instance.delete(vector_id=vector_id) 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): def test_update_vector(chromadb_instance):