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

app/chain/message.py

@@ -12,7 +12,7 @@ from typing import Any, Optional, Dict, Union, List, Tuple
 from urllib.parse import unquote, urlparse
 
 from app.application.agent import (
-    get_agent_manager,
+    get_running_agent_manager,
     is_audio_input_available,
     supports_image_input,
     transcribe_audio,
@@ -70,9 +70,14 @@ class MessageChain(ChainBase):
         """
         if not session_id:
             return
+        manager = get_running_agent_manager()
+        if manager is None:
+            return
         clear_task = None
         try:
-            clear_task = get_agent_manager().clear_session(session_id=session_id, user_id=str(userid))
+            clear_task = manager.clear_session(
+                session_id=session_id, user_id=str(userid)
+            )
             asyncio.run_coroutine_threadsafe(
                 clear_task,
                 global_vars.loop,
@@ -350,7 +355,8 @@ class MessageChain(ChainBase):
         if not session_info:
             return False
         session_id, _ = session_info
-        if not get_agent_manager().matches_secret_confirmation(
+        manager = get_running_agent_manager()
+        if manager is None or not manager.matches_secret_confirmation(
             session_id,
             str(userid),
             channel=channel.value,
@@ -968,19 +974,21 @@ class MessageChain(ChainBase):
 
         # 如果有会话ID，同时清除智能体的会话记忆
         if session_id:
+            manager = get_running_agent_manager()
             clear_task = None
-            try:
-                clear_task = get_agent_manager().clear_session(
-                    session_id=session_id, user_id=str(userid)
-                )
-                asyncio.run_coroutine_threadsafe(
-                    clear_task,
-                    global_vars.loop,
-                )
-            except Exception as e:
-                if clear_task:
-                    clear_task.close()
-                logger.warning(f"清除智能体会话记忆失败: {e}")
+            if manager is not None:
+                try:
+                    clear_task = manager.clear_session(
+                        session_id=session_id, user_id=str(userid)
+                    )
+                    asyncio.run_coroutine_threadsafe(
+                        clear_task,
+                        global_vars.loop,
+                    )
+                except Exception as e:
+                    if clear_task:
+                        clear_task.close()
+                    logger.warning(f"清除智能体会话记忆失败: {e}")
 
             self.post_message(
                 Notification(
@@ -1017,12 +1025,16 @@ class MessageChain(ChainBase):
         session_info = self._user_sessions.get(userid)
         if session_info:
             session_id, _ = session_info
+            manager = get_running_agent_manager()
             try:
-                future = asyncio.run_coroutine_threadsafe(
-                    get_agent_manager().stop_current_task(session_id=session_id),
-                    global_vars.loop,
-                )
-                stopped = future.result(timeout=10)
+                if manager is None:
+                    stopped = False
+                else:
+                    future = asyncio.run_coroutine_threadsafe(
+                        manager.stop_current_task(session_id=session_id),
+                        global_vars.loop,
+                    )
+                    stopped = future.result(timeout=10)
             except Exception as e:
                 logger.warning(f"停止Agent推理失败: {e}")
                 stopped = False
@@ -1184,7 +1196,19 @@ class MessageChain(ChainBase):
             return
 
         session_id, _ = session_info
-        status = get_agent_manager().get_session_status(session_id=session_id)
+        manager = get_running_agent_manager()
+        if manager is None:
+            self.post_message(
+                Notification(
+                    channel=channel,
+                    source=source,
+                    title="您当前没有活跃的智能体会话",
+                    userid=userid,
+                    save_history=False,
+                )
+            )
+            return
+        status = manager.get_session_status(session_id=session_id)
         self.post_message(
             Notification(
                 channel=channel,
@@ -1229,6 +1253,20 @@ class MessageChain(ChainBase):
                 )
                 return False
 
+            manager = get_running_agent_manager()
+            if manager is None:
+                self.post_message(
+                    Notification(
+                        channel=channel,
+                        source=source,
+                        userid=userid,
+                        username=username,
+                        title="MoviePilot智能助手服务尚未就绪，请稍后重试",
+                        save_history=False,
+                    )
+                )
+                return False
+
             images = CommingMessage.MessageImage.normalize_list(images)
 
             # 提取用户消息
@@ -1337,7 +1375,7 @@ class MessageChain(ChainBase):
                 process_kwargs["has_audio_input"] = True
             # 在事件循环中处理
             asyncio.run_coroutine_threadsafe(
-                get_agent_manager().process_message(**process_kwargs),
+                manager.process_message(**process_kwargs),
                 global_vars.loop,
             )
             return True
@@ -1854,4 +1892,4 @@ class MessageChain(ChainBase):
             return base64.b64decode(payload)
         except Exception as e:
             logger.error(e)
-            return None
+            return None