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