Ver código fonte

fix:external dataset weight rerank bug (#24533)

Co-authored-by: huangzhuo <huangzhuo1@xiaomi.com>
tags/1.8.0
huangzhuo1949 2 meses atrás
pai
commit
98473e9d4f
Nenhuma conta vinculada ao e-mail do autor do commit
1 arquivos alterados com 8 adições e 1 exclusões
  1. 8
    1
      api/core/rag/rerank/weight_rerank.py

+ 8
- 1
api/core/rag/rerank/weight_rerank.py Ver arquivo

unique_documents = [] unique_documents = []
doc_ids = set() doc_ids = set()
for document in documents: for document in documents:
if document.metadata is not None and document.metadata["doc_id"] not in doc_ids:
if (
document.provider == "dify"
and document.metadata is not None
and document.metadata["doc_id"] not in doc_ids
):
doc_ids.add(document.metadata["doc_id"]) doc_ids.add(document.metadata["doc_id"])
unique_documents.append(document) unique_documents.append(document)
else:
if document not in unique_documents:
unique_documents.append(document)


documents = unique_documents documents = unique_documents



Carregando…
Cancelar
Salvar