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

fix(chat): 修复消息打断会取消正在进行的回复任务的问题

Browse Source 
之前的消息打断逻辑会无差别地取消处理流中的所有任务。这会导致一个问题：当用户在机器人生成回复期间快速发送新消息时，回复任务会被意外中断，导致机器人无法正常完成回复。

本次修改通过引入 `is_replying` 状态来解决此问题：
1.  在 `StreamContext` 中新增 `is_replying` 状态标志，用于追踪回复生成过程。
2.  当开始生成回复时，设置该标志为 `True`，并在回复完成或取消后通过 `finally` 块确保其恢复为 `False`。
3.  `MessageManager` 的打断检查逻辑现在会首先检查此标志，如果为 `True` 则跳过打断，从而保护正在进行的回复。
4.  `cancel_all_stream_tasks` 也增加了 `exclude_reply` 选项，确保即使触发打断，也不会取消回复任务。
This commit is contained in:
tt-P607
2025-10-11 02:11:32 +08:00
committed by Windpicker-owo
parent fde8aaf514
commit 2e2691c8d5
4 changed files with 31 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
src/chat/message_manager/message_manager.py
View File

@@ -352,11 +352,16 @@ class MessageManager:
if not global_config.chat.interruption_enabled or not chat_stream:
return
# 🌟 修复：获取所有处理任务（包括多重回复）
# 获取所有处理任务
all_processing_tasks = self.chatter_manager.get_all_processing_tasks(chat_stream.stream_id)
if all_processing_tasks:
# 计算打断概率 - 使用新的线性概率模型
# 检查是否有回复任务正在进行
if chat_stream.context_manager.context.is_replying:
logger.debug(f"聊天流 {chat_stream.stream_id} 正在回复，跳过打断检查")
return
# 计算打断概率
interruption_probability = chat_stream.context_manager.context.calculate_interruption_probability(
global_config.chat.interruption_max_limit
)
@@ -364,39 +369,28 @@ class MessageManager:
# 检查是否已达到最大打断次数
if chat_stream.context_manager.context.interruption_count >= global_config.chat.interruption_max_limit:
logger.debug(
f"聊天流 {chat_stream.stream_id} 已达到最大打断次数 {chat_stream.context_manager.context.interruption_count}/{global_config.chat.interruption_max_limit}，跳过打断检查"
f"聊天流 {chat_stream.stream_id} 已达到最大打断次数，跳过打断检查"
)
return
# 根据概率决定是否打断
if random.random() < interruption_probability:
logger.info(f"聊天流 {chat_stream.stream_id} 触发消息打断，打断概率: {interruption_probability:.2f}，检测到 {len(all_processing_tasks)} 个任务")
logger.info(f"聊天流 {chat_stream.stream_id} 触发消息打断 (概率: {interruption_probability:.2f})")
# 🌟 修复：取消所有任务（包括多重回复）
cancelled_count = self.chatter_manager.cancel_all_stream_tasks(chat_stream.stream_id)
# 取消所有非回复任务
cancelled_count = self.chatter_manager.cancel_all_stream_tasks(chat_stream.stream_id, exclude_reply=True)
if cancelled_count > 0:
logger.info(f"消息打断成功取消 {cancelled_count} 个任务: {chat_stream.stream_id}")
else:
logger.warning(f"消息打断未能取消任何任务: {chat_stream.stream_id}")
# 增加打断计数
await chat_stream.context_manager.context.increment_interruption_count()
# 🚀 新增：打断后立即重新进入聊天流程
# 立即重新处理
await self._trigger_immediate_reprocess(chat_stream)
# 检查是否已达到最大次数
if chat_stream.context_manager.context.interruption_count >= global_config.chat.interruption_max_limit:
logger.warning(
f"聊天流 {chat_stream.stream_id} 已达到最大打断次数 {chat_stream.context_manager.context.interruption_count}/{global_config.chat.interruption_max_limit}，后续消息将不再打断"
)
else:
logger.info(
f"聊天流 {chat_stream.stream_id} 已打断并重新进入处理流程，当前打断次数: {chat_stream.context_manager.context.interruption_count}/{global_config.chat.interruption_max_limit}"
)
else:
logger.debug(f"聊天流 {chat_stream.stream_id} 未触发打断，打断概率: {interruption_probability:.2f}，检测到 {len(all_processing_tasks)} 个任务")
logger.debug(f"聊天流 {chat_stream.stream_id} 未触发打断 (概率: {interruption_probability:.2f})")
async def _trigger_immediate_reprocess(self, chat_stream: ChatStream):
"""打断后立即重新进入聊天流程"""