fix(workflow): "Max submit count reached" error occurred when executing workflow as tool in iteration (#8595)

This commit is contained in:
takatost
2024-09-20 19:47:25 +08:00
committed by GitHub
parent 7f3282ec04
commit e0a3307563
2 changed files with 8 additions and 3 deletions

View File

@@ -180,16 +180,20 @@ class GraphEngine:
# trigger graph run success event
yield GraphRunSucceededEvent(outputs=self.graph_runtime_state.outputs)
self._release_thread()
except GraphRunFailedError as e:
yield GraphRunFailedEvent(error=e.error)
self._release_thread()
return
except Exception as e:
logger.exception("Unknown Error when graph running")
yield GraphRunFailedEvent(error=str(e))
self._release_thread()
raise e
finally:
if self.is_main_thread_pool and self.thread_pool_id in GraphEngine.workflow_thread_pool_mapping:
del GraphEngine.workflow_thread_pool_mapping[self.thread_pool_id]
def _release_thread(self):
if self.is_main_thread_pool and self.thread_pool_id in GraphEngine.workflow_thread_pool_mapping:
del GraphEngine.workflow_thread_pool_mapping[self.thread_pool_id]
def _run(
self,