opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-08-28 20:03:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

r2

Browse Source
This commit is contained in:
jyong 2025-06-05 11:45:53 +08:00
parent 5fe5da7c1d
commit 3e0a10b7ed
1 changed files with 19 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
api/services/rag_pipeline/rag_pipeline.py
View File

@ -430,11 +430,16 @@ class RagPipelineService:
raise ValueError("Workflow not initialized") raise ValueError("Workflow not initialized")
# run draft workflow node # run draft workflow node
datasource_node_data = None
start_at = time.perf_counter() start_at = time.perf_counter()
datasource_nodes = published_workflow.graph_dict.get("nodes", [])
datasource_node_data = published_workflow.graph_dict.get("nodes", {}).get(node_id, {}).get("data", {}) for datasource_node in datasource_nodes:
if datasource_node.get("id") == node_id:
datasource_node_data = datasource_node.get("data", {})
break
if not datasource_node_data: if not datasource_node_data:
raise ValueError("Datasource node data not found") raise ValueError("Datasource node data not found")
datasource_parameters = datasource_node_data.get("datasource_parameters", {}) datasource_parameters = datasource_node_data.get("datasource_parameters", {})
for key, value in datasource_parameters.items(): for key, value in datasource_parameters.items():
if not user_inputs.get(key): if not user_inputs.get(key):
@ -665,7 +670,12 @@ class RagPipelineService:
raise ValueError("Workflow not initialized") raise ValueError("Workflow not initialized")
# get second step node # get second step node
datasource_node_data = published_workflow.graph_dict.get("nodes", {}).get(node_id, {}).get("data", {}) datasource_node_data = None
datasource_nodes = published_workflow.graph_dict.get("nodes", [])
for datasource_node in datasource_nodes:
if datasource_node.get("id") == node_id:
datasource_node_data = datasource_node.get("data", {})
break
if not datasource_node_data: if not datasource_node_data:
raise ValueError("Datasource node data not found") raise ValueError("Datasource node data not found")
datasource_parameters = datasource_node_data.get("datasource_parameters", {}) datasource_parameters = datasource_node_data.get("datasource_parameters", {})
@ -692,7 +702,12 @@ class RagPipelineService:
raise ValueError("Workflow not initialized") raise ValueError("Workflow not initialized")
# get second step node # get second step node
datasource_node_data = draft_workflow.graph_dict.get("nodes", {}).get(node_id, {}).get("data", {}) datasource_node_data = None
datasource_nodes = draft_workflow.graph_dict.get("nodes", [])
for datasource_node in datasource_nodes:
if datasource_node.get("id") == node_id:
datasource_node_data = datasource_node.get("data", {})
break
if not datasource_node_data: if not datasource_node_data:
raise ValueError("Datasource node data not found") raise ValueError("Datasource node data not found")
datasource_parameters = datasource_node_data.get("datasource_parameters", {}) datasource_parameters = datasource_node_data.get("datasource_parameters", {})