fix(milvus): wrap scalar vector_id in list for delete() (#5704)
This commit is contained in:
@@ -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):
|
||||
"""
|
||||
|
||||
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user