From 761f0e5b038a0ceb6f713e60ea5423887c347552 Mon Sep 17 00:00:00 2001
From: JOJO <1498581755@qq.com>
Date: Thu, 25 Jun 2026 11:52:42 +0800
Subject: [PATCH] fix(mcp_client_manager): restore missing imports and
 dataclass decorators after split

---
 modules/mcp_client_manager/manager.py | 2 ++
 modules/mcp_client_manager/models.py  | 4 ++++
 2 files changed, 6 insertions(+)

diff --git a/modules/mcp_client_manager/manager.py b/modules/mcp_client_manager/manager.py
index 4655a6e..0af7228 100644
--- a/modules/mcp_client_manager/manager.py
+++ b/modules/mcp_client_manager/manager.py
@@ -17,6 +17,8 @@ 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.models import MCPClientPoolEntry, MCPToolBinding
 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 客户端池条目。"""