diff --git a/src/chat/chatter_manager.py b/src/chat/chatter_manager.py index be70f4969..7aee10562 100644 --- a/src/chat/chatter_manager.py +++ b/src/chat/chatter_manager.py @@ -122,6 +122,13 @@ class ChatterManager: actions_count = result.get("actions_count", 0) logger.debug(f"流 {stream_id} 处理完成: 成功={success}, 动作数={actions_count}") + # 在处理完成后,清除该流的未读消息 + try: + from src.chat.message_manager.message_manager import message_manager + await message_manager.clear_stream_unread_messages(stream_id) + except Exception as clear_e: + logger.error(f"清除流 {stream_id} 未读消息时发生错误: {clear_e}") + return result except Exception as e: self.stats["failed_executions"] += 1 diff --git a/src/chat/message_manager/message_manager.py b/src/chat/message_manager/message_manager.py index 623250fde..5b715715b 100644 --- a/src/chat/message_manager/message_manager.py +++ b/src/chat/message_manager/message_manager.py @@ -354,6 +354,25 @@ class MessageManager: except Exception as e: logger.error(f"清除未读消息时发生错误: {e}") + async def clear_stream_unread_messages(self, stream_id: str): + """清除指定聊天流的所有未读消息""" + try: + chat_manager = get_chat_manager() + chat_stream = chat_manager.get_stream(stream_id) + if not chat_stream: + logger.warning(f"clear_stream_unread_messages: 聊天流 {stream_id} 不存在") + return + + context = chat_stream.context_manager.context + if hasattr(context, 'unread_messages') and context.unread_messages: + logger.debug(f"正在为流 {stream_id} 清除 {len(context.unread_messages)} 条未读消息") + context.unread_messages.clear() + else: + logger.debug(f"流 {stream_id} 没有需要清除的未读消息") + + except Exception as e: + logger.error(f"清除流 {stream_id} 的未读消息时发生错误: {e}") + # 创建全局消息管理器实例 message_manager = MessageManager()