d66e04baff
This commit applies automated code formatting across the project. The changes primarily involve removing unnecessary blank lines and ensuring consistent code style, improving readability and maintainability without altering functionality.
389 lines
14 KiB
Python
389 lines
14 KiB
Python
# 再用这个就写一行注释来混提交的我直接全部🌿飞😡
|
||
import asyncio
|
||
import time
|
||
import signal
|
||
import sys
|
||
from maim_message import MessageServer
|
||
|
||
from src.common.remote import TelemetryHeartBeatTask
|
||
from src.manager.async_task_manager import async_task_manager
|
||
from src.chat.utils.statistic import OnlineTimeRecordTask, StatisticOutputTask
|
||
from src.chat.emoji_system.emoji_manager import get_emoji_manager
|
||
from src.chat.message_receive.chat_stream import get_chat_manager
|
||
from src.config.config import global_config
|
||
from src.chat.message_receive.bot import chat_bot
|
||
from src.common.logger import get_logger
|
||
from src.individuality.individuality import get_individuality, Individuality
|
||
from src.common.server import get_global_server, Server
|
||
from src.mood.mood_manager import mood_manager
|
||
from rich.traceback import install
|
||
from src.schedule.schedule_manager import schedule_manager
|
||
from src.schedule.monthly_plan_manager import monthly_plan_manager
|
||
from src.plugin_system.core.event_manager import event_manager
|
||
from src.plugin_system.base.component_types import EventType
|
||
# from src.api.main import start_api_server
|
||
|
||
# 导入新的插件管理器和热重载管理器
|
||
from src.plugin_system.core.plugin_manager import plugin_manager
|
||
from src.plugin_system.core.plugin_hot_reload import hot_reload_manager
|
||
|
||
# 导入消息API和traceback模块
|
||
from src.common.message import get_global_api
|
||
|
||
from src.chat.memory_system.Hippocampus import hippocampus_manager
|
||
|
||
if not global_config.memory.enable_memory:
|
||
import src.chat.memory_system.Hippocampus as hippocampus_module
|
||
|
||
class MockHippocampusManager:
|
||
def initialize(self):
|
||
pass
|
||
|
||
def get_hippocampus(self):
|
||
return None
|
||
|
||
async def build_memory(self):
|
||
pass
|
||
|
||
async def forget_memory(self, percentage: float = 0.005):
|
||
pass
|
||
|
||
async def consolidate_memory(self):
|
||
pass
|
||
|
||
async def get_memory_from_text(
|
||
self,
|
||
text: str,
|
||
max_memory_num: int = 3,
|
||
max_memory_length: int = 2,
|
||
max_depth: int = 3,
|
||
fast_retrieval: bool = False,
|
||
) -> list:
|
||
return []
|
||
|
||
async def get_memory_from_topic(
|
||
self, valid_keywords: list[str], max_memory_num: int = 3, max_memory_length: int = 2, max_depth: int = 3
|
||
) -> list:
|
||
return []
|
||
|
||
async def get_activate_from_text(
|
||
self, text: str, max_depth: int = 3, fast_retrieval: bool = False
|
||
) -> tuple[float, list[str]]:
|
||
return 0.0, []
|
||
|
||
def get_memory_from_keyword(self, keyword: str, max_depth: int = 2) -> list:
|
||
return []
|
||
|
||
def get_all_node_names(self) -> list:
|
||
return []
|
||
|
||
hippocampus_module.hippocampus_manager = MockHippocampusManager()
|
||
|
||
# 插件系统现在使用统一的插件加载器
|
||
|
||
install(extra_lines=3)
|
||
|
||
logger = get_logger("main")
|
||
|
||
|
||
class MainSystem:
|
||
def __init__(self):
|
||
self.hippocampus_manager = hippocampus_manager
|
||
|
||
self.individuality: Individuality = get_individuality()
|
||
|
||
# 使用消息API替代直接的FastAPI实例
|
||
self.app: MessageServer = get_global_api()
|
||
self.server: Server = get_global_server()
|
||
|
||
# 设置信号处理器用于优雅退出
|
||
self._setup_signal_handlers()
|
||
|
||
def _setup_signal_handlers(self):
|
||
"""设置信号处理器"""
|
||
|
||
def signal_handler(signum, frame):
|
||
logger.info("收到退出信号，正在优雅关闭系统...")
|
||
self._cleanup()
|
||
sys.exit(0)
|
||
|
||
signal.signal(signal.SIGINT, signal_handler)
|
||
signal.signal(signal.SIGTERM, signal_handler)
|
||
|
||
def _cleanup(self):
|
||
"""清理资源"""
|
||
try:
|
||
# 停止消息重组器
|
||
from src.utils.message_chunker import reassembler
|
||
import asyncio
|
||
|
||
loop = asyncio.get_event_loop()
|
||
if loop.is_running():
|
||
asyncio.create_task(reassembler.stop_cleanup_task())
|
||
else:
|
||
loop.run_until_complete(reassembler.stop_cleanup_task())
|
||
logger.info("🛑 消息重组器已停止")
|
||
except Exception as e:
|
||
logger.error(f"停止消息重组器时出错: {e}")
|
||
|
||
try:
|
||
# 停止插件热重载系统
|
||
hot_reload_manager.stop()
|
||
logger.info("🛑 插件热重载系统已停止")
|
||
except Exception as e:
|
||
logger.error(f"停止热重载系统时出错: {e}")
|
||
|
||
try:
|
||
# 停止异步记忆管理器
|
||
if global_config.memory.enable_memory:
|
||
from src.chat.memory_system.async_memory_optimizer import async_memory_manager
|
||
import asyncio
|
||
|
||
loop = asyncio.get_event_loop()
|
||
if loop.is_running():
|
||
asyncio.create_task(async_memory_manager.shutdown())
|
||
else:
|
||
loop.run_until_complete(async_memory_manager.shutdown())
|
||
logger.info("🛑 记忆管理器已停止")
|
||
except Exception as e:
|
||
logger.error(f"停止记忆管理器时出错: {e}")
|
||
|
||
async def initialize(self):
|
||
"""初始化系统组件"""
|
||
logger.info(f"正在唤醒{global_config.bot.nickname}......")
|
||
|
||
# 其他初始化任务
|
||
await asyncio.gather(self._init_components())
|
||
phrases = [
|
||
("我们的代码里真的没有bug，只有‘特性’.", 10),
|
||
("你知道吗？阿范喜欢被切成臊子😡", 10), # 你加的提示出语法问题来了😡😡😡😡😡😡😡
|
||
("你知道吗,雅诺狐的耳朵其实很好摸", 5),
|
||
("你群最高技术力————言柒姐姐！", 20),
|
||
("初墨小姐宇宙第一(不是)", 10), # 15
|
||
("world.execute(me);", 10),
|
||
("正在尝试连接到MaiBot的服务器...连接失败...，正在转接到maimaiDX", 10),
|
||
("你的bug就像星星一样多，而我的代码像太阳一样，一出来就看不见了。", 10),
|
||
("温馨提示：请不要在代码中留下任何魔法数字，除非你知道它的含义。", 10),
|
||
("世界上只有10种人：懂二进制的和不懂的。", 10),
|
||
("喵喵~你的麦麦被猫娘入侵了喵~", 15),
|
||
("恭喜你触发了稀有彩蛋喵：诺狐嗷呜~ ~", 1),
|
||
("恭喜你！！！你的开发者模式已成功开启，快来加入我们吧！(๑•̀ㅂ•́)و✧ (小声bb:其实是当黑奴)", 10),
|
||
]
|
||
from random import choices
|
||
|
||
# 分离彩蛋和权重
|
||
egg_texts, weights = zip(*phrases, strict=True)
|
||
|
||
# 使用choices进行带权重的随机选择
|
||
selected_egg = choices(egg_texts, weights=weights, k=1)
|
||
eggs = selected_egg[0]
|
||
logger.info(f"""
|
||
全部系统初始化完成，{global_config.bot.nickname}已成功唤醒
|
||
=========================================================
|
||
MoFox_Bot(第三方修改版)
|
||
全部组件已成功启动!
|
||
=========================================================
|
||
🌐 项目地址: https://github.com/MoFox-Studio/MoFox_Bot
|
||
🏠 官方项目: https://github.com/MaiM-with-u/MaiBot
|
||
=========================================================
|
||
这是基于原版MMC的社区改版，包含增强功能和优化(同时也有更多的'特性')
|
||
=========================================================
|
||
小贴士:{eggs}
|
||
""")
|
||
|
||
async def _init_components(self):
|
||
"""初始化其他组件"""
|
||
init_start_time = time.time()
|
||
|
||
# 添加在线时间统计任务
|
||
await async_task_manager.add_task(OnlineTimeRecordTask())
|
||
|
||
# 添加统计信息输出任务
|
||
await async_task_manager.add_task(StatisticOutputTask())
|
||
|
||
# 添加遥测心跳任务
|
||
await async_task_manager.add_task(TelemetryHeartBeatTask())
|
||
|
||
# 注册默认事件
|
||
event_manager.init_default_events()
|
||
|
||
# 初始化权限管理器
|
||
from src.plugin_system.core.permission_manager import PermissionManager
|
||
from src.plugin_system.apis.permission_api import permission_api
|
||
|
||
permission_manager = PermissionManager()
|
||
permission_api.set_permission_manager(permission_manager)
|
||
logger.info("权限管理器初始化成功")
|
||
|
||
# 启动API服务器
|
||
# start_api_server()
|
||
# logger.info("API服务器启动成功")
|
||
|
||
# 加载所有actions，包括默认的和插件的
|
||
plugin_manager.load_all_plugins()
|
||
|
||
# 处理所有缓存的事件订阅（插件加载完成后）
|
||
event_manager.process_all_pending_subscriptions()
|
||
|
||
# 启动插件热重载系统
|
||
hot_reload_manager.start()
|
||
|
||
# 初始化表情管理器
|
||
get_emoji_manager().initialize()
|
||
logger.info("表情包管理器初始化成功")
|
||
|
||
# 启动情绪管理器
|
||
await mood_manager.start()
|
||
logger.info("情绪管理器初始化成功")
|
||
|
||
# 初始化聊天管理器
|
||
|
||
await get_chat_manager()._initialize()
|
||
asyncio.create_task(get_chat_manager()._auto_save_task())
|
||
|
||
logger.info("聊天管理器初始化成功")
|
||
|
||
# 初始化记忆系统
|
||
self.hippocampus_manager.initialize()
|
||
logger.info("记忆系统初始化成功")
|
||
|
||
# 初始化异步记忆管理器
|
||
try:
|
||
from src.chat.memory_system.async_memory_optimizer import async_memory_manager
|
||
|
||
await async_memory_manager.initialize()
|
||
logger.info("记忆管理器初始化成功")
|
||
except Exception as e:
|
||
logger.error(f"记忆管理器初始化失败: {e}")
|
||
|
||
# await asyncio.sleep(0.5) #防止logger输出飞了
|
||
|
||
# 将bot.py中的chat_bot.message_process消息处理函数注册到api.py的消息处理基类中
|
||
self.app.register_message_handler(chat_bot.message_process)
|
||
|
||
# 启动消息重组器的清理任务
|
||
from src.utils.message_chunker import reassembler
|
||
|
||
await reassembler.start_cleanup_task()
|
||
logger.info("消息重组器已启动")
|
||
|
||
# 初始化个体特征
|
||
await self.individuality.initialize()
|
||
|
||
# 初始化月度计划管理器
|
||
if global_config.monthly_plan_system.enable:
|
||
logger.info("正在初始化月度计划管理器...")
|
||
try:
|
||
await monthly_plan_manager.start_monthly_plan_generation()
|
||
logger.info("月度计划管理器初始化成功")
|
||
except Exception as e:
|
||
logger.error(f"月度计划管理器初始化失败: {e}")
|
||
|
||
# 初始化日程管理器
|
||
if global_config.schedule.enable:
|
||
logger.info("日程表功能已启用，正在初始化管理器...")
|
||
await schedule_manager.load_or_generate_today_schedule()
|
||
await schedule_manager.start_daily_schedule_generation()
|
||
logger.info("日程表管理器初始化成功。")
|
||
|
||
try:
|
||
await event_manager.trigger_event(EventType.ON_START, plugin_name="SYSTEM")
|
||
init_time = int(1000 * (time.time() - init_start_time))
|
||
logger.info(f"初始化完成，神经元放电{init_time}次")
|
||
except Exception as e:
|
||
logger.error(f"启动大脑和外部世界失败: {e}")
|
||
raise
|
||
|
||
async def schedule_tasks(self):
|
||
"""调度定时任务"""
|
||
while True:
|
||
tasks = [
|
||
get_emoji_manager().start_periodic_check_register(),
|
||
self.app.run(),
|
||
self.server.run(),
|
||
]
|
||
|
||
# 添加记忆系统相关任务
|
||
tasks.extend(
|
||
[
|
||
self.build_memory_task(),
|
||
self.forget_memory_task(),
|
||
self.consolidate_memory_task(),
|
||
]
|
||
)
|
||
|
||
await asyncio.gather(*tasks)
|
||
|
||
async def build_memory_task(self):
|
||
"""记忆构建任务"""
|
||
while True:
|
||
await asyncio.sleep(global_config.memory.memory_build_interval)
|
||
|
||
try:
|
||
# 使用异步记忆管理器进行非阻塞记忆构建
|
||
from src.chat.memory_system.async_memory_optimizer import build_memory_nonblocking
|
||
|
||
logger.info("正在启动记忆构建")
|
||
|
||
# 定义构建完成的回调函数
|
||
def build_completed(result):
|
||
if result:
|
||
logger.info("记忆构建完成")
|
||
else:
|
||
logger.warning("记忆构建失败")
|
||
|
||
# 启动异步构建，不等待完成
|
||
task_id = await build_memory_nonblocking()
|
||
logger.info(f"记忆构建任务已提交：{task_id}")
|
||
|
||
except ImportError:
|
||
# 如果异步优化器不可用，使用原有的同步方式（但在单独的线程中运行）
|
||
logger.warning("记忆优化器不可用，使用线性运行执行记忆构建")
|
||
|
||
def sync_build_memory():
|
||
"""在线程池中执行同步记忆构建"""
|
||
try:
|
||
loop = asyncio.new_event_loop()
|
||
asyncio.set_event_loop(loop)
|
||
result = loop.run_until_complete(self.hippocampus_manager.build_memory())
|
||
logger.info("记忆构建完成")
|
||
return result
|
||
except Exception as e:
|
||
logger.error(f"记忆构建失败: {e}")
|
||
return None
|
||
finally:
|
||
loop.close()
|
||
|
||
# 在线程池中执行记忆构建
|
||
asyncio.get_event_loop().run_in_executor(None, sync_build_memory)
|
||
|
||
except Exception as e:
|
||
logger.error(f"记忆构建任务启动失败: {e}")
|
||
# fallback到原有的同步方式
|
||
logger.info("正在进行记忆构建（同步模式）")
|
||
await self.hippocampus_manager.build_memory() # type: ignore
|
||
|
||
async def forget_memory_task(self):
|
||
"""记忆遗忘任务"""
|
||
while True:
|
||
await asyncio.sleep(global_config.memory.forget_memory_interval)
|
||
logger.info("[记忆遗忘] 开始遗忘记忆...")
|
||
await self.hippocampus_manager.forget_memory(percentage=global_config.memory.memory_forget_percentage) # type: ignore
|
||
logger.info("[记忆遗忘] 记忆遗忘完成")
|
||
|
||
|
||
|
||
|
||
async def main():
|
||
"""主函数"""
|
||
system = MainSystem()
|
||
await asyncio.gather(
|
||
system.initialize(),
|
||
system.schedule_tasks(),
|
||
)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
asyncio.run(main())
|
||
|
||
|