Merge branch 'dev' into plugin

src/chat/willing/mode_classical.py

@@ -0,0 +1,55 @@
+import asyncio
+
+from src.config.config import global_config
+from .willing_manager import BaseWillingManager
+
+
+class ClassicalWillingManager(BaseWillingManager):
+    def __init__(self):
+        super().__init__()
+        self._decay_task: asyncio.Task | None = None
+
+    async def _decay_reply_willing(self):
+        """定期衰减回复意愿"""
+        while True:
+            await asyncio.sleep(1)
+            for chat_id in self.chat_reply_willing:
+                self.chat_reply_willing[chat_id] = max(0.0, self.chat_reply_willing[chat_id] * 0.9)
+
+    async def async_task_starter(self):
+        if self._decay_task is None:
+            self._decay_task = asyncio.create_task(self._decay_reply_willing())
+
+    async def get_reply_probability(self, message_id):
+        willing_info = self.ongoing_messages[message_id]
+        chat_id = willing_info.chat_id
+        current_willing = self.chat_reply_willing.get(chat_id, 0)
+
+        interested_rate = willing_info.interested_rate * global_config.normal_chat.response_interested_rate_amplifier
+
+        if interested_rate > 0.4:
+            current_willing += interested_rate - 0.3
+
+        if willing_info.is_mentioned_bot:
+            current_willing += 1 if current_willing < 1.0 else 0.05
+
+        self.chat_reply_willing[chat_id] = min(current_willing, 3.0)
+
+        return min(max((current_willing - 0.5), 0.01) * 2, 1)
+
+    async def before_generate_reply_handle(self, message_id):
+        chat_id = self.ongoing_messages[message_id].chat_id
+        current_willing = self.chat_reply_willing.get(chat_id, 0)
+        self.chat_reply_willing[chat_id] = max(0.0, current_willing - 1.8)
+
+    async def after_generate_reply_handle(self, message_id):
+        if message_id not in self.ongoing_messages:
+            return
+
+        chat_id = self.ongoing_messages[message_id].chat_id
+        current_willing = self.chat_reply_willing.get(chat_id, 0)
+        if current_willing < 1:
+            self.chat_reply_willing[chat_id] = min(1.0, current_willing + 0.4)
+
+    async def not_reply_handle(self, message_id):
+        return await super().not_reply_handle(message_id)