outman/dify
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: resolved the issue of not releasing ThreadPoolExecutor when an exception occurs in _generate_worker

Browse Source 
Signed-off-by: Guowen.Bao <Guowen.Bao@desaysv.com>
This commit is contained in:
Guowen.Bao 2025-02-28 15:39:17 +08:00
parent a9e4f345e9
commit 526f6321d1
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -169,6 +169,7 @@ class GraphEngine:
error=item.route_node_state.failed_reason or "Unknown error.", error=item.route_node_state.failed_reason or "Unknown error.",
exceptions_count=len(handle_exceptions), exceptions_count=len(handle_exceptions),
) )
self._release_thread()
return return
elif isinstance(item, NodeRunSucceededEvent): elif isinstance(item, NodeRunSucceededEvent):
if item.node_type == NodeType.END: if item.node_type == NodeType.END:
@ -195,6 +196,7 @@ class GraphEngine:
except Exception as e: except Exception as e:
logger.exception("Graph run failed") logger.exception("Graph run failed")
yield GraphRunFailedEvent(error=str(e), exceptions_count=len(handle_exceptions)) yield GraphRunFailedEvent(error=str(e), exceptions_count=len(handle_exceptions))
self._release_thread()
return return
# count exceptions to determine partial success # count exceptions to determine partial success
if len(handle_exceptions) > 0: if len(handle_exceptions) > 0: