Merge branch 'refractor' of https://github.com/tcmofashi/MaiMBot into refractor

src/plugins/chat/willing_manager.py

@@ -2,6 +2,9 @@ import asyncio
 from typing import Dict
 from loguru import logger
 
+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
@@ -9,6 +12,7 @@ from .chat_stream import chat_manager,ChatStream
 
 class WillingManager:
     def __init__(self):
+        self.chat_reply_willing: Dict[str, float] = {}  # 存储每个聊天流的回复意愿
         self.chat_reply_willing: Dict[str, float] = {}  # 存储每个聊天流的回复意愿
         self._decay_task = None
         self._started = False
@@ -19,6 +23,8 @@ class WillingManager:
             await asyncio.sleep(5)
             for chat_id in self.chat_reply_willing:
                 self.chat_reply_willing[chat_id] = max(0, self.chat_reply_willing[chat_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,chat_stream:ChatStream) -> float:
         """获取指定聊天流的回复意愿"""
@@ -27,6 +33,9 @@ class WillingManager:
             return self.chat_reply_willing.get(stream.stream_id, 0)
         return 0
     
+    def set_willing(self, chat_id: str, willing: float):
+        """设置指定聊天流的回复意愿"""
+        self.chat_reply_willing[chat_id] = willing
     def set_willing(self, chat_id: str, willing: float):
         """设置指定聊天流的回复意愿"""
         self.chat_reply_willing[chat_id] = willing
@@ -81,6 +90,7 @@ class WillingManager:
         if reply_probability < 0:
             reply_probability = 0
             
+        self.chat_reply_willing[chat_id] = min(current_willing, 3.0)
         self.chat_reply_willing[chat_id] = min(current_willing, 3.0)
         return reply_probability