|
|
|
|
@ -15,7 +15,6 @@ from core.workflow.nodes.base.entities import BaseNodeData
|
|
|
|
|
from core.workflow.nodes.base.node import BaseNode
|
|
|
|
|
from core.workflow.nodes.enums import ErrorStrategy
|
|
|
|
|
from core.workflow.nodes.event import RunCompletedEvent
|
|
|
|
|
from core.workflow.nodes.event.event import SingleStepRetryEvent
|
|
|
|
|
from core.workflow.nodes.node_mapping import LATEST_VERSION, NODE_TYPE_CLASSES_MAPPING
|
|
|
|
|
from core.workflow.workflow_entry import WorkflowEntry
|
|
|
|
|
from events.app_event import app_draft_workflow_was_synced, app_published_workflow_was_updated
|
|
|
|
|
@ -221,14 +220,8 @@ class WorkflowService:
|
|
|
|
|
|
|
|
|
|
# run draft workflow node
|
|
|
|
|
start_at = time.perf_counter()
|
|
|
|
|
retries = 0
|
|
|
|
|
max_retries = 0
|
|
|
|
|
should_retry = True
|
|
|
|
|
retry_events = []
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
while retries <= max_retries and should_retry:
|
|
|
|
|
retry_start_at = time.perf_counter()
|
|
|
|
|
node_instance, generator = WorkflowEntry.single_step_run(
|
|
|
|
|
workflow=draft_workflow,
|
|
|
|
|
node_id=node_id,
|
|
|
|
|
@ -236,10 +229,6 @@ class WorkflowService:
|
|
|
|
|
user_id=account.id,
|
|
|
|
|
)
|
|
|
|
|
node_instance = cast(BaseNode[BaseNodeData], node_instance)
|
|
|
|
|
max_retries = (
|
|
|
|
|
node_instance.node_data.retry_config.max_retries if node_instance.node_data.retry_config else 0
|
|
|
|
|
)
|
|
|
|
|
retry_interval = node_instance.node_data.retry_config.retry_interval_seconds
|
|
|
|
|
node_run_result: NodeRunResult | None = None
|
|
|
|
|
for event in generator:
|
|
|
|
|
if isinstance(event, RunCompletedEvent):
|
|
|
|
|
@ -252,36 +241,7 @@ class WorkflowService:
|
|
|
|
|
if not node_run_result:
|
|
|
|
|
raise ValueError("Node run failed with no run result")
|
|
|
|
|
# single step debug mode error handling return
|
|
|
|
|
if node_run_result.status == WorkflowNodeExecutionStatus.FAILED:
|
|
|
|
|
if (
|
|
|
|
|
retries == max_retries
|
|
|
|
|
and node_instance.node_type == NodeType.HTTP_REQUEST
|
|
|
|
|
and node_run_result.outputs
|
|
|
|
|
and not node_instance.should_continue_on_error
|
|
|
|
|
):
|
|
|
|
|
node_run_result.status = WorkflowNodeExecutionStatus.SUCCEEDED
|
|
|
|
|
should_retry = False
|
|
|
|
|
else:
|
|
|
|
|
if node_instance.should_retry:
|
|
|
|
|
node_run_result.status = WorkflowNodeExecutionStatus.RETRY
|
|
|
|
|
retries += 1
|
|
|
|
|
node_run_result.retry_index = retries
|
|
|
|
|
retry_events.append(
|
|
|
|
|
SingleStepRetryEvent(
|
|
|
|
|
elapsed_time=time.perf_counter() - retry_start_at,
|
|
|
|
|
inputs=WorkflowEntry.handle_special_values(node_run_result.inputs),
|
|
|
|
|
process_data=WorkflowEntry.handle_special_values(node_run_result.process_data),
|
|
|
|
|
outputs=WorkflowEntry.handle_special_values(node_run_result.outputs),
|
|
|
|
|
metadata=node_run_result.metadata,
|
|
|
|
|
llm_usage=node_run_result.llm_usage,
|
|
|
|
|
error=node_run_result.error,
|
|
|
|
|
retry_index=node_run_result.retry_index,
|
|
|
|
|
)
|
|
|
|
|
)
|
|
|
|
|
time.sleep(retry_interval)
|
|
|
|
|
else:
|
|
|
|
|
should_retry = False
|
|
|
|
|
if node_instance.should_continue_on_error:
|
|
|
|
|
if node_run_result.status == WorkflowNodeExecutionStatus.FAILED and node_instance.should_continue_on_error:
|
|
|
|
|
node_error_args = {
|
|
|
|
|
"status": WorkflowNodeExecutionStatus.EXCEPTION,
|
|
|
|
|
"error": node_run_result.error,
|
|
|
|
|
@ -358,7 +318,6 @@ class WorkflowService:
|
|
|
|
|
|
|
|
|
|
db.session.add(workflow_node_execution)
|
|
|
|
|
db.session.commit()
|
|
|
|
|
workflow_node_execution.retry_events = retry_events
|
|
|
|
|
|
|
|
|
|
return workflow_node_execution
|
|
|
|
|
|
|
|
|
|
|