소스 검색

fix mivlus delete by ids error (#2573)

Co-authored-by: jyong <jyong@dify.ai>
tags/0.5.7
Jyong 1 년 전
부모
커밋
fc64cdee64
No account linked to committer's email address
2개의 변경된 파일7개의 추가작업 그리고 2개의 파일을 삭제
  1. 1
    1
      api/core/rag/datasource/vdb/field.py
  2. 6
    1
      api/core/rag/datasource/vdb/milvus/milvus_vector.py

+ 1
- 1
api/core/rag/datasource/vdb/field.py 파일 보기

@@ -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 파일 보기

@@ -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

Loading…
취소
저장