youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-03-04 07:16:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: ensure document id is not missing (#32765)

Browse Source
This commit is contained in:
wangxiaolei
2026-03-02 16:28:31 +08:00
committed by GitHub
parent 8a7ba87349
commit 5c7a293ba7
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
api/core/rag/retrieval/dataset_retrieval.py
View File

@ -248,19 +248,22 @@ class DatasetRetrieval:
retrieval_resource_list = []
# deal with external documents
for item in external_documents:
ext_meta = item.metadata or {}
title = ext_meta.get("title") or ""
doc_id = ext_meta.get("document_id") or title
source = Source(
metadata=SourceMetadata(
source="knowledge",
dataset_id=item.metadata.get("dataset_id"),
dataset_name=item.metadata.get("dataset_name"),
document_id=item.metadata.get("document_id"),
document_name=item.metadata.get("title"),
dataset_id=ext_meta.get("dataset_id") or "",
dataset_name=ext_meta.get("dataset_name") or "",
document_id=str(doc_id),
document_name=ext_meta.get("title") or "",
data_source_type="external",
retriever_from="workflow",
score=item.metadata.get("score"),
doc_metadata=item.metadata,
score=float(ext_meta.get("score") or 0.0),
doc_metadata=ext_meta,
),
title=item.metadata.get("title"),
title=title,
content=item.page_content,
)
retrieval_resource_list.append(source)