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

fix: 添加对话stream流式结束标记

Browse Source
This commit is contained in:
shaohuzhang1 2024-11-07 15:55:46 +08:00 committed by shaohuzhang1
parent 7bf7df5e27
commit 33d88954c6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -304,11 +304,11 @@ class WorkflowManage:
if chunk is None: if chunk is None:
break break
yield chunk yield chunk
yield self.get_chunk_content('', True)
finally: finally:
self.work_flow_post_handler.handler(self.params['chat_id'], self.params['chat_record_id'], self.work_flow_post_handler.handler(self.params['chat_id'], self.params['chat_record_id'],
self.answer, self.answer,
self) self)
yield self.get_chunk_content('', True)
def run_chain_async(self, current_node): def run_chain_async(self, current_node):
future = executor.submit(self.run_chain, current_node) future = executor.submit(self.run_chain, current_node)