youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-06 02:18:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: code

Browse Source
This commit is contained in:
Yeuoly
2024-03-17 21:08:25 +08:00
parent 73c2b35dfe
commit d8ab611480
9 changed files with 24 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/app/apps/advanced_chat/workflow_event_trigger_callback.py
View File

@ -97,6 +97,7 @@ class WorkflowEventTriggerCallback(BaseWorkflowCallback):
node_data: BaseNodeData,
error: str,
inputs: Optional[dict] = None,
outputs: Optional[dict] = None,
process_data: Optional[dict] = None) -> None:
"""
Workflow node execute failed
@ -107,6 +108,7 @@ class WorkflowEventTriggerCallback(BaseWorkflowCallback):
node_type=node_type,
node_data=node_data,
inputs=inputs,
outputs=outputs,
process_data=process_data,
error=error
),

2
api/core/app/apps/workflow/workflow_event_trigger_callback.py
View File

@ -96,6 +96,7 @@ class WorkflowEventTriggerCallback(BaseWorkflowCallback):
node_data: BaseNodeData,
error: str,
inputs: Optional[dict] = None,
outputs: Optional[dict] = None,
process_data: Optional[dict] = None) -> None:
"""
Workflow node execute failed
@ -106,6 +107,7 @@ class WorkflowEventTriggerCallback(BaseWorkflowCallback):
node_type=node_type,
node_data=node_data,
inputs=inputs,
outputs=outputs,
process_data=process_data,
error=error
),

1
api/core/app/entities/queue_entities.py
View File

@ -168,6 +168,7 @@ class QueueNodeFailedEvent(AppQueueEvent):
node_data: BaseNodeData
inputs: Optional[dict] = None
outputs: Optional[dict] = None
process_data: Optional[dict] = None
error: str

14
api/core/app/task_pipeline/workflow_cycle_manage.py
View File

@ -218,7 +218,11 @@ class WorkflowCycleManage:
def _workflow_node_execution_failed(self, workflow_node_execution: WorkflowNodeExecution,
start_at: float,
error: str) -> WorkflowNodeExecution:
error: str,
inputs: Optional[dict] = None,
process_data: Optional[dict] = None,
outputs: Optional[dict] = None,
) -> WorkflowNodeExecution:
"""
Workflow node execution failed
:param workflow_node_execution: workflow node execution
@ -230,6 +234,9 @@ class WorkflowCycleManage:
workflow_node_execution.error = error
workflow_node_execution.elapsed_time = time.perf_counter() - start_at
workflow_node_execution.finished_at = datetime.utcnow()
workflow_node_execution.inputs = json.dumps(inputs) if inputs else None
workflow_node_execution.process_data = json.dumps(process_data) if process_data else None
workflow_node_execution.outputs = json.dumps(outputs) if outputs else None
db.session.commit()
db.session.refresh(workflow_node_execution)
@ -402,7 +409,10 @@ class WorkflowCycleManage:
workflow_node_execution = self._workflow_node_execution_failed(
workflow_node_execution=workflow_node_execution,
start_at=current_node_execution.start_at,
error=event.error
error=event.error,
inputs=event.inputs,
process_data=event.process_data,
outputs=event.outputs
)
db.session.close()