From ff26835c93d4df0813db96f29110ca872bcb97e1 Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com>
Date: Mon, 8 Dec 2025 15:48:48 +0800
Subject: [PATCH] fix: Error in total time consumption statistics for executing
 records (#4459)

---
 apps/application/flow/i_step_node.py               | 5 ++++-
 apps/application/flow/knowledge_workflow_manage.py | 2 ++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/apps/application/flow/i_step_node.py b/apps/application/flow/i_step_node.py
index e963b74ae..40b3b04d0 100644
--- a/apps/application/flow/i_step_node.py
+++ b/apps/application/flow/i_step_node.py
@@ -105,8 +105,11 @@ class KnowledgeWorkflowPostHandler(WorkFlowPostHandler):
         self.knowledge_action_id = knowledge_action_id
 
     def handler(self, workflow):
+        err = [True for key, value in workflow.get_runtime_details().items() if value.get('status') == 500]
         QuerySet(KnowledgeAction).filter(id=self.knowledge_action_id).update(
-            state=State.SUCCESS)
+            state=State.FAILURE if any(err) else State.SUCCESS,
+            run_time=time.time() - workflow.context.get('start_time') if workflow.context.get(
+                'start_time') is not None else 0)
 
 
 class NodeResult:
diff --git a/apps/application/flow/knowledge_workflow_manage.py b/apps/application/flow/knowledge_workflow_manage.py
index 07a184df6..6fb25c1f1 100644
--- a/apps/application/flow/knowledge_workflow_manage.py
+++ b/apps/application/flow/knowledge_workflow_manage.py
@@ -6,6 +6,7 @@
     @date：2025/11/13 19:02
     @desc:
 """
+import time
 import traceback
 from concurrent.futures import ThreadPoolExecutor
 
@@ -43,6 +44,7 @@ class KnowledgeWorkflowManage(WorkflowManage):
         return start_node_list[0]
 
     def run(self):
+        self.context['start_time'] = time.time()
         executor.submit(self._run)
 
     def _run(self):