zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 修复节点直接回答,不同节点响应用回车隔开 (#935)

Browse Source
This commit is contained in:
shaohuzhang1 2024-08-06 16:56:05 +08:00 committed by GitHub
parent 864bca6450
commit 8453ea2b4b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
apps/application/flow/i_step_node.py
View File

@ -28,8 +28,9 @@ def write_context(step_variable: Dict, global_variable: Dict, node, workflow):
for key in step_variable: for key in step_variable:
node.context[key] = step_variable[key] node.context[key] = step_variable[key]
if workflow.is_result() and 'answer' in step_variable: if workflow.is_result() and 'answer' in step_variable:
yield step_variable['answer'] answer = step_variable['answer'] + '\n'
workflow.answer += step_variable['answer'] yield answer
workflow.answer += answer
if global_variable is not None: if global_variable is not None:
for key in global_variable: for key in global_variable:
workflow.context[key] = global_variable[key] workflow.context[key] = global_variable[key]

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

@ -45,6 +45,8 @@ def write_context_stream(node_variable: Dict, workflow_variable: Dict, node: INo
for chunk in response: for chunk in response:
answer += chunk.content answer += chunk.content
yield chunk.content yield chunk.content
answer += '\n'
yield '\n'
_write_context(node_variable, workflow_variable, node, workflow, answer) _write_context(node_variable, workflow_variable, node, workflow, answer)

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

@ -45,6 +45,8 @@ def write_context_stream(node_variable: Dict, workflow_variable: Dict, node: INo
for chunk in response: for chunk in response:
answer += chunk.content answer += chunk.content
yield chunk.content yield chunk.content
answer += '\n'
yield '\n'
_write_context(node_variable, workflow_variable, node, workflow, answer) _write_context(node_variable, workflow_variable, node, workflow, answer)