feat；新的情绪系统

src/mood/mood_manager.py

@@ -1,5 +1,7 @@
 import math
 import random
+import time
+import asyncio
 
 from src.chat.message_receive.message import MessageRecv
 from src.llm_models.utils_model import LLMRequest
@@ -7,6 +9,7 @@ from ..common.logger import get_logger
 from src.chat.utils.chat_message_builder import build_readable_messages, get_raw_msg_by_timestamp_with_chat_inclusive
 from src.config.config import global_config
 from src.chat.utils.prompt_builder import Prompt, global_prompt_manager
+from src.manager.async_task_manager import AsyncTask, async_task_manager
 logger = get_logger("mood")
 
 def init_prompt():
@@ -23,15 +26,29 @@ def init_prompt():
 """,
         "change_mood_prompt",
     )
+    Prompt(
+        """
+{chat_talking_prompt}
+以上是群里最近的聊天记录
+
+{indentify_block}
+你之前的情绪状态是：{mood_state}
+
+距离你上次关注群里消息已经过去了一段时间，你冷静了下来，请你输出一句话描述你现在的情绪状态
+请只输出情绪状态，不要输出其他内容：
+""",
+        "regress_mood_prompt",
+    )
 
 class ChatMood:
     def __init__(self,chat_id:str):
         self.chat_id:str = chat_id
         self.mood_state:str = "感觉很平静"
         
+        self.regression_count:int = 0
         
         self.mood_model = LLMRequest(
-            model=global_config.model.utils,
+            model=global_config.model.emotion,
             temperature=0.7,
             request_type="mood",
         )
@@ -39,6 +56,7 @@ class ChatMood:
         self.last_change_time = 0
         
     async def update_mood_by_message(self,message:MessageRecv,interested_rate:float):
+        self.regression_count = 0
         
         during_last_time = message.message_info.time - self.last_change_time
         
@@ -104,13 +122,95 @@ class ChatMood:
         
         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
+        indentify_block = f"你的名字是{bot_name}{bot_nickname}，你{prompt_personality}："
+        
+        prompt = await global_prompt_manager.format_prompt(
+            "regress_mood_prompt",
+            chat_talking_prompt=chat_talking_prompt,
+            indentify_block=indentify_block,
+            mood_state=self.mood_state,
+        )
+        
+        logger.info(f"prompt: {prompt}")
+        response, (reasoning_content, model_name) = await self.mood_model.generate_response_async(prompt=prompt)
+        logger.info(f"response: {response}")
+        logger.info(f"reasoning_content: {reasoning_content}")
+        
+        
+        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("Running mood regression task...")
+        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.info(f"chat {mood.chat_id} 开始情绪回归, 这是第 {mood.regression_count+1} 次")
+                await mood.regress_mood()
+
 class MoodManager:
 
     def __init__(self):
         self.mood_list:list[ChatMood] = []
         """当前情绪状态"""
+        self.task_started: bool = False
+
+    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:
@@ -124,6 +224,7 @@ class MoodManager:
         for mood in self.mood_list:
             if mood.chat_id == chat_id:
                 mood.mood_state = "感觉很平静"
+                mood.regression_count = 0
                 return
         self.mood_list.append(ChatMood(chat_id))