From d698ee57ac87fe0ef930037b6e521066e89221f0 Mon Sep 17 00:00:00 2001 From: shaohuzhang1 Date: Thu, 27 Jun 2024 11:42:08 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=20=E6=89=A7=E8=A1=8C?= =?UTF-8?q?=E8=AF=A6=E6=83=85=E4=B8=AD=E7=9A=84=E8=80=97=E6=97=B6=E5=92=8C?= =?UTF-8?q?=E5=9B=9E=E7=AD=94=E8=80=97=E6=97=B6=E5=AF=B9=E4=B8=8D=E4=B8=8A?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../flow/step_node/ai_chat_step_node/impl/base_chat_node.py | 3 +++ .../flow/step_node/question_node/impl/base_question_node.py | 3 +++ 2 files changed, 6 insertions(+) diff --git a/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py b/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py index 7f8f61bbd..48df8a463 100644 --- a/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py +++ b/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py @@ -7,6 +7,7 @@ @desc: """ import json +import time from functools import reduce from typing import List, Dict @@ -42,6 +43,7 @@ def write_context_stream(node_variable: Dict, workflow_variable: Dict, node: INo node.context['answer'] = answer node.context['history_message'] = node_variable['history_message'] node.context['question'] = node_variable['question'] + node.context['run_time'] = time.time() - node.context['start_time'] def write_context(node_variable: Dict, workflow_variable: Dict, node: INode, workflow): @@ -79,6 +81,7 @@ def get_to_response_write_context(node_variable: Dict, node: INode): node.context['message_tokens'] = message_tokens node.context['answer_tokens'] = answer_tokens node.context['answer'] = answer + node.context['run_time'] = time.time() - node.context['start_time'] return _write_context diff --git a/apps/application/flow/step_node/question_node/impl/base_question_node.py b/apps/application/flow/step_node/question_node/impl/base_question_node.py index e5ed6a660..65fc52c32 100644 --- a/apps/application/flow/step_node/question_node/impl/base_question_node.py +++ b/apps/application/flow/step_node/question_node/impl/base_question_node.py @@ -7,6 +7,7 @@ @desc: """ import json +import time from functools import reduce from typing import List, Dict @@ -42,6 +43,7 @@ def write_context_stream(node_variable: Dict, workflow_variable: Dict, node: INo node.context['answer'] = answer node.context['history_message'] = node_variable['history_message'] node.context['question'] = node_variable['question'] + node.context['run_time'] = time.time() - node.context['start_time'] def write_context(node_variable: Dict, workflow_variable: Dict, node: INode, workflow): @@ -79,6 +81,7 @@ def get_to_response_write_context(node_variable: Dict, node: INode): node.context['message_tokens'] = message_tokens node.context['answer_tokens'] = answer_tokens node.context['answer'] = answer + node.context['run_time'] = time.time() - node.context['start_time'] return _write_context