15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 09:54:54 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix: 修复 执行详情中的耗时和回答耗时对不上

This commit is contained in:
shaohuzhang1 2024-06-27 11:42:08 +08:00
parent 848735e4f5
commit d698ee57ac
2 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py
View File

@ -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

3
apps/application/flow/step_node/question_node/impl/base_question_node.py
View File

@ -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