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

perf: 完善历史消息处理逻辑

Browse Source
This commit is contained in:
RockChinQ
2024-02-01 16:43:44 +08:00
parent 532a713355
commit 6f2d7d96d0
1 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
pkg/pipeline/process/handlers/chat.py
View File

@@ -62,27 +62,27 @@ class ChatMessageHandler(handler.MessageHandler):
query.user_message
)
query.session.using_conversation.messages.append(
query.user_message
)
text_length = 0
start_time = time.time()
async for result in query.use_model.requester.request(query):
query.resp_messages.append(result)
try:
# 消息同步到会话
query.session.using_conversation.messages.append(result)
async for result in query.use_model.requester.request(query):
query.resp_messages.append(result)
if result.content is not None:
text_length += len(result.content)
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
)
finally:
query.session.using_conversation.messages.append(query.user_message)
query.session.using_conversation.messages.extend(query.resp_messages)
print(query.session.using_conversation.messages)
await self.ap.ctr_mgr.usage.post_query_record(
session_type=query.session.launcher_type.value,