修正willing_manager内因为代码合并导致的冗余,去除几个print

src/plugins/chat/willing_manager.py

@@ -5,101 +5,98 @@ from typing import Dict
 from .config import global_config
 from .chat_stream import ChatStream
 
+from loguru import logger
+
 
 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
-        
+
     async def _decay_reply_willing(self):
         """定期衰减回复意愿"""
         while True:
             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:
-                
-    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, chat_id: str, willing: float):
         """设置指定聊天流的回复意愿"""
         self.chat_reply_willing[chat_id] = willing
-    def set_willing(self, chat_id: str, willing: float):
+
-        """设置指定聊天流的回复意愿"""
+    async def change_reply_willing_received(
-        self.chat_reply_willing[chat_id] = willing
+        self,
-        
+        chat_stream: ChatStream,
-    async def change_reply_willing_received(self, 
+        topic: str = None,
-                                          chat_stream:ChatStream,
+        is_mentioned_bot: bool = False,
-                                          topic: str = None,
+        config=None,
-                                          is_mentioned_bot: bool = False,
+        is_emoji: bool = False,
-                                          config = None,
+        interested_rate: float = 0,
-                                          is_emoji: bool = False,
+    ) -> float:
-                                          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:
             current_willing += 0.9
-            print(f"被提及, 当前意愿: {current_willing}")
+            logger.debug(f"被提及, 当前意愿: {current_willing}")
         elif is_mentioned_bot:
             current_willing += 0.05
-            print(f"被重复提及, 当前意愿: {current_willing}")
+            logger.debug(f"被重复提及, 当前意愿: {current_willing}")
-        
+
         if is_emoji:
             current_willing *= 0.1
-            print(f"表情包, 当前意愿: {current_willing}")
+            logger.debug(f"表情包, 当前意愿: {current_willing}")
-        
+
-        print(f"放大系数_interested_rate: {global_config.response_interested_rate_amplifier}")
+        logger.debug(f"放大系数_interested_rate: {global_config.response_interested_rate_amplifier}")
-        interested_rate *= global_config.response_interested_rate_amplifier #放大回复兴趣度
+        interested_rate *= global_config.response_interested_rate_amplifier  # 放大回复兴趣度
         if interested_rate > 0.4:
             # print(f"兴趣度: {interested_rate}, 当前意愿: {current_willing}")
-            current_willing += interested_rate-0.4
+            current_willing += interested_rate - 0.4
-        
+
-        current_willing *= global_config.response_willing_amplifier #放大回复意愿
+        current_willing *= global_config.response_willing_amplifier  # 放大回复意愿
         # print(f"放大系数_willing: {global_config.response_willing_amplifier}, 当前意愿: {current_willing}")
-        
+
         reply_probability = max((current_willing - 0.45) * 2, 0)
-        
+
         # 检查群组权限（如果是群聊）
-        if chat_stream.group_info:                
+        if chat_stream.group_info:
             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.chat_reply_willing[chat_id] = min(current_willing, 3.0)
         return reply_probability
-    
+
-    def change_reply_willing_sent(self, chat_stream:ChatStream):
+    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,chat_stream:ChatStream):
+    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):
         """确保衰减任务已启动"""
         if not self._started:
@@ -107,5 +104,6 @@ class WillingManager:
                 self._decay_task = asyncio.create_task(self._decay_reply_willing())
             self._started = True
 
+
 # 创建全局实例
-willing_manager = WillingManager() 
+willing_manager = WillingManager()