fix(mcp_client_manager): restore missing imports and dataclass decorators after split

This commit is contained in:
JOJO 2026-06-25 11:52:42 +08:00
parent 72a49a7c8d
commit 761f0e5b03
2 changed files with 6 additions and 0 deletions

View File

@ -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:

View File

@ -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 客户端池条目。"""