From c0ac9f81fa54a34b982b05654fa45ec43717c9dc Mon Sep 17 00:00:00 2001 From: Hrushikesh Yadav <136978914+HrushiYadav@users.noreply.github.com> Date: Tue, 23 Jun 2026 16:48:47 +0530 Subject: [PATCH] fix(milvus): wrap scalar vector_id in list for delete() (#5704) --- mem0/vector_stores/milvus.py | 2 +- tests/vector_stores/test_milvus.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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):