mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-09-01 21:47:50 +08:00
jxxghp
7e851dbfa7
- ChainBase 拆分为 RecognitionMixin/MessageProcessingMixin/NotificationMixin - TransferChain 拆分为 7 个功能 mixin（_mixins.py），SubscribeChain 音乐订阅域拆出 _music.py - 斜杠命令交互四件套收敛为 InteractionChainMixin 委托，会话管理器移至 application 层，chain 层不再 re-export - 模块基础类收敛到 app/modules/_base（notification/mediaserver 语义重命名） - 清理 app/chain/__init__.py 24 个未使用导入，修正 49 处测试 patch 目标到实际命名空间 - 兼容层 legacy 符号不再并入 __all__，根治 schemas 初始化反向拉起 application.transfer 的循环导入 - 修复 bangumi 集数为字符串时 set_bangumi_info 抛 TypeError - 新增重复代码等架构门禁测试；capability 清单校验排除下划线内部目录
89 lines
2.5 KiB
Python
89 lines
2.5 KiB
Python
from app.agent.llm import AgentCapabilityManager, LLMHelper
|
|
from app.agent.orchestrator import agent_manager
|
|
from app.agent.prompt import prompt_manager
|
|
from app.agent.prompt.transfer_redo import build_manual_redo_prompt
|
|
from app.application.agent import register_agent_services
|
|
from app.runtime.config import settings
|
|
from app.runtime.log import logger
|
|
|
|
# 导入期即向 application 门面注册实现，保证任何先于 initialize 的
|
|
# 链层调用都能通过门面取到 Agent 服务对象。
|
|
register_agent_services(
|
|
agent_manager=agent_manager,
|
|
prompt_manager=prompt_manager,
|
|
capability_manager=AgentCapabilityManager,
|
|
llm_helper=LLMHelper,
|
|
manual_redo_prompt_builder=build_manual_redo_prompt,
|
|
)
|
|
|
|
|
|
class AgentInitializer:
|
|
"""
|
|
AI智能体初始化器
|
|
"""
|
|
|
|
def __init__(self):
|
|
self._initialized = False
|
|
|
|
async def initialize(self) -> bool:
|
|
"""
|
|
初始化AI智能体管理器
|
|
"""
|
|
try:
|
|
if not settings.AI_AGENT_ENABLE:
|
|
logger.info("AI智能体功能未启用")
|
|
return True
|
|
|
|
await agent_manager.initialize()
|
|
self._initialized = True
|
|
logger.info("AI智能体管理器初始化成功")
|
|
return True
|
|
|
|
except Exception as e:
|
|
logger.error(f"AI智能体管理器初始化失败: {e}")
|
|
return False
|
|
|
|
async def cleanup(self) -> None:
|
|
"""
|
|
清理AI智能体管理器
|
|
"""
|
|
try:
|
|
if not self._initialized:
|
|
return
|
|
await agent_manager.close()
|
|
self._initialized = False
|
|
logger.info("AI智能体管理器已关闭")
|
|
|
|
except Exception as e:
|
|
logger.debug(f"关闭AI智能体管理器时发生错误: {e}")
|
|
|
|
|
|
# 全局AI智能体初始化器实例
|
|
agent_initializer = AgentInitializer()
|
|
|
|
|
|
async def init_agent() -> bool:
|
|
"""
|
|
在应用事件循环中初始化AI智能体。
|
|
"""
|
|
try:
|
|
if not settings.AI_AGENT_ENABLE:
|
|
logger.info("AI智能体功能未启用")
|
|
return True
|
|
|
|
return await agent_initializer.initialize()
|
|
|
|
except Exception as e:
|
|
logger.error(f"初始化AI智能体时发生错误: {e}")
|
|
return False
|
|
|
|
|
|
async def stop_agent():
|
|
"""
|
|
停止AI智能体（异步版本，用于在应用关闭时调用）
|
|
"""
|
|
try:
|
|
await agent_initializer.cleanup()
|
|
except Exception as e:
|
|
logger.error(f"停止AI智能体时发生错误: {e}")
|