209 lines
7.5 KiB
Python
209 lines
7.5 KiB
Python
import asyncio
|
|
import time
|
|
|
|
from nonebot import get_driver, on_message, on_notice, require
|
|
from nonebot.adapters.onebot.v11 import Bot, MessageEvent, NoticeEvent
|
|
from nonebot.typing import T_State
|
|
|
|
from ..moods.moods import MoodManager # 导入情绪管理器
|
|
from ..schedule.schedule_generator import bot_schedule
|
|
from ..utils.statistic import LLMStatistics
|
|
from .bot import chat_bot
|
|
from ..config.config import global_config
|
|
from .emoji_manager import emoji_manager
|
|
from .relationship_manager import relationship_manager
|
|
from ..willing.willing_manager import willing_manager
|
|
from .chat_stream import chat_manager
|
|
from .auto_speak import auto_speak_manager # 导入自动发言管理器
|
|
# from ..memory_system.memory import hippocampus
|
|
from src.plugins.memory_system.Hippocampus import HippocampusManager
|
|
from .message_sender import message_manager, message_sender
|
|
from .storage import MessageStorage
|
|
from src.common.logger import get_module_logger
|
|
from src.think_flow_demo.outer_world import outer_world
|
|
from src.think_flow_demo.heartflow import subheartflow_manager
|
|
|
|
logger = get_module_logger("chat_init")
|
|
|
|
# 创建LLM统计实例
|
|
llm_stats = LLMStatistics("llm_statistics.txt")
|
|
|
|
# 添加标志变量
|
|
_message_manager_started = False
|
|
|
|
# 获取驱动器
|
|
driver = get_driver()
|
|
config = driver.config
|
|
|
|
# 初始化表情管理器
|
|
emoji_manager.initialize()
|
|
logger.success("--------------------------------")
|
|
logger.success(f"正在唤醒{global_config.BOT_NICKNAME}......使用版本：{global_config.MAI_VERSION}")
|
|
logger.success("--------------------------------")
|
|
# 注册消息处理器
|
|
msg_in = on_message(priority=5)
|
|
# 注册和bot相关的通知处理器
|
|
notice_matcher = on_notice(priority=1)
|
|
# 创建定时任务
|
|
scheduler = require("nonebot_plugin_apscheduler").scheduler
|
|
|
|
|
|
async def start_think_flow():
|
|
"""启动外部世界"""
|
|
try:
|
|
outer_world_task = asyncio.create_task(outer_world.open_eyes())
|
|
logger.success("大脑和外部世界启动成功")
|
|
# 启动心流系统
|
|
heartflow_task = asyncio.create_task(subheartflow_manager.heartflow_start_working())
|
|
logger.success("心流系统启动成功")
|
|
return outer_world_task, heartflow_task
|
|
except Exception as e:
|
|
logger.error(f"启动大脑和外部世界失败: {e}")
|
|
raise
|
|
|
|
async def start_memory():
|
|
"""启动记忆系统"""
|
|
try:
|
|
start_time = time.time()
|
|
logger.info("开始初始化记忆系统...")
|
|
|
|
# 使用HippocampusManager初始化海马体
|
|
hippocampus_manager = HippocampusManager.get_instance()
|
|
hippocampus_manager.initialize(global_config=global_config)
|
|
|
|
end_time = time.time()
|
|
logger.success(f"记忆系统初始化完成，耗时: {end_time - start_time:.2f} 秒")
|
|
except Exception as e:
|
|
logger.error(f"记忆系统初始化失败: {e}")
|
|
raise
|
|
|
|
|
|
@driver.on_startup
|
|
async def start_background_tasks():
|
|
"""启动后台任务"""
|
|
# 启动LLM统计
|
|
llm_stats.start()
|
|
logger.success("LLM统计功能启动成功")
|
|
|
|
# 初始化并启动情绪管理器
|
|
mood_manager = MoodManager.get_instance()
|
|
mood_manager.start_mood_update(update_interval=global_config.mood_update_interval)
|
|
logger.success("情绪管理器启动成功")
|
|
|
|
# 启动大脑和外部世界
|
|
if global_config.enable_think_flow:
|
|
logger.success("启动测试功能：心流系统")
|
|
await start_think_flow()
|
|
|
|
# 启动自动发言检查任务
|
|
# await auto_speak_manager.start_auto_speak_check()
|
|
# logger.success("自动发言检查任务启动成功")
|
|
|
|
# 只启动表情包管理任务
|
|
asyncio.create_task(emoji_manager.start_periodic_check())
|
|
|
|
asyncio.create_task(start_memory())
|
|
|
|
|
|
@driver.on_startup
|
|
async def init_schedule():
|
|
"""在 NoneBot2 启动时初始化日程系统"""
|
|
bot_schedule.initialize(
|
|
name=global_config.BOT_NICKNAME,
|
|
personality=global_config.PROMPT_PERSONALITY,
|
|
behavior=global_config.PROMPT_SCHEDULE_GEN,
|
|
interval=global_config.SCHEDULE_DOING_UPDATE_INTERVAL)
|
|
asyncio.create_task(bot_schedule.mai_schedule_start())
|
|
|
|
@driver.on_startup
|
|
async def init_relationships():
|
|
"""在 NoneBot2 启动时初始化关系管理器"""
|
|
logger.debug("正在加载用户关系数据...")
|
|
await relationship_manager.load_all_relationships()
|
|
asyncio.create_task(relationship_manager._start_relationship_manager())
|
|
|
|
|
|
@driver.on_bot_connect
|
|
async def _(bot: Bot):
|
|
"""Bot连接成功时的处理"""
|
|
global _message_manager_started
|
|
logger.debug(f"-----------{global_config.BOT_NICKNAME}成功连接！-----------")
|
|
await willing_manager.ensure_started()
|
|
|
|
message_sender.set_bot(bot)
|
|
logger.success("-----------消息发送器已启动！-----------")
|
|
|
|
if not _message_manager_started:
|
|
asyncio.create_task(message_manager.start_processor())
|
|
_message_manager_started = True
|
|
logger.success("-----------消息处理器已启动！-----------")
|
|
|
|
asyncio.create_task(emoji_manager._periodic_scan())
|
|
logger.success("-----------开始偷表情包！-----------")
|
|
asyncio.create_task(chat_manager._initialize())
|
|
asyncio.create_task(chat_manager._auto_save_task())
|
|
|
|
|
|
@msg_in.handle()
|
|
async def _(bot: Bot, event: MessageEvent, state: T_State):
|
|
# 处理合并转发消息
|
|
if "forward" in event.message:
|
|
await chat_bot.handle_forward_message(event, bot)
|
|
else:
|
|
await chat_bot.handle_message(event, bot)
|
|
|
|
|
|
@notice_matcher.handle()
|
|
async def _(bot: Bot, event: NoticeEvent, state: T_State):
|
|
logger.debug(f"收到通知：{event}")
|
|
await chat_bot.handle_notice(event, bot)
|
|
|
|
|
|
# 添加build_memory定时任务
|
|
@scheduler.scheduled_job("interval", seconds=global_config.build_memory_interval, id="build_memory")
|
|
async def build_memory_task():
|
|
"""每build_memory_interval秒执行一次记忆构建"""
|
|
await HippocampusManager.get_instance().build_memory()
|
|
|
|
|
|
@scheduler.scheduled_job("interval", seconds=global_config.forget_memory_interval, id="forget_memory")
|
|
async def forget_memory_task():
|
|
"""每30秒执行一次记忆构建"""
|
|
print("\033[1;32m[记忆遗忘]\033[0m 开始遗忘记忆...")
|
|
await HippocampusManager.get_instance().forget_memory(percentage=global_config.memory_forget_percentage)
|
|
print("\033[1;32m[记忆遗忘]\033[0m 记忆遗忘完成")
|
|
|
|
|
|
@scheduler.scheduled_job("interval", seconds=global_config.build_memory_interval + 10, id="merge_memory")
|
|
async def merge_memory_task():
|
|
"""每30秒执行一次记忆构建"""
|
|
# print("\033[1;32m[记忆整合]\033[0m 开始整合")
|
|
# await hippocampus.operation_merge_memory(percentage=0.1)
|
|
# print("\033[1;32m[记忆整合]\033[0m 记忆整合完成")
|
|
|
|
|
|
@scheduler.scheduled_job("interval", seconds=15, id="print_mood")
|
|
async def print_mood_task():
|
|
"""每30秒打印一次情绪状态"""
|
|
mood_manager = MoodManager.get_instance()
|
|
mood_manager.print_mood_status()
|
|
|
|
|
|
# @scheduler.scheduled_job("interval", seconds=7200, id="generate_schedule")
|
|
# async def generate_schedule_task():
|
|
# """每2小时尝试生成一次日程"""
|
|
# logger.debug("尝试生成日程")
|
|
# await bot_schedule.initialize()
|
|
# if not bot_schedule.enable_output:
|
|
# bot_schedule.print_schedule()
|
|
|
|
|
|
@scheduler.scheduled_job("interval", seconds=3600, id="remove_recalled_message")
|
|
async def remove_recalled_message() -> None:
|
|
"""删除撤回消息"""
|
|
try:
|
|
storage = MessageStorage()
|
|
await storage.remove_recalled_message(time.time())
|
|
except Exception:
|
|
logger.exception("删除撤回消息失败")
|