浏览代码

fix mivlus delete by ids error (#2573)

Co-authored-by: jyong <jyong@dify.ai>
tags/0.5.7
Jyong 1年前
父节点
当前提交
fc64cdee64
没有帐户链接到提交者的电子邮件
共有 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

正在加载...
取消
保存