refractor: 几乎写完了，进入测试阶段

src/plugins/chat/willing_manager.py

@@ -1,10 +1,15 @@
 import asyncio
+from typing import Dict
+from loguru import logger
+
 from .config import global_config
+from .message_base import UserInfo, GroupInfo
+from .chat_stream import chat_manager,ChatStream
 
 
 class WillingManager:
     def __init__(self):
-        self.group_reply_willing = {}  # 存储每个群的回复意愿
+        self.chat_reply_willing: Dict[str, float] = {}  # 存储每个聊天流的回复意愿
         self._decay_task = None
         self._started = False
         
@@ -12,20 +17,33 @@ class WillingManager:
         """定期衰减回复意愿"""
         while True:
             await asyncio.sleep(5)
-            for group_id in self.group_reply_willing:
-                self.group_reply_willing[group_id] = max(0, self.group_reply_willing[group_id] * 0.6)
+            for chat_id in self.chat_reply_willing:
+                self.chat_reply_willing[chat_id] = max(0, self.chat_reply_willing[chat_id] * 0.6)
                 
-    def get_willing(self, group_id: int) -> float:
-        """获取指定群组的回复意愿"""
-        return self.group_reply_willing.get(group_id, 0)
+    def get_willing(self,chat_stream:ChatStream) -> float:
+        """获取指定聊天流的回复意愿"""
+        stream = chat_stream
+        if stream:
+            return self.chat_reply_willing.get(stream.stream_id, 0)
+        return 0
     
-    def set_willing(self, group_id: int, willing: float):
-        """设置指定群组的回复意愿"""
-        self.group_reply_willing[group_id] = willing
+    def set_willing(self, chat_id: str, willing: float):
+        """设置指定聊天流的回复意愿"""
+        self.chat_reply_willing[chat_id] = willing
         
-    def change_reply_willing_received(self, group_id: int, topic: str, is_mentioned_bot: bool, config, user_id: int = None, is_emoji: bool = False, interested_rate: float = 0) -> float:
-        """改变指定群组的回复意愿并返回回复概率"""
-        current_willing = self.group_reply_willing.get(group_id, 0)
+    async def change_reply_willing_received(self, 
+                                          chat_stream:ChatStream,
+                                          topic: str = None,
+                                          is_mentioned_bot: bool = False,
+                                          config = None,
+                                          is_emoji: bool = False,
+                                          interested_rate: float = 0) -> float:
+        """改变指定聊天流的回复意愿并返回回复概率"""
+        # 获取或创建聊天流
+        stream = chat_stream
+        chat_id = stream.stream_id
+        
+        current_willing = self.chat_reply_willing.get(chat_id, 0)
         
         # print(f"初始意愿: {current_willing}")
         if is_mentioned_bot and current_willing < 1.0:
@@ -49,31 +67,37 @@ class WillingManager:
         # print(f"放大系数_willing: {global_config.response_willing_amplifier}, 当前意愿: {current_willing}")
         
         reply_probability = max((current_willing - 0.45) * 2, 0)
-        if group_id not in config.talk_allowed_groups:
-            current_willing = 0
-            reply_probability = 0
-            
-        if group_id in config.talk_frequency_down_groups:
-            reply_probability = reply_probability / global_config.down_frequency_rate
+        
+        # 检查群组权限（如果是群聊）
+        if chat_stream.group_info:
+            if chat_stream.group_info.group_id not in config.talk_allowed_groups:
+                current_willing = 0
+                reply_probability = 0
+                
+            if chat_stream.group_info.group_id in config.talk_frequency_down_groups:
+                reply_probability = reply_probability / global_config.down_frequency_rate
 
         reply_probability = min(reply_probability, 1)
         if reply_probability < 0:
             reply_probability = 0
             
-            
-        self.group_reply_willing[group_id] = min(current_willing, 3.0)
+        self.chat_reply_willing[chat_id] = min(current_willing, 3.0)
         return reply_probability
     
-    def change_reply_willing_sent(self, group_id: int):
-        """开始思考后降低群组的回复意愿"""
-        current_willing = self.group_reply_willing.get(group_id, 0)
-        self.group_reply_willing[group_id] = max(0, current_willing - 2)
+    def change_reply_willing_sent(self, chat_stream:ChatStream):
+        """开始思考后降低聊天流的回复意愿"""
+        stream = chat_stream
+        if stream:
+            current_willing = self.chat_reply_willing.get(stream.stream_id, 0)
+            self.chat_reply_willing[stream.stream_id] = max(0, current_willing - 2)
         
-    def change_reply_willing_after_sent(self, group_id: int):
-        """发送消息后提高群组的回复意愿"""
-        current_willing = self.group_reply_willing.get(group_id, 0)
-        if current_willing < 1:
-            self.group_reply_willing[group_id] = min(1, current_willing + 0.2)
+    def change_reply_willing_after_sent(self,chat_stream:ChatStream):
+        """发送消息后提高聊天流的回复意愿"""
+        stream = chat_stream
+        if stream:
+            current_willing = self.chat_reply_willing.get(stream.stream_id, 0)
+            if current_willing < 1:
+                self.chat_reply_willing[stream.stream_id] = min(1, current_willing + 0.2)
         
     async def ensure_started(self):
         """确保衰减任务已启动"""