v0.5.11 添加了情绪管理器

src/plugins/moods/moods.py

@@ -0,0 +1,230 @@
+import math
+import time
+import threading
+from typing import Dict, Tuple, Optional
+from dataclasses import dataclass
+from ..chat.config import global_config
+
+@dataclass
+class MoodState:
+    valence: float  # 愉悦度 (-1 到 1)
+    arousal: float  # 唤醒度 (0 到 1)
+    text: str       # 心情文本描述
+
+class MoodManager:
+    _instance = None
+    _lock = threading.Lock()
+    
+    def __new__(cls):
+        with cls._lock:
+            if cls._instance is None:
+                cls._instance = super().__new__(cls)
+                cls._instance._initialized = False
+            return cls._instance
+    
+    def __init__(self):
+        # 确保初始化代码只运行一次
+        if self._initialized:
+            return
+            
+        self._initialized = True
+        
+        # 初始化心情状态
+        self.current_mood = MoodState(
+            valence=0.0,
+            arousal=0.5,
+            text="平静"
+        )
+        
+        # 从配置文件获取衰减率
+        self.decay_rate_valence = 1 - global_config.mood_decay_rate  # 愉悦度衰减率
+        self.decay_rate_arousal = 1 - global_config.mood_decay_rate  # 唤醒度衰减率
+        
+        # 上次更新时间
+        self.last_update = time.time()
+        
+        # 线程控制
+        self._running = False
+        self._update_thread = None
+        
+        # 情绪词映射表 (valence, arousal)
+        self.emotion_map = {
+            'happy': (0.8, 0.6),      # 高愉悦度，中等唤醒度
+            'angry': (-0.7, 0.8),     # 负愉悦度，高唤醒度
+            'sad': (-0.6, 0.3),       # 负愉悦度，低唤醒度
+            'surprised': (0.4, 0.9),  # 中等愉悦度，高唤醒度
+            'disgusted': (-0.8, 0.5), # 高负愉悦度，中等唤醒度
+            'fearful': (-0.7, 0.7),   # 负愉悦度，高唤醒度
+            'neutral': (0.0, 0.5),    # 中性愉悦度，中等唤醒度
+        }
+        
+        # 情绪文本映射表
+        self.mood_text_map = {
+            # 第一象限：高唤醒，正愉悦
+            (0.5, 0.7): "兴奋",
+            (0.3, 0.8): "快乐",
+            # 第二象限：高唤醒，负愉悦
+            (-0.5, 0.7): "愤怒",
+            (-0.3, 0.8): "焦虑",
+            # 第三象限：低唤醒，负愉悦
+            (-0.5, 0.3): "悲伤",
+            (-0.3, 0.2): "疲倦",
+            # 第四象限：低唤醒，正愉悦
+            (0.5, 0.3): "放松",
+            (0.3, 0.2): "平静"
+        }
+
+    @classmethod
+    def get_instance(cls) -> 'MoodManager':
+        """获取MoodManager的单例实例"""
+        if cls._instance is None:
+            cls._instance = MoodManager()
+        return cls._instance
+
+    def start_mood_update(self, update_interval: float = 1.0) -> None:
+        """
+        启动情绪更新线程
+        :param update_interval: 更新间隔（秒）
+        """
+        if self._running:
+            return
+            
+        self._running = True
+        self._update_thread = threading.Thread(
+            target=self._continuous_mood_update,
+            args=(update_interval,),
+            daemon=True
+        )
+        self._update_thread.start()
+
+    def stop_mood_update(self) -> None:
+        """停止情绪更新线程"""
+        self._running = False
+        if self._update_thread and self._update_thread.is_alive():
+            self._update_thread.join()
+
+    def _continuous_mood_update(self, update_interval: float) -> None:
+        """
+        持续更新情绪状态的线程函数
+        :param update_interval: 更新间隔（秒）
+        """
+        while self._running:
+            self._apply_decay()
+            self._update_mood_text()
+            time.sleep(update_interval)
+
+    def _apply_decay(self) -> None:
+        """应用情绪衰减"""
+        current_time = time.time()
+        time_diff = current_time - self.last_update
+        
+        # 应用衰减公式
+        self.current_mood.valence *= math.pow(1 - self.decay_rate_valence, time_diff)
+        self.current_mood.arousal *= math.pow(1 - self.decay_rate_arousal, time_diff)
+        
+        # 确保值在合理范围内
+        self.current_mood.valence = max(-1.0, min(1.0, self.current_mood.valence))
+        self.current_mood.arousal = max(0.0, min(1.0, self.current_mood.arousal))
+        
+        self.last_update = current_time
+
+    def update_mood_from_text(self, text: str, valence_change: float, arousal_change: float) -> None:
+        """根据输入文本更新情绪状态"""
+        
+        self.current_mood.valence += valence_change
+        self.current_mood.arousal += arousal_change
+        
+        # 限制范围
+        self.current_mood.valence = max(-1.0, min(1.0, self.current_mood.valence))
+        self.current_mood.arousal = max(0.0, min(1.0, self.current_mood.arousal))
+        
+        self._update_mood_text()
+
+    def set_mood_text(self, text: str) -> None:
+        """直接设置心情文本"""
+        self.current_mood.text = text
+
+    def _update_mood_text(self) -> None:
+        """根据当前情绪状态更新文本描述"""
+        closest_mood = None
+        min_distance = float('inf')
+        
+        for (v, a), text in self.mood_text_map.items():
+            distance = math.sqrt(
+                (self.current_mood.valence - v) ** 2 + 
+                (self.current_mood.arousal - a) ** 2
+            )
+            if distance < min_distance:
+                min_distance = distance
+                closest_mood = text
+        
+        if closest_mood:
+            self.current_mood.text = closest_mood
+
+    def update_mood_by_user(self, user_id: str, valence_change: float, arousal_change: float) -> None:
+        """根据用户ID更新情绪状态"""
+        
+        # 这里可以根据用户ID添加特定的权重或规则
+        weight = 1.0  # 默认权重
+        
+        self.current_mood.valence += valence_change * weight
+        self.current_mood.arousal += arousal_change * weight
+        
+        # 限制范围
+        self.current_mood.valence = max(-1.0, min(1.0, self.current_mood.valence))
+        self.current_mood.arousal = max(0.0, min(1.0, self.current_mood.arousal))
+        
+        self._update_mood_text()
+
+    def get_prompt(self) -> str:
+        """根据当前情绪状态生成提示词"""
+        
+        base_prompt = f"当前心情：{self.current_mood.text}。"
+        
+        # 根据情绪状态添加额外的提示信息
+        if self.current_mood.valence > 0.5:
+            base_prompt += "你现在心情很好，"
+        elif self.current_mood.valence < -0.5:
+            base_prompt += "你现在心情不太好，"
+            
+        if self.current_mood.arousal > 0.7:
+            base_prompt += "情绪比较激动。"
+        elif self.current_mood.arousal < 0.3:
+            base_prompt += "情绪比较平静。"
+            
+        return base_prompt
+
+    def get_current_mood(self) -> MoodState:
+        """获取当前情绪状态"""
+        return self.current_mood
+
+    def print_mood_status(self) -> None:
+        """打印当前情绪状态"""
+        print(f"\033[1;35m[情绪状态]\033[0m 愉悦度: {self.current_mood.valence:.2f}, " 
+              f"唤醒度: {self.current_mood.arousal:.2f}, "
+              f"心情: {self.current_mood.text}")
+
+    def update_mood_from_emotion(self, emotion: str, intensity: float = 1.0) -> None:
+        """
+        根据情绪词更新心情状态
+        :param emotion: 情绪词（如'happy', 'sad'等）
+        :param intensity: 情绪强度（0.0-1.0）
+        """
+        if emotion not in self.emotion_map:
+            return
+            
+        valence_change, arousal_change = self.emotion_map[emotion]
+        
+        # 应用情绪强度
+        valence_change *= intensity
+        arousal_change *= intensity
+        
+        # 更新当前情绪状态
+        self.current_mood.valence += valence_change
+        self.current_mood.arousal += arousal_change
+        
+        # 限制范围
+        self.current_mood.valence = max(-1.0, min(1.0, self.current_mood.valence))
+        self.current_mood.arousal = max(0.0, min(1.0, self.current_mood.arousal))
+        
+        self._update_mood_text()