update:输出单模块部署的mcp接入点

Author: openrz

src/core/connection_manager.py

@@ -7,7 +7,6 @@
 import json
 import time
 import uuid
-import re
 from typing import Dict, Any, List, Optional, Tuple
 from websockets.server import WebSocketServerProtocol
 from websockets.exceptions import ConnectionClosed

src/handlers/websocket_handler.py

@@ -4,10 +4,6 @@
 """
 
 import json
-from typing import Optional, Any
-from urllib.parse import parse_qs, urlparse
-from websockets.server import WebSocketServerProtocol
-from websockets.exceptions import ConnectionClosed
 from ..core.connection_manager import connection_manager
 from ..utils.logger import get_logger
 from ..utils.jsonrpc import (
@@ -116,17 +112,6 @@ async def _handle_robot_message(
         except Exception as e:
             logger.error(f"处理小智端消息时发生错误: {e}")
 
-    def _extract_agent_id(self, path: str) -> Optional[str]:
-        """从路径中提取agentId参数"""
-        try:
-            parsed_url = urlparse(f"ws://localhost{path}")
-            query_params = parse_qs(parsed_url.query)
-            agent_ids = query_params.get("agentId", [])
-            return agent_ids[0] if agent_ids else None
-        except Exception as e:
-            logger.error(f"提取agentId参数失败: {e}")
-            return None
-
 
 # 全局WebSocket处理器实例
 websocket_handler = WebSocketHandler()

src/server.py

@@ -7,9 +7,10 @@
 import json
 import signal
 import uvicorn
+from urllib.parse import quote
 from .utils.config import config
 from .utils.logger import get_logger
-from .utils.aes_utils import decrypt
+from .utils.aes_utils import decrypt, encrypt
 from .utils.jsonrpc import (
     JSONRPCProtocol,
 )
@@ -65,12 +66,21 @@ async def lifespan(app: FastAPI):
     """应用生命周期管理"""
     # 启动时
     logger.info("MCP Endpoint Server 正在启动...")
-    logger.info(f"======================================================")
+    logger.info(f"=====下面的地址分别是智控台/单模块MCP接入点地址====")
     local_ip = get_local_ip()
     logger.info(
-        f"接口地址: http://{local_ip}:{config.getint('server', 'port', 8004)}/mcp_endpoint/health?key={config.get('server', 'key', '')}"
+        f"智控台MCP参数配置: http://{local_ip}:{config.getint('server', 'port', 8004)}/mcp_endpoint/health?key={config.get('server', 'key', '')}"
+    )
+    encrypted_token = encrypt(
+        config.get("server", "key", ""), '{"agentId":"single_module"}'
+    )
+    token = quote(encrypted_token)
+    logger.info(
+        f"单模块部署MCP接入点: ws://{local_ip}:{config.getint('server', 'port', 8004)}/mcp_endpoint/mcp/?token={token}"
+    )
+    logger.info(
+        "=====请根据具体部署选择使用，请勿泄露给任何人======",
     )
-    logger.info("=======上面的地址是MCP接入点地址，请勿泄露给任何人============")
     yield
     # 关闭时
     logger.info("MCP Endpoint Server 已关闭")