Explorar el Código

fix: optimize unique document filtering with set (#10082)

tags/0.11.0
omr hace 1 año
padre
commit
11ca1bec0b
No account linked to committer's email address
Se han modificado 1 ficheros con 8 adiciones y 9 borrados
  1. 8
    9
      api/core/rag/rerank/rerank_model.py

+ 8
- 9
api/core/rag/rerank/rerank_model.py Ver fichero

:return: :return:
""" """
docs = [] docs = []
doc_id = []
doc_id = set()
unique_documents = [] unique_documents = []
dify_documents = [item for item in documents if item.provider == "dify"]
external_documents = [item for item in documents if item.provider == "external"]
for document in dify_documents:
if document.metadata["doc_id"] not in doc_id:
doc_id.append(document.metadata["doc_id"])
for document in documents:
if document.provider == "dify" and document.metadata["doc_id"] not in doc_id:
doc_id.add(document.metadata["doc_id"])
docs.append(document.page_content) docs.append(document.page_content)
unique_documents.append(document) unique_documents.append(document)
for document in external_documents:
docs.append(document.page_content)
unique_documents.append(document)
elif document.provider == "external":
if document not in unique_documents:
docs.append(document.page_content)
unique_documents.append(document)


documents = unique_documents documents = unique_documents



Cargando…
Cancelar
Guardar