From fee344db851501a38270b6692f51b18a0dd22edd Mon Sep 17 00:00:00 2001 From: Hrushikesh Yadav <136978914+HrushiYadav@users.noreply.github.com> Date: Wed, 24 Jun 2026 10:54:36 +0530 Subject: [PATCH] fix(chroma): wrap scalar vector_id in list for delete() (#5703) --- mem0/vector_stores/chroma.py | 2 +- tests/vector_stores/test_chroma.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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):