From d7984e1192b0653dc7049d7ee2b3bc4e0176a972 Mon Sep 17 00:00:00 2001 From: shaohuzhang1 Date: Thu, 27 Jun 2024 11:35:35 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=20=E5=BD=93=E5=89=8D?= =?UTF-8?q?=E6=97=B6=E9=97=B4=E5=8F=98=E9=87=8F=E8=BE=93=E5=87=BA=E6=97=B6?= =?UTF-8?q?=E5=BB=BA=E8=AE=AE=E8=BE=93=E5=87=BA=E6=AD=A3=E5=B8=B8=E7=9A=84?= =?UTF-8?q?=E6=97=B6=E9=97=B4=E6=A0=BC=E5=BC=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/application/flow/i_step_node.py | 2 +- .../flow/step_node/start_node/impl/base_start_node.py | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/apps/application/flow/i_step_node.py b/apps/application/flow/i_step_node.py index 2b48869e0..0aa620e26 100644 --- a/apps/application/flow/i_step_node.py +++ b/apps/application/flow/i_step_node.py @@ -55,7 +55,7 @@ class WorkFlowPostHandler: details=details, message_tokens=message_tokens, answer_tokens=answer_tokens, - run_time=time.time() - workflow.context['time'], + run_time=time.time() - workflow.context['start_time'], index=0) self.chat_info.append_chat_record(chat_record, self.client_id) # 重新设置缓存 diff --git a/apps/application/flow/step_node/start_node/impl/base_start_node.py b/apps/application/flow/step_node/start_node/impl/base_start_node.py index aedc4f984..7043e42eb 100644 --- a/apps/application/flow/step_node/start_node/impl/base_start_node.py +++ b/apps/application/flow/step_node/start_node/impl/base_start_node.py @@ -7,6 +7,7 @@ @desc: """ import time +from datetime import datetime from application.flow.i_step_node import NodeResult from application.flow.step_node.start_node.i_start_node import IStarNode @@ -17,7 +18,8 @@ class BaseStartStepNode(IStarNode): """ 开始节点 初始化全局变量 """ - return NodeResult({'question': question}, {'time': time.time()}) + return NodeResult({'question': question}, + {'time': datetime.now().strftime('%Y-%m-%d %H:%M:%S'), 'start_time': time.time()}) def get_details(self, index: int, **kwargs): return {