feat(sleep): 引入休眠阶段消息唤醒机制与愤怒情绪模式

- 扩展日程休眠逻辑，支持在睡眠时段通过@/私聊唤醒 - 新增 WakeUpManager 持续追踪唤醒阈值并触发愤怒状态 - MoodManager 中追加“被吵醒”情绪标识与愤怒提示模板 - HeartFChat 主循环集成唤醒处理管线，兼容原逻辑 - 配置中心开放阈值、衰减、持续时间等可调参数此举允许机器人即便在“睡觉”时也能被重要消息强行“闹醒”，并以生气口吻做出回应，提升拟人化交互体验。
2025-08-21 16:19:34 +08:00
parent 747816b6e3
commit 2c98f64cdd
9 changed files with 609 additions and 11 deletions
--- a/src/chat/chat_loop/wakeup_manager.py
+++ b/src/chat/chat_loop/wakeup_manager.py
@@ -0,0 +1,172 @@
+import asyncio
+import time
+from typing import Optional
+from src.common.logger import get_logger
+from src.config.config import global_config
+from .hfc_context import HfcContext
+
+logger = get_logger("wakeup")
+
+class WakeUpManager:
+    def __init__(self, context: HfcContext):
+        """
+        初始化唤醒度管理器
+        
+        Args:
+            context: HFC聊天上下文对象
+            
+        功能说明:
+        - 管理休眠状态下的唤醒度累积
+        - 处理唤醒度的自然衰减
+        - 控制愤怒状态的持续时间
+        """
+        self.context = context
+        self.wakeup_value = 0.0  # 当前唤醒度
+        self.is_angry = False  # 是否处于愤怒状态
+        self.angry_start_time = 0.0  # 愤怒状态开始时间
+        self.last_decay_time = time.time()  # 上次衰减时间
+        self._decay_task: Optional[asyncio.Task] = None
+        
+        # 从配置文件获取参数
+        wakeup_config = global_config.wakeup_system
+        self.wakeup_threshold = wakeup_config.wakeup_threshold
+        self.private_message_increment = wakeup_config.private_message_increment
+        self.group_mention_increment = wakeup_config.group_mention_increment
+        self.decay_rate = wakeup_config.decay_rate
+        self.decay_interval = wakeup_config.decay_interval
+        self.angry_duration = wakeup_config.angry_duration
+        self.enabled = wakeup_config.enable
+
+    async def start(self):
+        """启动唤醒度管理器"""
+        if not self.enabled:
+            logger.info(f"{self.context.log_prefix} 唤醒度系统已禁用，跳过启动")
+            return
+            
+        if not self._decay_task:
+            self._decay_task = asyncio.create_task(self._decay_loop())
+            self._decay_task.add_done_callback(self._handle_decay_completion)
+            logger.info(f"{self.context.log_prefix} 唤醒度管理器已启动")
+
+    async def stop(self):
+        """停止唤醒度管理器"""
+        if self._decay_task and not self._decay_task.done():
+            self._decay_task.cancel()
+            await asyncio.sleep(0)
+            logger.info(f"{self.context.log_prefix} 唤醒度管理器已停止")
+
+    def _handle_decay_completion(self, task: asyncio.Task):
+        """处理衰减任务完成"""
+        try:
+            if exception := task.exception():
+                logger.error(f"{self.context.log_prefix} 唤醒度衰减任务异常: {exception}")
+            else:
+                logger.info(f"{self.context.log_prefix} 唤醒度衰减任务正常结束")
+        except asyncio.CancelledError:
+            logger.info(f"{self.context.log_prefix} 唤醒度衰减任务被取消")
+
+    async def _decay_loop(self):
+        """唤醒度衰减循环"""
+        while self.context.running:
+            await asyncio.sleep(self.decay_interval)
+            
+            current_time = time.time()
+            
+            # 检查愤怒状态是否过期
+            if self.is_angry and current_time - self.angry_start_time >= self.angry_duration:
+                self.is_angry = False
+                # 通知情绪管理系统清除愤怒状态
+                from src.mood.mood_manager import mood_manager
+                mood_manager.clear_angry_from_wakeup(self.context.stream_id)
+                logger.info(f"{self.context.log_prefix} 愤怒状态结束，恢复正常")
+            
+            # 唤醒度自然衰减
+            if self.wakeup_value > 0:
+                old_value = self.wakeup_value
+                self.wakeup_value = max(0, self.wakeup_value - self.decay_rate)
+                if old_value != self.wakeup_value:
+                    logger.debug(f"{self.context.log_prefix} 唤醒度衰减: {old_value:.1f} -> {self.wakeup_value:.1f}")
+
+    def add_wakeup_value(self, is_private_chat: bool, is_mentioned: bool = False) -> bool:
+        """
+        增加唤醒度值
+        
+        Args:
+            is_private_chat: 是否为私聊
+            is_mentioned: 是否被艾特(仅群聊有效)
+            
+        Returns:
+            bool: 是否达到唤醒阈值
+        """
+        # 如果系统未启用，直接返回
+        if not self.enabled:
+            return False
+            
+        from src.manager.schedule_manager import schedule_manager
+        
+        # 只有在休眠状态下才累积唤醒度
+        if not schedule_manager.is_sleeping():
+            return False
+            
+        old_value = self.wakeup_value
+        
+        if is_private_chat:
+            # 私聊每条消息都增加唤醒度
+            self.wakeup_value += self.private_message_increment
+            logger.debug(f"{self.context.log_prefix} 私聊消息增加唤醒度: +{self.private_message_increment}")
+        elif is_mentioned:
+            # 群聊只有被艾特才增加唤醒度
+            self.wakeup_value += self.group_mention_increment
+            logger.debug(f"{self.context.log_prefix} 群聊艾特增加唤醒度: +{self.group_mention_increment}")
+        else:
+            # 群聊未被艾特，不增加唤醒度
+            return False
+        
+        logger.info(f"{self.context.log_prefix} 唤醒度变化: {old_value:.1f} -> {self.wakeup_value:.1f} (阈值: {self.wakeup_threshold})")
+        
+        # 检查是否达到唤醒阈值
+        if self.wakeup_value >= self.wakeup_threshold:
+            self._trigger_wakeup()
+            return True
+            
+        return False
+
+    def _trigger_wakeup(self):
+        """触发唤醒，进入愤怒状态"""
+        self.is_angry = True
+        self.angry_start_time = time.time()
+        self.wakeup_value = 0.0  # 重置唤醒度
+        
+        # 通知情绪管理系统进入愤怒状态
+        from src.mood.mood_manager import mood_manager
+        mood_manager.set_angry_from_wakeup(self.context.stream_id)
+        
+        logger.info(f"{self.context.log_prefix} 唤醒度达到阈值({self.wakeup_threshold})，被吵醒进入愤怒状态！")
+
+    def get_angry_prompt_addition(self) -> str:
+        """获取愤怒状态下的提示词补充"""
+        if self.is_angry:
+            return "你被人吵醒了非常生气，说话带着怒气"
+        return ""
+
+    def is_in_angry_state(self) -> bool:
+        """检查是否处于愤怒状态"""
+        if self.is_angry:
+            current_time = time.time()
+            if current_time - self.angry_start_time >= self.angry_duration:
+                self.is_angry = False
+                # 通知情绪管理系统清除愤怒状态
+                from src.mood.mood_manager import mood_manager
+                mood_manager.clear_angry_from_wakeup(self.context.stream_id)
+                logger.info(f"{self.context.log_prefix} 愤怒状态自动过期")
+                return False
+        return self.is_angry
+
+    def get_status_info(self) -> dict:
+        """获取当前状态信息"""
+        return {
+            "wakeup_value": self.wakeup_value,
+            "wakeup_threshold": self.wakeup_threshold,
+            "is_angry": self.is_angry,
+            "angry_remaining_time": max(0, self.angry_duration - (time.time() - self.angry_start_time)) if self.is_angry else 0
+        }