From 91cfa9050384f4eb0ebf7af784cb84c9694313e9 Mon Sep 17 00:00:00 2001 From: briqt Date: Fri, 11 Apr 2025 15:37:27 +0800 Subject: [PATCH] Fix external knowledge Issues: (#17685) (#17843) --- api/core/rag/retrieval/dataset_retrieval.py | 1 + api/core/tools/utils/dataset_retriever/dataset_retriever_tool.py | 1 + .../nodes/knowledge_retrieval/knowledge_retrieval_node.py | 1 + 3 files changed, 3 insertions(+) diff --git a/api/core/rag/retrieval/dataset_retrieval.py b/api/core/rag/retrieval/dataset_retrieval.py index e00c989c9..23ea775de 100644 --- a/api/core/rag/retrieval/dataset_retrieval.py +++ b/api/core/rag/retrieval/dataset_retrieval.py @@ -206,6 +206,7 @@ class DatasetRetrieval: source = { "dataset_id": item.metadata.get("dataset_id"), "dataset_name": item.metadata.get("dataset_name"), + "document_id": item.metadata.get("document_id") or item.metadata.get("title"), "document_name": item.metadata.get("title"), "data_source_type": "external", "retriever_from": invoke_from.to_source(), diff --git a/api/core/tools/utils/dataset_retriever/dataset_retriever_tool.py b/api/core/tools/utils/dataset_retriever/dataset_retriever_tool.py index 63260cfac..f661294ec 100644 --- a/api/core/tools/utils/dataset_retriever/dataset_retriever_tool.py +++ b/api/core/tools/utils/dataset_retriever/dataset_retriever_tool.py @@ -86,6 +86,7 @@ class DatasetRetrieverTool(DatasetRetrieverBaseTool): "position": position, "dataset_id": item.metadata.get("dataset_id"), "dataset_name": item.metadata.get("dataset_name"), + "document_id": item.metadata.get("document_id") or item.metadata.get("title"), "document_name": item.metadata.get("title"), "data_source_type": "external", "retriever_from": self.retriever_from, diff --git a/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py b/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py index 860373948..d1e6e5b67 100644 --- a/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py +++ b/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py @@ -259,6 +259,7 @@ class KnowledgeRetrievalNode(LLMNode): "_source": "knowledge", "dataset_id": item.metadata.get("dataset_id"), "dataset_name": item.metadata.get("dataset_name"), + "document_id": item.metadata.get("document_id") or item.metadata.get("title"), "document_name": item.metadata.get("title"), "data_source_type": "external", "retriever_from": "workflow",