AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 12:45:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(graph_engine): parent_parallel_id missing

Browse Source
This commit is contained in:
takatost 2024-08-27 16:45:14 +08:00
parent 4256e9d47f
commit cd52633b0e
1 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
api/core/workflow/graph_engine/graph_engine.py
View File

@ -143,7 +143,13 @@ class GraphEngine:
yield GraphRunFailedEvent(error=str(e)) yield GraphRunFailedEvent(error=str(e))
raise e raise e
def _run(self, start_node_id: str, in_parallel_id: Optional[str] = None) -> Generator[GraphEngineEvent, None, None]: def _run(
self,
start_node_id: str,
in_parallel_id: Optional[str] = None,
parent_parallel_id: Optional[str] = None,
parent_parallel_start_node_id: Optional[str] = None
) -> Generator[GraphEngineEvent, None, None]:
parallel_start_node_id = None parallel_start_node_id = None
if in_parallel_id: if in_parallel_id:
parallel_start_node_id = start_node_id parallel_start_node_id = start_node_id
@ -197,7 +203,9 @@ class GraphEngine:
node_instance=node_instance, node_instance=node_instance,
route_node_state=route_node_state, route_node_state=route_node_state,
parallel_id=in_parallel_id, parallel_id=in_parallel_id,
parallel_start_node_id=parallel_start_node_id parallel_start_node_id=parallel_start_node_id,
parent_parallel_id=parent_parallel_id,
parent_parallel_start_node_id=parent_parallel_start_node_id
) )
for item in generator: for item in generator:
@ -227,8 +235,8 @@ class GraphEngine:
route_node_state=route_node_state, route_node_state=route_node_state,
parallel_id=in_parallel_id, parallel_id=in_parallel_id,
parallel_start_node_id=parallel_start_node_id, parallel_start_node_id=parallel_start_node_id,
parent_parallel_id=in_parallel_id, parent_parallel_id=parent_parallel_id,
parent_parallel_start_node_id=parallel_start_node_id parent_parallel_start_node_id=parent_parallel_start_node_id
) )
raise e raise e
@ -374,7 +382,9 @@ class GraphEngine:
# run node # run node
generator = self._run( generator = self._run(
start_node_id=parallel_start_node_id, start_node_id=parallel_start_node_id,
in_parallel_id=parallel_id in_parallel_id=parallel_id,
parent_parallel_id=parent_parallel_id,
parent_parallel_start_node_id=parent_parallel_start_node_id
) )
for item in generator: for item in generator: