import math import random import time from src.common.logger import get_logger from src.config.config import global_config, model_config from src.chat.message_receive.message import MessageRecv from src.common.data_models.database_data_model import DatabaseMessages from src.chat.message_receive.chat_stream import get_chat_manager from src.chat.utils.prompt import Prompt, global_prompt_manager from src.chat.utils.chat_message_builder import build_readable_messages, get_raw_msg_by_timestamp_with_chat_inclusive from src.llm_models.utils_model import LLMRequest from src.manager.async_task_manager import AsyncTask, async_task_manager logger = get_logger("mood") def init_prompt(): Prompt( """ {chat_talking_prompt} 以上是群里正在进行的聊天记录 {identity_block} 你刚刚的情绪状态是：{mood_state} 现在，发送了消息，引起了你的注意，你对其进行了阅读和思考，请你输出一句话描述你新的情绪状态 请只输出情绪状态，不要输出其他内容： """, "change_mood_prompt", ) Prompt( """ {chat_talking_prompt} 以上是群里最近的聊天记录 {identity_block} 你之前的情绪状态是：{mood_state} 距离你上次关注群里消息已经过去了一段时间，你冷静了下来，请你输出一句话描述你现在的情绪状态 请只输出情绪状态，不要输出其他内容： """, "regress_mood_prompt", ) class ChatMood: def __init__(self, chat_id: str): self.chat_id: str = chat_id chat_manager = get_chat_manager() self.chat_stream = chat_manager.get_stream(self.chat_id) if not self.chat_stream: raise ValueError(f"Chat stream for chat_id {chat_id} not found") self.log_prefix = f"[{self.chat_stream.group_info.group_name if self.chat_stream.group_info else self.chat_stream.user_info.user_nickname}]" self.mood_state: str = "感觉很平静" self.is_angry_from_wakeup: bool = False # 是否因被吵醒而愤怒 self.regression_count: int = 0 self.mood_model = LLMRequest(model_set=model_config.model_task_config.emotion, request_type="mood") self.last_change_time: float = 0 async def update_mood_by_message(self, message: MessageRecv | DatabaseMessages, interested_rate: float): # 如果当前聊天处于失眠状态，则锁定情绪，不允许更新 if self.chat_id in mood_manager.insomnia_chats: logger.debug(f"{self.log_prefix} 处于失眠状态，情绪已锁定，跳过更新。") return self.regression_count = 0 # 处理不同类型的消息对象 if isinstance(message, MessageRecv): message_time = message.message_info.time else: # DatabaseMessages message_time = message.time during_last_time = message_time - self.last_change_time base_probability = 0.05 time_multiplier = 4 * (1 - math.exp(-0.01 * during_last_time)) if interested_rate <= 0: interest_multiplier = 0 else: interest_multiplier = 2 * math.pow(interested_rate, 0.25) logger.debug( f"base_probability: {base_probability}, time_multiplier: {time_multiplier}, interest_multiplier: {interest_multiplier}" ) update_probability = global_config.mood.mood_update_threshold * min( 1.0, base_probability * time_multiplier * interest_multiplier ) if random.random() > update_probability: return logger.debug( f"{self.log_prefix} 更新情绪状态，感兴趣度: {interested_rate:.2f}, 更新概率: {update_probability:.2f}" ) message_list_before_now = get_raw_msg_by_timestamp_with_chat_inclusive( chat_id=self.chat_id, timestamp_start=self.last_change_time, timestamp_end=message_time, limit=int(global_config.chat.max_context_size / 3), limit_mode="last", ) chat_talking_prompt = build_readable_messages( message_list_before_now, replace_bot_name=True, merge_messages=False, timestamp_mode="normal_no_YMD", read_mark=0.0, truncate=True, show_actions=True, ) bot_name = global_config.bot.nickname if global_config.bot.alias_names: bot_nickname = f",也有人叫你{','.join(global_config.bot.alias_names)}" else: bot_nickname = "" prompt_personality = global_config.personality.personality_core identity_block = f"你的名字是{bot_name}{bot_nickname}，你{prompt_personality}：" prompt = await global_prompt_manager.format_prompt( "change_mood_prompt", chat_talking_prompt=chat_talking_prompt, identity_block=identity_block, mood_state=self.mood_state, ) response, (reasoning_content, _, _) = await self.mood_model.generate_response_async( prompt=prompt, temperature=0.7 ) if global_config.debug.show_prompt: logger.info(f"{self.log_prefix} prompt: {prompt}") logger.info(f"{self.log_prefix} response: {response}") logger.info(f"{self.log_prefix} reasoning_content: {reasoning_content}") logger.info(f"{self.log_prefix} 情绪状态更新为: {response}") self.mood_state = response self.last_change_time = message_time async def regress_mood(self): message_time = time.time() message_list_before_now = get_raw_msg_by_timestamp_with_chat_inclusive( chat_id=self.chat_id, timestamp_start=self.last_change_time, timestamp_end=message_time, limit=15, limit_mode="last", ) chat_talking_prompt = build_readable_messages( message_list_before_now, replace_bot_name=True, merge_messages=False, timestamp_mode="normal_no_YMD", read_mark=0.0, truncate=True, show_actions=True, ) bot_name = global_config.bot.nickname if global_config.bot.alias_names: bot_nickname = f",也有人叫你{','.join(global_config.bot.alias_names)}" else: bot_nickname = "" prompt_personality = global_config.personality.personality_core identity_block = f"你的名字是{bot_name}{bot_nickname}，你{prompt_personality}：" prompt = await global_prompt_manager.format_prompt( "regress_mood_prompt", chat_talking_prompt=chat_talking_prompt, identity_block=identity_block, mood_state=self.mood_state, ) response, (reasoning_content, _, _) = await self.mood_model.generate_response_async( prompt=prompt, temperature=0.7 ) if global_config.debug.show_prompt: logger.info(f"{self.log_prefix} prompt: {prompt}") logger.info(f"{self.log_prefix} response: {response}") logger.info(f"{self.log_prefix} reasoning_content: {reasoning_content}") logger.info(f"{self.log_prefix} 情绪状态转变为: {response}") self.mood_state = response self.regression_count += 1 class MoodRegressionTask(AsyncTask): def __init__(self, mood_manager: "MoodManager"): super().__init__(task_name="MoodRegressionTask", run_interval=30) self.mood_manager = mood_manager async def run(self): logger.debug("开始情绪回归任务...") now = time.time() for mood in self.mood_manager.mood_list: if mood.last_change_time == 0: continue if now - mood.last_change_time > 180: if mood.regression_count >= 3: continue logger.debug(f"{mood.log_prefix} 开始情绪回归, 第 {mood.regression_count + 1} 次") await mood.regress_mood() class MoodManager: def __init__(self): self.mood_list: list[ChatMood] = [] """当前情绪状态""" self.task_started: bool = False self.insomnia_chats: set[str] = set() # 正在失眠的聊天ID列表 async def start(self): """启动情绪回归后台任务""" if self.task_started: return logger.info("启动情绪回归任务...") task = MoodRegressionTask(self) await async_task_manager.add_task(task) self.task_started = True logger.info("情绪回归任务已启动") def get_mood_by_chat_id(self, chat_id: str) -> ChatMood: for mood in self.mood_list: if mood.chat_id == chat_id: return mood new_mood = ChatMood(chat_id) self.mood_list.append(new_mood) return new_mood def reset_mood_by_chat_id(self, chat_id: str): for mood in self.mood_list: if mood.chat_id == chat_id: mood.mood_state = "感觉很平静" mood.regression_count = 0 mood.is_angry_from_wakeup = False return self.mood_list.append(ChatMood(chat_id)) def set_angry_from_wakeup(self, chat_id: str): """设置因被吵醒而愤怒的状态""" mood = self.get_mood_by_chat_id(chat_id) mood.is_angry_from_wakeup = True mood.mood_state = "被人吵醒了非常生气" mood.last_change_time = time.time() logger.info(f"{mood.log_prefix} 因被吵醒设置为愤怒状态") def clear_angry_from_wakeup(self, chat_id: str): """清除因被吵醒而愤怒的状态""" mood = self.get_mood_by_chat_id(chat_id) if mood.is_angry_from_wakeup: mood.is_angry_from_wakeup = False mood.mood_state = "感觉很平静" logger.info(f"{mood.log_prefix} 清除被吵醒的愤怒状态") def start_insomnia(self, chat_id: str): """开始一个聊天的失眠状态，锁定情绪更新""" logger.info(f"Chat [{chat_id}]进入失眠状态，情绪已锁定。") self.insomnia_chats.add(chat_id) def stop_insomnia(self, chat_id: str): """停止一个聊天的失眠状态，解锁情绪更新""" logger.info(f"Chat [{chat_id}]失眠状态结束，情绪已解锁。") self.insomnia_chats.discard(chat_id) def get_angry_prompt_addition(self, chat_id: str) -> str: """获取愤怒状态下的提示词补充""" mood = self.get_mood_by_chat_id(chat_id) if mood.is_angry_from_wakeup: return "你被人吵醒了非常生气，说话带着怒气" return "" init_prompt() mood_manager = MoodManager() """全局情绪管理器"""