xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-27 07:54:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

chat中的流式修改

Browse Source
This commit is contained in:
fdc
2025-07-01 18:03:05 +08:00
committed by Junyan Qin
parent b65670cd1a
commit 0d53843230
2 changed files with 17 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/pipeline/process/handlers/chat.py
+8 -13
View File
@@ -71,23 +71,18 @@ class ChatMessageHandler(handler.MessageHandler):
raise ValueError(f'Request runner not found: {query.pipeline_config["ai"]["runner"]["runner"]}')
if is_stream:
accumulated_messages = []
async for result in runner.run(query):
accumulated_messages.append(result)
query.resp_messages.append(result)
async for results in runner.run(query):
async for result in results:
self.ap.logger.info(f'对话({query.query_id})流式响应: {self.cut_str(result.readable_str())}')
query.resp_messages.append(result)
if result.content is not None:
text_length += len(result.content)
self.ap.logger.info(f'对话({query.query_id})流式响应: {self.cut_str(result.readable_str())}')
# current_chain = platform_message.MessageChain([])
# for msg in accumulated_messages:
# if msg.content is not None:
# current_chain.append(platform_message.Plain(msg.content))
# query.resp_message_chain = [current_chain]
if result.content is not None:
text_length += len(result.content)
yield entities.StageProcessResult(result_type=entities.ResultType.CONTINUE, new_query=query)
yield entities.StageProcessResult(result_type=entities.ResultType.CONTINUE, new_query=query)
else:
async for result in runner.run(query):