Merge branch 'dev' of https://github.com/MoFox-Studio/MoFox_Bot into dev
This commit is contained in:
Windpicker-owo
@@ -420,7 +420,9 @@ class Config(ValidatedConfigBase):
|
||||
default_factory=lambda: CrossContextConfig(), description="跨群聊上下文共享配置"
|
||||
)
|
||||
affinity_flow: AffinityFlowConfig = Field(default_factory=lambda: AffinityFlowConfig(), description="亲和流配置")
|
||||
ProactiveThinking: ProactiveThinkingConfig = Field(default_factory=lambda: AffinityFlowConfig(), description="主动思考配置")
|
||||
proactive_thinking: ProactiveThinkingConfig = Field(
|
||||
default_factory=lambda: ProactiveThinkingConfig(), description="主动思考配置"
|
||||
)
|
||||
|
||||
|
||||
class APIAdapterConfig(ValidatedConfigBase):
|
||||
|
||||
@@ -23,7 +23,7 @@ logger = get_logger(__name__)
|
||||
class ProactiveThinkerPlugin(BasePlugin):
|
||||
"""一个主动思考的插件，但现在还只是个空壳子"""
|
||||
plugin_name: str = "proactive_thinker"
|
||||
enable_plugin: bool = True
|
||||
enable_plugin: bool = False
|
||||
dependencies: list[str] = []
|
||||
python_dependencies: list[str] = []
|
||||
config_file_name: str = "config.toml"
|
||||
|
||||
@@ -1,23 +1,217 @@
|
||||
import asyncio
|
||||
import random
|
||||
import time
|
||||
from datetime import datetime
|
||||
from typing import List, Union, Type, Optional
|
||||
|
||||
from maim_message import UserInfo
|
||||
|
||||
from src.chat.message_receive.chat_stream import get_chat_manager
|
||||
from src.common.logger import get_logger
|
||||
from src.config.config import global_config
|
||||
from src.manager.async_task_manager import async_task_manager, AsyncTask
|
||||
from src.plugin_system import EventType, BaseEventHandler
|
||||
from src.plugin_system.apis import chat_api, person_api
|
||||
from src.plugin_system.base.base_event import HandlerResult
|
||||
|
||||
logger = get_logger(__name__)
|
||||
from src.plugin_system import (
|
||||
EventType,
|
||||
BaseEventHandler,
|
||||
HandlerResult,
|
||||
)
|
||||
|
||||
|
||||
class ColdStartTask(AsyncTask):
|
||||
"""
|
||||
冷启动任务，专门用于处理那些在白名单里，但从未与机器人发生过交互的用户。
|
||||
它的核心职责是“破冰”，主动创建聊天流并发起第一次问候。
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
super().__init__(task_name="ColdStartTask")
|
||||
self.chat_manager = get_chat_manager()
|
||||
|
||||
async def run(self):
|
||||
"""任务主循环，周期性地检查是否有需要“破冰”的新用户。"""
|
||||
logger.info("冷启动任务已启动，将周期性检查白名单中的新朋友。")
|
||||
# 初始等待一段时间，确保其他服务（如数据库）完全启动
|
||||
await asyncio.sleep(20)
|
||||
|
||||
while True:
|
||||
try:
|
||||
logger.info("【冷启动】开始扫描白名单，寻找从未聊过的用户...")
|
||||
|
||||
# 从全局配置中获取私聊白名单
|
||||
enabled_private_chats = global_config.proactive_thinking.enabled_private_chats
|
||||
if not enabled_private_chats:
|
||||
logger.debug("【冷启动】私聊白名单为空，任务暂停一小时。")
|
||||
await asyncio.sleep(3600) # 白名单为空时，没必要频繁检查
|
||||
continue
|
||||
|
||||
# 遍历白名单中的每一个用户
|
||||
for chat_id in enabled_private_chats:
|
||||
try:
|
||||
platform, user_id_str = chat_id.split(":")
|
||||
user_id = int(user_id_str)
|
||||
|
||||
# 【核心逻辑】使用 chat_api 检查该用户是否已经存在聊天流（ChatStream）
|
||||
# 如果返回了 ChatStream 对象，说明已经聊过天了，不是本次任务的目标
|
||||
if chat_api.get_stream_by_user_id(user_id_str, platform):
|
||||
continue # 跳过已存在的用户
|
||||
|
||||
logger.info(f"【冷启动】发现白名单新用户 {chat_id}，准备发起第一次问候。")
|
||||
|
||||
# 【增强体验】尝试从关系数据库中获取该用户的昵称
|
||||
# 这样打招呼时可以更亲切，而不是只知道一个冷冰冰的ID
|
||||
person_id = person_api.get_person_id(platform, user_id)
|
||||
nickname = await person_api.get_person_value(person_id, "nickname")
|
||||
|
||||
# 如果数据库里有昵称，就用数据库里的；如果没有，就用 "用户+ID" 作为备用
|
||||
user_nickname = nickname or f"用户{user_id}"
|
||||
|
||||
# 创建 UserInfo 对象，这是创建聊天流的必要信息
|
||||
user_info = UserInfo(platform=platform, user_id=str(user_id), user_nickname=user_nickname)
|
||||
|
||||
# 【关键步骤】主动创建聊天流。
|
||||
# 创建后，该用户就进入了机器人的“好友列表”，后续将由 ProactiveThinkingTask 接管
|
||||
await self.chat_manager.get_or_create_stream(platform, user_info)
|
||||
|
||||
# TODO: 在这里调用LLM，生成一句自然的、符合人设的“破冰”问候语，并发送给用户。
|
||||
logger.info(f"【冷启动】已为新用户 {chat_id} (昵称: {user_nickname}) 创建聊天流并发送问候。")
|
||||
|
||||
except ValueError:
|
||||
logger.warning(f"【冷启动】白名单条目格式错误或用户ID无效，已跳过: {chat_id}")
|
||||
except Exception as e:
|
||||
logger.error(f"【冷启动】处理用户 {chat_id} 时发生未知错误: {e}", exc_info=True)
|
||||
|
||||
# 完成一轮检查后，进入长时休眠
|
||||
await asyncio.sleep(3600)
|
||||
|
||||
except asyncio.CancelledError:
|
||||
logger.info("冷启动任务被正常取消。")
|
||||
break
|
||||
except Exception as e:
|
||||
logger.error(f"【冷启动】任务出现严重错误，将在5分钟后重试: {e}", exc_info=True)
|
||||
await asyncio.sleep(300)
|
||||
|
||||
|
||||
class ProactiveThinkingTask(AsyncTask):
|
||||
"""
|
||||
主动思考的后台任务（日常唤醒），负责在聊天“冷却”后重新活跃气氛。
|
||||
它只处理已经存在的聊天流。
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
super().__init__(task_name="ProactiveThinkingTask")
|
||||
self.chat_manager = get_chat_manager()
|
||||
|
||||
def _get_next_interval(self) -> float:
|
||||
"""
|
||||
动态计算下一次执行的时间间隔，模拟人类行为的随机性。
|
||||
结合了基础间隔、随机偏移和每日不同时段的活跃度调整。
|
||||
"""
|
||||
# 从配置中读取基础间隔和随机范围
|
||||
base_interval = global_config.proactive_thinking.interval
|
||||
sigma = global_config.proactive_thinking.interval_sigma
|
||||
|
||||
# 1. 在 [base - sigma, base + sigma] 范围内随机取一个值
|
||||
interval = random.uniform(base_interval - sigma, base_interval + sigma)
|
||||
|
||||
# 2. 根据当前时间，应用活跃度调整因子
|
||||
now = datetime.now()
|
||||
current_time_str = now.strftime("%H:%M")
|
||||
|
||||
adjust_rules = global_config.proactive_thinking.talk_frequency_adjust
|
||||
if adjust_rules and adjust_rules[0]:
|
||||
# 按时间对规则排序，确保能找到正确的时间段
|
||||
rules = sorted([rule.split(",") for rule in adjust_rules[0][1:]], key=lambda x: x[0])
|
||||
|
||||
factor = 1.0
|
||||
# 找到最后一个小于等于当前时间的规则
|
||||
for time_str, factor_str in rules:
|
||||
if current_time_str >= time_str:
|
||||
factor = float(factor_str)
|
||||
else:
|
||||
break # 后面的时间都比当前晚，无需再找
|
||||
# factor > 1 表示更活跃，所以用除法来缩短间隔
|
||||
interval /= factor
|
||||
|
||||
# 保证最小间隔，防止过于频繁的骚扰
|
||||
return max(60.0, interval)
|
||||
|
||||
async def run(self):
|
||||
"""任务主循环，周期性地检查所有已存在的聊天是否需要“唤醒”。"""
|
||||
logger.info("日常唤醒任务已启动，将根据动态间隔检查聊天活跃度。")
|
||||
await asyncio.sleep(15) # 初始等待
|
||||
|
||||
while True:
|
||||
# 计算下一次检查前的休眠时间
|
||||
next_interval = self._get_next_interval()
|
||||
try:
|
||||
logger.debug(f"【日常唤醒】下一次检查将在 {next_interval:.2f} 秒后进行。")
|
||||
await asyncio.sleep(next_interval)
|
||||
|
||||
logger.info("【日常唤醒】开始检查不活跃的聊天...")
|
||||
|
||||
# 加载白名单配置
|
||||
enabled_private = set(global_config.proactive_thinking.enabled_private_chats)
|
||||
enabled_groups = set(global_config.proactive_thinking.enabled_group_chats)
|
||||
|
||||
# 获取当前所有聊天流的快照
|
||||
all_streams = list(self.chat_manager.streams.values())
|
||||
|
||||
for stream in all_streams:
|
||||
# 1. 检查该聊天是否在白名单内（或白名单为空时默认允许）
|
||||
is_whitelisted = False
|
||||
if stream.group_info: # 群聊
|
||||
if not enabled_groups or f"qq:{stream.group_info.group_id}" in enabled_groups:
|
||||
is_whitelisted = True
|
||||
else: # 私聊
|
||||
if not enabled_private or f"qq:{stream.user_info.user_id}" in enabled_private:
|
||||
is_whitelisted = True
|
||||
|
||||
if not is_whitelisted:
|
||||
continue # 不在白名单内，跳过
|
||||
|
||||
# 2. 【核心逻辑】检查聊天冷却时间是否足够长
|
||||
time_since_last_active = time.time() - stream.last_active_time
|
||||
if time_since_last_active > next_interval:
|
||||
logger.info(f"【日常唤醒】聊天流 {stream.stream_id} 已冷却 {time_since_last_active:.2f} 秒，触发主动对话。")
|
||||
|
||||
# TODO: 在这里调用LLM，生成一句自然的、符合上下文的问候语，并发送。
|
||||
|
||||
# 【关键步骤】在触发后，立刻更新活跃时间并保存。
|
||||
# 这可以防止在同一个检查周期内，对同一个目标因为意外的延迟而发送多条消息。
|
||||
stream.update_active_time()
|
||||
await self.chat_manager._save_stream(stream)
|
||||
|
||||
except asyncio.CancelledError:
|
||||
logger.info("日常唤醒任务被正常取消。")
|
||||
break
|
||||
except Exception as e:
|
||||
logger.error(f"【日常唤醒】任务出现错误，将在60秒后重试: {e}", exc_info=True)
|
||||
await asyncio.sleep(60)
|
||||
|
||||
|
||||
class ProactiveThinkerEventHandler(BaseEventHandler):
|
||||
"""主动思考需要的启动时触发的事件处理器"""
|
||||
"""主动思考插件的启动事件处理器，负责根据配置启动一个或两个后台任务。"""
|
||||
|
||||
handler_name: str = "proactive_thinker_on_start"
|
||||
handler_description: str = "主动思考插件的启动事件处理器"
|
||||
init_subscribe: List[Union[EventType, str]] = [EventType.ON_START]
|
||||
|
||||
async def execute(self, kwargs: dict | None) -> "HandlerResult":
|
||||
"""执行事件处理"""
|
||||
logger.info("ProactiveThinkerPlugin on_start event triggered.")
|
||||
# 返回 (是否执行成功, 是否需要继续处理, 可选的返回消息)
|
||||
"""在机器人启动时执行，根据配置决定是否启动后台任务。"""
|
||||
logger.info("检测到插件启动事件，正在初始化【主动思考】插件...")
|
||||
# 检查总开关
|
||||
if global_config.proactive_thinking.enable:
|
||||
# 启动负责“日常唤醒”的核心任务
|
||||
logger.info("【主动思考】功能已启用，正在启动“日常唤醒”任务...")
|
||||
proactive_task = ProactiveThinkingTask()
|
||||
await async_task_manager.add_task(proactive_task)
|
||||
|
||||
# 检查“冷启动”功能的独立开关
|
||||
if global_config.proactive_thinking.enable_cold_start:
|
||||
logger.info("“冷启动”功能已启用，正在启动“破冰”任务...")
|
||||
cold_start_task = ColdStartTask()
|
||||
await async_task_manager.add_task(cold_start_task)
|
||||
|
||||
else:
|
||||
logger.info("【主动思考】功能未启用，所有任务均跳过启动。")
|
||||
return HandlerResult(success=True, continue_process=True, message=None)
|
||||
|
||||
@@ -133,54 +133,6 @@ dynamic_distribution_max_interval = 30.0 # 最大分发间隔（秒）
|
||||
dynamic_distribution_jitter_factor = 0.2 # 分发间隔随机扰动因子
|
||||
max_concurrent_distributions = 10 # 最大并发处理的消息流数量，可以根据API性能和服务器负载调整
|
||||
|
||||
talk_frequency_adjust = [
|
||||
["", "8:00,1", "12:00,1.2", "18:00,1.5", "01:00,0.6"],
|
||||
["qq:114514:group", "12:20,1", "16:10,2", "20:10,1", "00:10,0.3"],
|
||||
["qq:1919810:private", "8:20,1", "12:10,2", "20:10,1.5", "00:10,0.2"]
|
||||
]
|
||||
# 基于聊天流的个性化活跃度配置
|
||||
# 格式：[["platform:chat_id:type", "HH:MM,frequency", "HH:MM,frequency", ...], ...]
|
||||
|
||||
# 全局配置示例：
|
||||
# [["", "8:00,1", "12:00,2", "18:00,1.5", "00:00,0.5"]]
|
||||
|
||||
# 特定聊天流配置示例：
|
||||
# [
|
||||
# ["", "8:00,1", "12:00,1.2", "18:00,1.5", "01:00,0.6"], # 全局默认配置
|
||||
# ["qq:1026294844:group", "12:20,1", "16:10,2", "20:10,1", "00:10,0.3"], # 特定群聊配置
|
||||
# ["qq:729957033:private", "8:20,1", "12:10,2", "20:10,1.5", "00:10,0.2"] # 特定私聊配置
|
||||
# ]
|
||||
|
||||
# 说明：
|
||||
# - 当第一个元素为空字符串""时，表示全局默认配置
|
||||
# - 当第一个元素为"platform:id:type"格式时，表示特定聊天流配置
|
||||
# - 后续元素是"时间,频率"格式，表示从该时间开始使用该活跃度，直到下一个时间点
|
||||
# - 优先级：特定聊天流配置 > 全局配置 > 默认 talk_frequency
|
||||
|
||||
# 主动思考功能配置（仅在focus模式下生效）
|
||||
|
||||
enable_proactive_thinking = false # 是否启用主动思考功能
|
||||
proactive_thinking_interval = 1500 # 主动思考触发间隔时间（秒），默认1500秒（25分钟）
|
||||
# TIPS:
|
||||
# 创意玩法：可以设置为0！设置为0时将基于delta_sigma生成纯随机间隔
|
||||
# 负数保险：如果出现了负数，会自动使用绝对值
|
||||
|
||||
proactive_thinking_in_private = true # 主动思考可以在私聊里面启用
|
||||
proactive_thinking_in_group = true # 主动思考可以在群聊里面启用
|
||||
# 主动思考启用范围配置 - 按平台和类型分别配置，建议平台配置为小写
|
||||
# 格式：["platform:user_id", "platform:user_id", ...]
|
||||
# 示例：["qq:123456789", "telegram:user123", "bilibili:987654321"]
|
||||
proactive_thinking_enable_in_private = [] # 启用主动思考的私聊范围，为空则不限制
|
||||
proactive_thinking_enable_in_groups = [] # 启用主动思考的群聊范围，为空则不限制
|
||||
|
||||
delta_sigma = 120 # 正态分布的标准差，控制时间间隔的随机程度
|
||||
# 特殊用法：
|
||||
# - 设置为0：禁用正态分布，使用固定间隔
|
||||
# - 设置得很大（如6000）：产生高度随机的间隔，即使基础间隔为0也能工作
|
||||
# - 负数会自动转换为正数，不用担心配置错误以及极端边界情况
|
||||
# 实验建议：试试 proactive_thinking_interval=0 + delta_sigma 非常大 的纯随机模式！
|
||||
# 结果保证：生成的间隔永远为正数（负数会取绝对值），最小1秒，最大24小时
|
||||
|
||||
|
||||
[relationship]
|
||||
enable_relationship = true # 是否启用关系系统
|
||||
@@ -606,3 +558,32 @@ relationship_weight = 0.3 # 人物关系分数权重
|
||||
mention_bot_adjustment_threshold = 0.3 # 提及bot后的调整阈值
|
||||
mention_bot_interest_score = 0.6 # 提及bot的兴趣分
|
||||
base_relationship_score = 0.3 # 基础人物关系分
|
||||
|
||||
[proactive_thinking] # 主动思考（主动发起对话）功能配置
|
||||
# --- 总开关 ---
|
||||
enable = true # 是否启用主动发起对话功能
|
||||
|
||||
# --- 触发时机 ---
|
||||
# 基础触发间隔（秒），AI会围绕这个时间点主动发起对话
|
||||
interval = 1500 # 默认25分钟
|
||||
# 间隔随机化标准差（秒），让触发时间更自然。设为0则为固定间隔。
|
||||
interval_sigma = 120
|
||||
# 每日活跃度调整，格式：[["", "HH:MM,factor", ...], ["stream_id", ...]]
|
||||
# factor > 1.0 会缩短思考间隔，更活跃；factor < 1.0 会延长间隔。
|
||||
talk_frequency_adjust = [['', '8:00,1', '12:00,1.2', '18:00,1.5', '01:00,0.6']]
|
||||
|
||||
# --- 作用范围 ---
|
||||
enable_in_private = true # 是否允许在私聊中主动发起对话
|
||||
enable_in_group = true # 是否允许在群聊中主动发起对话
|
||||
# 私聊白名单，为空则对所有私聊生效
|
||||
# 格式: ["platform:user_id", ...] e.g., ["qq:123456"]
|
||||
enabled_private_chats = []
|
||||
# 群聊白名单，为空则对所有群聊生效
|
||||
# 格式: ["platform:group_id", ...] e.g., ["qq:7891011"]
|
||||
enabled_group_chats = []
|
||||
|
||||
# --- 冷启动配置 (针对私聊) ---
|
||||
# 对于白名单中不活跃的私聊，是否允许进行一次“冷启动”问候
|
||||
enable_cold_start = true
|
||||
# 冷启动后，该私聊的下一次主动思考需要等待的最小时间（秒）
|
||||
cold_start_cooldown = 86400 # 默认24小时
|
||||
Reference in New Issue
Block a user