Przeglądaj źródła

fix mivlus delete by ids error (#2573)

Co-authored-by: jyong <jyong@dify.ai>
tags/0.5.7
Jyong 1 rok temu
rodzic
commit
fc64cdee64
No account linked to committer's email address

+ 1
- 1
api/core/rag/datasource/vdb/field.py Wyświetl plik

@@ -7,4 +7,4 @@ class Field(Enum):
GROUP_KEY = "group_id"
VECTOR = "vector"
TEXT_KEY = "text"
PRIMARY_KEY = " id"
PRIMARY_KEY = "id"

+ 6
- 1
api/core/rag/datasource/vdb/milvus/milvus_vector.py Wyświetl plik

@@ -124,7 +124,12 @@ class MilvusVector(BaseVector):

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:
alias = uuid4().hex

Ładowanie…
Anuluj
Zapisz