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

fix: 修复ai对话节点输出内容回添加回车

Browse Source
This commit is contained in:
shaohuzhang1 2024-08-27 11:32:27 +08:00 committed by shaohuzhang1
parent cb97de66a8
commit b19ed0d8db
4 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -29,7 +29,7 @@ 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:
answer = step_variable['answer'] + '\n' answer = step_variable['answer']
yield answer yield answer
workflow.answer += answer workflow.answer += answer
if global_variable is not None: if global_variable is not None:

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

@ -45,8 +45,6 @@ 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,8 +45,6 @@ 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)

8
apps/application/flow/workflow_manage.py
View File

@ -209,9 +209,13 @@ class WorkflowManage:
self.current_result = self.current_node.run() self.current_result = self.current_node.run()
result = self.current_result.write_context(self.current_node, self) result = self.current_result.write_context(self.current_node, self)
if result is not None: if result is not None:
for r in result: if self.is_result():
if self.is_result(): for r in result:
yield self.get_chunk_content(r) yield self.get_chunk_content(r)
yield self.get_chunk_content('\n')
self.answer += '\n'
else:
list(result)
if not self.has_next_node(self.current_result): if not self.has_next_node(self.current_result):
yield self.get_chunk_content('', True) yield self.get_chunk_content('', True)
break break