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

优化代码格式和异常处理

Browse Source 
- 修复异常处理链，使用from语法保留原始异常
- 格式化代码以符合项目规范
- 优化导入模块的顺序

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
春河晴
2025-03-19 20:27:34 +09:00
parent a829dfdb77
commit fdc098d0db
52 changed files with 3156 additions and 2778 deletions
Download Patch File Download Diff File Expand all files Collapse all files

120
src/plugins/moods/moods.py
View File

@@ -8,59 +8,57 @@ from src.common.logger import get_module_logger
logger = get_module_logger("mood_manager")
@dataclass
class MoodState:
valence: float # 愉悦度 (-1 到 1)
arousal: float # 唤醒度 (0 到 1)
text: str # 心情文本描述
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.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.7), # 负愉悦度，高唤醒度
'sad': (-0.6, 0.3), # 负愉悦度，低唤醒度
'surprised': (0.4, 0.8), # 中等愉悦度，高唤醒度
'disgusted': (-0.8, 0.5), # 高负愉悦度，中等唤醒度
'fearful': (-0.7, 0.6), # 负愉悦度，高唤醒度
'neutral': (0.0, 0.5), # 中性愉悦度，中等唤醒度
"happy": (0.8, 0.6), # 高愉悦度，中等唤醒度
"angry": (-0.7, 0.7), # 负愉悦度，高唤醒度
"sad": (-0.6, 0.3), # 负愉悦度，低唤醒度
"surprised": (0.4, 0.8), # 中等愉悦度，高唤醒度
"disgusted": (-0.8, 0.5), # 高负愉悦度，中等唤醒度
"fearful": (-0.7, 0.6), # 负愉悦度，高唤醒度
"neutral": (0.0, 0.5), # 中性愉悦度，中等唤醒度
}
# 情绪文本映射表
self.mood_text_map = {
# 第一象限：高唤醒，正愉悦
@@ -78,12 +76,11 @@ class MoodManager:
# 第四象限：低唤醒，正愉悦
(0.2, 0.45): "平静",
(0.3, 0.4): "安宁",
(0.5, 0.3): "放松"
(0.5, 0.3): "放松",
}
@classmethod
def get_instance(cls) -> 'MoodManager':
def get_instance(cls) -> "MoodManager":
"""获取MoodManager的单例实例"""
if cls._instance is None:
cls._instance = MoodManager()
@@ -96,12 +93,10 @@ class MoodManager:
"""
if self._running:
return
self._running = True
self._update_thread = threading.Thread(
target=self._continuous_mood_update,
args=(update_interval,),
daemon=True
target=self._continuous_mood_update, args=(update_interval,), daemon=True
)
self._update_thread.start()
@@ -125,31 +120,35 @@ class MoodManager:
"""应用情绪衰减"""
current_time = time.time()
time_diff = current_time - self.last_update
# Valence 向中性（0）回归
valence_target = 0.0
self.current_mood.valence = valence_target + (self.current_mood.valence - valence_target) * math.exp(-self.decay_rate_valence * time_diff)
self.current_mood.valence = valence_target + (self.current_mood.valence - valence_target) * math.exp(
-self.decay_rate_valence * time_diff
)
# Arousal 向中性（0.5）回归
arousal_target = 0.5
self.current_mood.arousal = arousal_target + (self.current_mood.arousal - arousal_target) * math.exp(-self.decay_rate_arousal * time_diff)
self.current_mood.arousal = arousal_target + (self.current_mood.arousal - arousal_target) * math.exp(
-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:
@@ -159,51 +158,48 @@ class MoodManager:
def _update_mood_text(self) -> None:
"""根据当前情绪状态更新文本描述"""
closest_mood = None
min_distance = float('inf')
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
)
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:
@@ -212,9 +208,11 @@ class MoodManager:
def print_mood_status(self) -> None:
"""打印当前情绪状态"""
logger.info(f"[情绪状态]愉悦度: {self.current_mood.valence:.2f}, "
f"唤醒度: {self.current_mood.arousal:.2f}, "
f"心情: {self.current_mood.text}")
logger.info(
f"[情绪状态]愉悦度: {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:
"""
@@ -224,19 +222,19 @@ class MoodManager:
"""
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()