Skip to content

Commit 91cfa90

Browse files
authored
Fix external knowledge Issues: (#17685) (#17843)
1 parent cc08451 commit 91cfa90

File tree

3 files changed

+3
-0
lines changed

3 files changed

+3
-0
lines changed

api/core/rag/retrieval/dataset_retrieval.py

+1
Original file line numberDiff line numberDiff line change
@@ -206,6 +206,7 @@ def retrieve(
206206
source = {
207207
"dataset_id": item.metadata.get("dataset_id"),
208208
"dataset_name": item.metadata.get("dataset_name"),
209+
"document_id": item.metadata.get("document_id") or item.metadata.get("title"),
209210
"document_name": item.metadata.get("title"),
210211
"data_source_type": "external",
211212
"retriever_from": invoke_from.to_source(),

api/core/tools/utils/dataset_retriever/dataset_retriever_tool.py

+1
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,7 @@ def _run(self, query: str) -> str:
8686
"position": position,
8787
"dataset_id": item.metadata.get("dataset_id"),
8888
"dataset_name": item.metadata.get("dataset_name"),
89+
"document_id": item.metadata.get("document_id") or item.metadata.get("title"),
8990
"document_name": item.metadata.get("title"),
9091
"data_source_type": "external",
9192
"retriever_from": self.retriever_from,

api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py

+1
Original file line numberDiff line numberDiff line change
@@ -259,6 +259,7 @@ def _fetch_dataset_retriever(self, node_data: KnowledgeRetrievalNodeData, query:
259259
"_source": "knowledge",
260260
"dataset_id": item.metadata.get("dataset_id"),
261261
"dataset_name": item.metadata.get("dataset_name"),
262+
"document_id": item.metadata.get("document_id") or item.metadata.get("title"),
262263
"document_name": item.metadata.get("title"),
263264
"data_source_type": "external",
264265
"retriever_from": "workflow",

0 commit comments

Comments
 (0)