diff --git a/mem0/vector_stores/milvus.py b/mem0/vector_stores/milvus.py index 80a4d0ffa..6c8ae885f 100644 --- a/mem0/vector_stores/milvus.py +++ b/mem0/vector_stores/milvus.py @@ -262,7 +262,7 @@ class MilvusDB(VectorStoreBase): Args: vector_id (str): ID of the vector to delete. """ - self.client.delete(collection_name=self.collection_name, ids=vector_id) + self.client.delete(collection_name=self.collection_name, ids=[vector_id]) def update(self, vector_id=None, vector=None, payload=None): """ diff --git a/tests/vector_stores/test_milvus.py b/tests/vector_stores/test_milvus.py index b0109ca56..9acd84140 100644 --- a/tests/vector_stores/test_milvus.py +++ b/tests/vector_stores/test_milvus.py @@ -178,7 +178,7 @@ class TestMilvusDB: mock_milvus_client.delete.assert_called_once_with( collection_name="test_collection", - ids=vector_id + ids=[vector_id] ) def test_get(self, milvus_db, mock_milvus_client):