Unverified Commit fc64cdee authored by Jyong's avatar Jyong Committed by GitHub

fix mivlus delete by ids error (#2573)

Co-authored-by: 's avatarjyong <jyong@dify.ai>
parent 0c0e96c5
...@@ -7,4 +7,4 @@ class Field(Enum): ...@@ -7,4 +7,4 @@ class Field(Enum):
GROUP_KEY = "group_id" GROUP_KEY = "group_id"
VECTOR = "vector" VECTOR = "vector"
TEXT_KEY = "text" TEXT_KEY = "text"
PRIMARY_KEY = " id" PRIMARY_KEY = "id"
...@@ -124,7 +124,12 @@ class MilvusVector(BaseVector): ...@@ -124,7 +124,12 @@ class MilvusVector(BaseVector):
def delete_by_ids(self, doc_ids: list[str]) -> None: def delete_by_ids(self, doc_ids: list[str]) -> None:
self._client.delete(collection_name=self._collection_name, pks=doc_ids) result = self._client.query(collection_name=self._collection_name,
filter=f'metadata["doc_id"] in {doc_ids}',
output_fields=["id"])
if result:
ids = [item["id"] for item in result]
self._client.delete(collection_name=self._collection_name, pks=ids)
def delete(self) -> None: def delete(self) -> None:
alias = uuid4().hex alias = uuid4().hex
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment