gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(chat): 优化异步任务处理和消息管理逻辑

Browse Source 
- 使用asyncio.create_task替代await调用，提升并发性能
- 简化流管理器的槽位获取逻辑，移除回退方案
- 重构上下文管理器的消息添加和更新机制
- 移除StreamContext中的冗余方法，保持数据模型的简洁性
- 优化兴趣度评分系统的更新流程，减少阻塞操作

这些改动主要关注性能优化和代码结构简化，不涉及功能变更。
This commit is contained in:
Windpicker-owo
2025-10-05 15:17:30 +08:00
parent 5edf50705c
commit 23e523c7b9
9 changed files with 92 additions and 255 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/chat/planner_actions/action_manager.py
View File

@@ -185,7 +185,7 @@ class ChatterActionManager:
logger.info(f"{log_prefix} 选择不回复，原因: {reason}")
# 存储no_reply信息到数据库
await database_api.store_action_info(
asyncio.create_task(database_api.store_action_info(
chat_stream=chat_stream,
action_build_into_prompt=False,
action_prompt_display=reason,
@@ -193,10 +193,10 @@ class ChatterActionManager:
thinking_id=thinking_id,
action_data={"reason": reason},
action_name="no_reply",
)
))
# 自动清空所有未读消息
await self._clear_all_unread_messages(chat_stream.stream_id, "no_reply")
asyncio.create_task(self._clear_all_unread_messages(chat_stream.stream_id, "no_reply"))
return {"action_type": "no_reply", "success": True, "reply_text": "", "command": ""}
@@ -214,12 +214,12 @@ class ChatterActionManager:
# 记录执行的动作到目标消息
if success:
await self._record_action_to_message(chat_stream, action_name, target_message, action_data)
asyncio.create_task(self._record_action_to_message(chat_stream, action_name, target_message, action_data))
# 自动清空所有未读消息
if clear_unread_messages:
await self._clear_all_unread_messages(chat_stream.stream_id, action_name)
asyncio.create_task(self._clear_all_unread_messages(chat_stream.stream_id, action_name))
# 重置打断计数
await self._reset_interruption_count_after_action(chat_stream.stream_id)
asyncio.create_task(self._reset_interruption_count_after_action(chat_stream.stream_id))
return {
"action_type": action_name,
@@ -260,13 +260,13 @@ class ChatterActionManager:
)
# 记录回复动作到目标消息
await self._record_action_to_message(chat_stream, "reply", target_message, action_data)
asyncio.create_task(self._record_action_to_message(chat_stream, "reply", target_message, action_data))
if clear_unread_messages:
await self._clear_all_unread_messages(chat_stream.stream_id, "reply")
asyncio.create_task(self._clear_all_unread_messages(chat_stream.stream_id, "reply"))
# 回复成功，重置打断计数
await self._reset_interruption_count_after_action(chat_stream.stream_id)
asyncio.create_task(self._reset_interruption_count_after_action(chat_stream.stream_id))
return {"action_type": "reply", "success": True, "reply_text": reply_text, "loop_info": loop_info}