Fix external knowledge Issues: (#17685) (#17843)

This commit is contained in:
briqt 2025-04-11 15:37:27 +08:00 committed by GitHub
parent cc08451eb8
commit 91cfa90503
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 3 additions and 0 deletions

View File

@ -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(),

View File

@ -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,

View File

@ -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",