hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-10 03:16:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add WORKFLOW_CALL_MAX_DEPTH env var. (#4713)

Browse Source
This commit is contained in:
majian
2024-05-29 13:39:11 +08:00
committed by GitHub
parent 705a6e3a8e
commit 72a90074bc
3 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
api/core/workflow/workflow_engine_manager.py
View File

@@ -54,8 +54,6 @@ node_classes = {
NodeType.PARAMETER_EXTRACTOR: ParameterExtractorNode
}
WORKFLOW_CALL_MAX_DEPTH = 5
logger = logging.getLogger(__name__)
@@ -128,8 +126,9 @@ class WorkflowEngineManager:
user_inputs=user_inputs
)
if call_depth > WORKFLOW_CALL_MAX_DEPTH:
raise ValueError('Max workflow call depth reached.')
workflow_call_max_depth = current_app.config.get("WORKFLOW_CALL_MAX_DEPTH")
if call_depth > workflow_call_max_depth:
raise ValueError('Max workflow call depth {} reached.'.format(workflow_call_max_depth))
# init workflow run state
workflow_run_state = WorkflowRunState(