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

feat: 支持节点参数设置直接输出

This commit is contained in:
shaohuzhang1 2024-08-02 13:55:30 +08:00
parent 98cc8cd0e8
commit f2b46225ac
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -6,6 +6,7 @@
@date：2024/6/4 14:30
@desc:
"""
import time
from functools import reduce
from typing import List, Dict
@ -26,6 +27,7 @@ def _write_context(node_variable: Dict, workflow_variable: Dict, node: INode, wo
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']
if workflow.is_result():
workflow.answer += answer

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

@ -6,6 +6,7 @@
@date：2024/6/4 14:30
@desc:
"""
import time
from functools import reduce
from typing import List, Dict
@ -26,6 +27,7 @@ def _write_context(node_variable: Dict, workflow_variable: Dict, node: INode, wo
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']
if workflow.is_result():
workflow.answer += answer