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

fix:del some print ,and amend respback on stream judge ,and del in dingtalk this is_stream_output_supported() use

Browse Source
This commit is contained in:
Dong_master
2025-07-29 23:09:02 +08:00
parent 8f8c8ff367
commit 307f6acd8c
11 changed files with 101 additions and 188 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/pipeline/process/handlers/chat.py
+6 -29
View File
@@ -63,7 +63,6 @@ class ChatMessageHandler(handler.MessageHandler):
is_stream = await query.adapter.is_stream_output_supported()
except AttributeError:
is_stream = False
print(is_stream)
try:
for r in runner_module.preregistered_runners:
@@ -74,7 +73,7 @@ class ChatMessageHandler(handler.MessageHandler):
raise ValueError(f'未找到请求运行器: {query.pipeline_config["ai"]["runner"]["runner"]}')
if is_stream:
resp_message_id = uuid.uuid4()
if await query.adapter.create_message_card(resp_message_id,query.message_event):
if await query.adapter.create_message_card(resp_message_id, query.message_event):
async for result in runner.run(query):
result.resp_message_id = resp_message_id
if query.resp_messages:
@@ -82,43 +81,21 @@ class ChatMessageHandler(handler.MessageHandler):
if query.resp_message_chain:
query.resp_message_chain.pop()
query.resp_messages.append(result)
self.ap.logger.info(f'对话({query.query_id})流式响应: {self.cut_str(result.readable_str())}')
self.ap.logger.info(
f'对话({query.query_id})流式响应: {self.cut_str(result.readable_str())}'
)
if result.content is not None:
text_length += len(result.content)
yield entities.StageProcessResult(result_type=entities.ResultType.CONTINUE, new_query=query)
else:
yield entities.StageProcessResult(result_type=entities.ResultType.INTERRUPT,
new_query=query)
# for result in results:
#
# query.resp_messages.append(result)
# print(result)
#
# self.ap.logger.info(f'对话({query.query_id})流式响应: {self.cut_str(result.content)}')
#
# if result.content is not None:
# text_length += len(result.content)
#
# # 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]
#
# yield entities.StageProcessResult(result_type=entities.ResultType.CONTINUE, new_query=query)
# query.resp_messages.append(results)
# self.ap.logger.info(f'对话({query.query_id})响应')
# yield entities.StageProcessResult(result_type=entities.ResultType.CONTINUE, new_query=query)
yield entities.StageProcessResult(result_type=entities.ResultType.INTERRUPT, new_query=query)
else:
print("非流式")
async for result in runner.run(query):
query.resp_messages.append(result)
print(result)
self.ap.logger.info(f'对话({query.query_id})响应: {self.cut_str(result.readable_str())}')
@@ -128,7 +105,7 @@ class ChatMessageHandler(handler.MessageHandler):
yield entities.StageProcessResult(result_type=entities.ResultType.CONTINUE, new_query=query)
query.session.using_conversation.messages.append(query.user_message)
query.session.using_conversation.messages.extend(query.resp_messages)
except Exception as e:
self.ap.logger.error(f'对话({query.query_id})请求失败: {type(e).__name__} {str(e)}')