[预设分享]低活跃多说话，高活跃少插嘴

[fatinghenji]

import asyncio
from typing import Dict, Optional, Union
from datetime import datetime, timedelta
from ..chat.chat_stream import ChatStream

class WillingManager:
    def __init__(self):
        self.chat_reply_willing: Dict[str, float] = {}
        self._decay_task = None
        self._started = False
        self.MAX_WILLING = 5.0
        
        # 群活跃度追踪
        self.group_activity: Dict[str, Dict[str, Union[datetime, int]]] = {}
        self.activity_window = timedelta(minutes=30)
        self.max_activity = 50

    async def ensure_started(self):
        """确保衰减任务已启动"""
        if not self._started:
            self._decay_task = asyncio.create_task(self._decay_reply_willing())
            self._started = True

    async def _decay_reply_willing(self):
        """固定速率线性衰减"""
        while True:
            await asyncio.sleep(3)
            for chat_id in list(self.chat_reply_willing.keys()):
                current = self.chat_reply_willing[chat_id]
                self.chat_reply_willing[chat_id] = max(0.0, current - 0.4)

    # ================== 新增的意愿值调整方法 ==================
    def change_reply_willing_not_sent(self, chat_stream: Union[ChatStream, str]):
        """消息未发送时调整意愿值"""
        current = self.get_willing(chat_stream)
        self.set_willing(chat_stream, current - 0.3)

    def change_reply_willing_sent(self, chat_stream: Union[ChatStream, str]):
        """消息发送后调整意愿值"""
        current = self.get_willing(chat_stream)
        self.set_willing(chat_stream, current - 1.0)

    # ================== 活跃度相关方法 ==================
    def _update_activity(self, chat_stream: Union[ChatStream, str]):
        """更新群活跃度统计"""
        chat_id = self._get_chat_key(chat_stream)
        now = datetime.now()
        
        if chat_id not in self.group_activity:
            self.group_activity[chat_id] = {
                'start_time': now,
                'message_count': 1
            }
        else:
            activity = self.group_activity[chat_id]
            if now - activity['start_time'] > self.activity_window:
                activity['start_time'] = now
                activity['message_count'] = 1
            else:
                activity['message_count'] += 1

    def _get_activity_level(self, chat_stream: Union[ChatStream, str]) -> float:
        """获取群活跃度(0.0-1.0)"""
        chat_id = self._get_chat_key(chat_stream)
        if chat_id not in self.group_activity:
            return 0.0
            
        activity = self.group_activity[chat_id]
        elapsed = (datetime.now() - activity['start_time']).total_seconds() / 60
        messages_per_min = activity['message_count'] / max(1, elapsed)
        return min(1.0, messages_per_min / (self.max_activity / 30))

    def _calculate_activity_adjustment(self, activity_level: float) -> float:
        """根据活跃度调整意愿值"""
        if activity_level < 0.3:
            return 0.5 * (0.3 - activity_level) / 0.3
        elif activity_level > 0.7:
            return -0.5 * (activity_level - 0.7) / 0.3
        return 0.0

    # ================== 基础方法 ==================
    def _willing_to_probability(self, current_willing: float) -> float:
        """意愿值转概率"""
        return max(0.0, min(1.0, current_willing / self.MAX_WILLING))

    def get_willing(self, chat_stream: Union[ChatStream, str]) -> float:
        """获取当前意愿值"""
        chat_id = self._get_chat_key(chat_stream)
        return self.chat_reply_willing.get(chat_id, 0.0)

    def set_willing(self, chat_stream: Union[ChatStream, str], willing: float):
        """设置意愿值"""
        chat_id = self._get_chat_key(chat_stream)
        self.chat_reply_willing[chat_id] = max(0.0, min(willing, self.MAX_WILLING))

    def _get_chat_key(self, chat_stream: Union[ChatStream, str]) -> str:
        """生成聊天标识"""
        return chat_stream if isinstance(chat_stream, str) else chat_stream.stream_id

    async def change_reply_willing_received(
        self,
        chat_stream: Union[ChatStream, str],
        topic: Optional[str] = None,
        is_mentioned_bot: bool = False,
        config=None,
        is_emoji: bool = False,
        interested_rate: float = 0,
        sender_id: Optional[str] = None,
    ) -> float:
        """处理接收消息的意愿调整"""
        self._update_activity(chat_stream)
        activity_level = self._get_activity_level(chat_stream)
        activity_adjustment = self._calculate_activity_adjustment(activity_level)

        current = self.get_willing(chat_stream)
        LINEAR_GAIN = 1.0

        current += LINEAR_GAIN * (0.5 + activity_adjustment)

        if topic:
            current += LINEAR_GAIN * 0.4
        if is_mentioned_bot:
            current += LINEAR_GAIN * 1.0
        current += LINEAR_GAIN * interested_rate * 0.6
        if is_emoji:
            current -= LINEAR_GAIN * 0.8

        self.set_willing(chat_stream, current)
        return self._willing_to_probability(current)

# 全局单例
willing_manager = WillingManager()

替换文件内容~\src\plugins\willing\mode_custom.py，然后在配置文件中启用custom即可