diff --git a/api/services/rag_pipeline/rag_pipeline.py b/api/services/rag_pipeline/rag_pipeline.py index 1e0578d12f..fdaaa73bcc 100644 --- a/api/services/rag_pipeline/rag_pipeline.py +++ b/api/services/rag_pipeline/rag_pipeline.py @@ -1381,8 +1381,8 @@ class RagPipelineService: datasource_nodes = workflow.graph_dict.get("nodes", []) datasource_plugins = [] for datasource_node in datasource_nodes: - if datasource_node.get("type") == "datasource": - datasource_node_data = datasource_node.get("data", {}) + if datasource_node.get("data", {}).get("type") == "datasource": + datasource_node_data = datasource_node["data"] if not datasource_node_data: continue