[Improvement] add delete functionality to zilliz DB (#912)

Co-authored-by: Deven Patel <deven298@yahoo.com>
This commit is contained in:
Deven Patel
2023-11-07 02:46:39 -08:00
committed by GitHub
parent a1de238716
commit 0533da72d7
2 changed files with 14 additions and 1 deletions

View File

@@ -222,3 +222,16 @@ class ZillizVectorDB(BaseVectorDB):
if not isinstance(name, str):
raise TypeError("Collection name must be a string")
self.config.collection_name = name
def delete(self, keys: Union[list, str, int]):
"""
Delete the embeddings from DB. Zilliz only support deleting with keys.
:param keys: Primary keys of the table entries to delete.
:type keys: Union[list, str, int]
"""
self.client.delete(
collection_name=self.config.collection_name,
pks=keys,
)