youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-06 02:18:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor single-node workflow runner helpers

Browse Source
This commit is contained in:
-LAN-
2026-01-24 00:35:19 +08:00
parent 8d45755303
commit 74e7e6fa8b
1 changed files with 6 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
api/core/app/apps/workflow_app_runner.py
View File

@ -166,18 +166,22 @@ class WorkflowBasedAppRunner:
# Determine which type of single node execution and get graph/variable_pool # Determine which type of single node execution and get graph/variable_pool
if single_iteration_run: if single_iteration_run:
graph, variable_pool = self._get_graph_and_variable_pool_of_single_iteration( graph, variable_pool = self._get_graph_and_variable_pool_for_single_node_run(
workflow=workflow, workflow=workflow,
node_id=single_iteration_run.node_id, node_id=single_iteration_run.node_id,
user_inputs=dict(single_iteration_run.inputs), user_inputs=dict(single_iteration_run.inputs),
graph_runtime_state=graph_runtime_state, graph_runtime_state=graph_runtime_state,
node_type_filter_key="iteration_id",
node_type_label="iteration",
) )
elif single_loop_run: elif single_loop_run:
graph, variable_pool = self._get_graph_and_variable_pool_of_single_loop( graph, variable_pool = self._get_graph_and_variable_pool_for_single_node_run(
workflow=workflow, workflow=workflow,
node_id=single_loop_run.node_id, node_id=single_loop_run.node_id,
user_inputs=dict(single_loop_run.inputs), user_inputs=dict(single_loop_run.inputs),
graph_runtime_state=graph_runtime_state, graph_runtime_state=graph_runtime_state,
node_type_filter_key="loop_id",
node_type_label="loop",
) )
else: else:
raise ValueError("Neither single_iteration_run nor single_loop_run is specified") raise ValueError("Neither single_iteration_run nor single_loop_run is specified")
@ -314,44 +318,6 @@ class WorkflowBasedAppRunner:
return graph, variable_pool return graph, variable_pool
def _get_graph_and_variable_pool_of_single_iteration(
self,
workflow: Workflow,
node_id: str,
user_inputs: dict[str, Any],
graph_runtime_state: GraphRuntimeState,
) -> tuple[Graph, VariablePool]:
"""
Get variable pool of single iteration
"""
return self._get_graph_and_variable_pool_for_single_node_run(
workflow=workflow,
node_id=node_id,
user_inputs=user_inputs,
graph_runtime_state=graph_runtime_state,
node_type_filter_key="iteration_id",
node_type_label="iteration",
)
def _get_graph_and_variable_pool_of_single_loop(
self,
workflow: Workflow,
node_id: str,
user_inputs: dict[str, Any],
graph_runtime_state: GraphRuntimeState,
) -> tuple[Graph, VariablePool]:
"""
Get variable pool of single loop
"""
return self._get_graph_and_variable_pool_for_single_node_run(
workflow=workflow,
node_id=node_id,
user_inputs=user_inputs,
graph_runtime_state=graph_runtime_state,
node_type_filter_key="loop_id",
node_type_label="loop",
)
def _handle_event(self, workflow_entry: WorkflowEntry, event: GraphEngineEvent): def _handle_event(self, workflow_entry: WorkflowEntry, event: GraphEngineEvent):
""" """
Handle event Handle event