jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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
commit 72a90074bc
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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(