refactor(agent): 按需加载 Agent 运行时 (#6336)

app/startup/agent_initializer.py

@@ -1,20 +1,81 @@
-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
+from typing import Any
 
-# 导入期即向 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,
+from app.agent.runtime_loader import (
+    activate_agent_service,
+    begin_agent_shutdown,
+    get_agent_manager as get_runtime_agent_manager,
+    get_running_agent_manager as get_runtime_running_agent_manager,
+    is_tool_factory_materialized,
+    reconcile_agent_service,
 )
+from app.application.agent import register_agent_service_providers
+from app.runtime.config import settings
+from app.runtime.events import Event, eventmanager
+from app.runtime.log import logger
+from app.schemas.types import EventType
+
+
+# 嵌入式启动器可显式注入 manager；常规进程使用 Capability Runtime。
+agent_manager: Any = None
+
+
+def _event_changed_keys(event: Event | None) -> set[str]:
+    """兼容对象和 dict 两种配置事件载荷。"""
+    if event is None:
+        return set()
+    event_data = event.event_data
+    if isinstance(event_data, dict):
+        keys = event_data.get("key", set())
+    else:
+        keys = getattr(event_data, "key", set())
+    if isinstance(keys, str):
+        return {keys}
+    return {str(key) for key in (keys or set())}
+
+
+def _get_agent_manager() -> Any:
+    """兼容显式注入对象，否则按需解析 canonical manager。"""
+    return agent_manager if agent_manager is not None else get_runtime_agent_manager()
+
+
+def _get_running_agent_manager() -> Any | None:
+    """只返回已运行实例，状态探测不得触发 Agent 物化。"""
+    if agent_initializer._compat_injected:
+        return agent_initializer._manager
+    return get_runtime_running_agent_manager()
+
+
+def _get_prompt_manager() -> Any:
+    """首个提示词调用才导入模板管理器。"""
+    from app.agent.prompt import prompt_manager
+
+    return prompt_manager
+
+
+def _get_capability_manager() -> Any:
+    """首个多模态调用才导入 Agent 能力管理器。"""
+    from app.agent.llm import AgentCapabilityManager
+
+    return AgentCapabilityManager
+
+
+def _get_llm_helper() -> Any:
+    """首个模型能力查询才导入 LLM helper。"""
+    from app.agent.llm import LLMHelper
+
+    return LLMHelper
+
+
+def _get_manual_redo_prompt_builder() -> Any:
+    """首个整理接管请求才导入对应提示词构建器。"""
+    from app.agent.prompt.transfer_redo import build_manual_redo_prompt
+
+    return build_manual_redo_prompt
+
+
+async def _handle_agent_config_changed(event: Event) -> None:
+    """把配置事件交给当前全局 initializer，避免监听器持有过期实例。"""
+    await agent_initializer.handle_config_changed(event)
 
 
 class AgentInitializer:
@@ -24,17 +85,33 @@ class AgentInitializer:
 
     def __init__(self):
         self._initialized = False
+        self._manager: Any = None
+        self._compat_injected = False
+        self._shutdown_complete = False
+        eventmanager.add_event_listener(
+            EventType.ConfigChanged,
+            _handle_agent_config_changed,
+        )
 
     async def initialize(self) -> bool:
         """
         初始化AI智能体管理器
         """
         try:
-            if not settings.AI_AGENT_ENABLE:
-                logger.info("AI智能体功能未启用")
-                return True
-
-            await agent_manager.initialize()
+            self._shutdown_complete = False
+            if agent_manager is not None:
+                if not settings.AI_AGENT_ENABLE:
+                    logger.info("AI智能体功能未启用")
+                    return True
+                self._manager = agent_manager
+                self._compat_injected = True
+                await agent_manager.initialize()
+            else:
+                self._manager = await activate_agent_service()
+                self._compat_injected = False
+                if self._manager is None:
+                    logger.info("AI智能体功能未启用")
+                    return True
             self._initialized = True
             logger.info("AI智能体管理器初始化成功")
             return True
@@ -43,16 +120,38 @@ class AgentInitializer:
             logger.error(f"AI智能体管理器初始化失败: {e}")
             return False
 
-    async def cleanup(self) -> None:
-        """
-        清理AI智能体管理器
-        """
+    async def handle_config_changed(self, event: Event) -> None:
+        """仅在 manifest watch 命中时协调 service，关闭态保持 fail closed。"""
+        changed_keys = _event_changed_keys(event)
+        if not changed_keys or self._compat_injected or self._shutdown_complete:
+            return
         try:
-            if not self._initialized:
-                return
-            await agent_manager.close()
+            self._manager = await reconcile_agent_service(
+                reason="agent_service_config_changed",
+                changed_keys=changed_keys,
+                retry=True,
+            )
+            self._initialized = self._manager is not None
+        except Exception as error:
+            self._manager = None
             self._initialized = False
-            logger.info("AI智能体管理器已关闭")
+            logger.debug(f"配置变更协调AI智能体失败: {error}")
+
+    async def cleanup(self) -> None:
+        """清理 initializer 引用；显式注入对象同时在此关闭。"""
+        try:
+            manager = self._manager
+            compat_injected = self._compat_injected
+            if manager is None:
+                return
+            try:
+                if compat_injected:
+                    await manager.close()
+                logger.info("AI智能体管理器已关闭")
+            finally:
+                self._initialized = False
+                self._manager = None
+                self._compat_injected = False
 
         except Exception as e:
             logger.debug(f"关闭AI智能体管理器时发生错误: {e}")
@@ -61,16 +160,22 @@ class AgentInitializer:
 # 全局AI智能体初始化器实例
 agent_initializer = AgentInitializer()
 
+# application 门面仅保存 provider；下列注册不会导入 Agent 实现。
+register_agent_service_providers(
+    agent_manager_provider=_get_agent_manager,
+    running_agent_manager_provider=_get_running_agent_manager,
+    prompt_manager_provider=_get_prompt_manager,
+    capability_manager_provider=_get_capability_manager,
+    llm_helper_provider=_get_llm_helper,
+    manual_redo_prompt_builder_provider=_get_manual_redo_prompt_builder,
+)
+
 
 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:
@@ -83,6 +188,16 @@ async def stop_agent():
     停止AI智能体（异步版本，用于在应用关闭时调用）
     """
     try:
-        await agent_initializer.cleanup()
+        if not agent_initializer._shutdown_complete:
+            if agent_initializer._compat_injected:
+                await agent_initializer.cleanup()
+            else:
+                await begin_agent_shutdown()
+                await agent_initializer.cleanup()
+            agent_initializer._shutdown_complete = True
+        if is_tool_factory_materialized():
+            from app.agent.tools.base import shutdown_blocking_executors
+
+            shutdown_blocking_executors(cancel_futures=True)
     except Exception as e:
         logger.error(f"停止AI智能体时发生错误: {e}")