[Improvement] add delete functionality to zilliz DB (#912)
Co-authored-by: Deven Patel <deven298@yahoo.com>
This commit is contained in:
@@ -138,7 +138,7 @@ app = App.from_config(yaml_path="config.yaml")
|
|||||||
vectordb:
|
vectordb:
|
||||||
provider: zilliz
|
provider: zilliz
|
||||||
config:
|
config:
|
||||||
collection_name: 'zilliz-app'
|
collection_name: 'zilliz_app'
|
||||||
uri: https://xxxx.api.gcp-region.zillizcloud.com
|
uri: https://xxxx.api.gcp-region.zillizcloud.com
|
||||||
token: xxx
|
token: xxx
|
||||||
vector_dim: 1536
|
vector_dim: 1536
|
||||||
|
|||||||
@@ -222,3 +222,16 @@ class ZillizVectorDB(BaseVectorDB):
|
|||||||
if not isinstance(name, str):
|
if not isinstance(name, str):
|
||||||
raise TypeError("Collection name must be a string")
|
raise TypeError("Collection name must be a string")
|
||||||
self.config.collection_name = name
|
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,
|
||||||
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user