diff --git a/modules/mcp_client_manager/manager.py b/modules/mcp_client_manager/manager.py
index 4655a6e..b47c8e7 100644
--- a/modules/mcp_client_manager/manager.py
+++ b/modules/mcp_client_manager/manager.py
@@ -17,6 +17,10 @@ from typing import Any, Callable, Dict, Iterable, List, Optional, Tuple, TYPE_CH
 import httpx
 
 from config import MCP_PROTOCOL_VERSION, MCP_DEFAULT_TIMEOUT_SECONDS
+from modules.mcp_client_manager.exceptions import MCPClientError
+from modules.mcp_client_manager.http_client import _StreamableHTTPMCPClient
+from modules.mcp_client_manager.models import MCPClientPoolEntry, MCPToolBinding
+from modules.mcp_client_manager.stdio_client import _StdioMCPClient
 from modules.mcp_server_registry import MCPServerRegistry
 
 if TYPE_CHECKING:
diff --git a/modules/mcp_client_manager/models.py b/modules/mcp_client_manager/models.py
index d007c7d..00f5fd6 100644
--- a/modules/mcp_client_manager/models.py
+++ b/modules/mcp_client_manager/models.py
@@ -23,6 +23,7 @@ if TYPE_CHECKING:
     from modules.user_container_manager import ContainerHandle
 
 
+@dataclass
 class MCPToolBinding:
     alias: str
     server_id: str
@@ -30,6 +31,9 @@ class MCPToolBinding:
     remote_name: str
     transport: str
     timeout_seconds: int
+
+
+@dataclass
 class MCPClientPoolEntry:
     """长连接 MCP 客户端池条目。"""