JOJO/agent-Specialization
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
JOJO 2026-06-25 11:52:42 +08:00
parent 72a49a7c8d
commit 38ff51416f
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/mcp_client_manager/manager.py
View File

@ -17,6 +17,10 @@ from typing import Any, Callable, Dict, Iterable, List, Optional, Tuple, TYPE_CH
import httpx import httpx
from config import MCP_PROTOCOL_VERSION, MCP_DEFAULT_TIMEOUT_SECONDS 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 from modules.mcp_server_registry import MCPServerRegistry
if TYPE_CHECKING: if TYPE_CHECKING:

4
modules/mcp_client_manager/models.py
View File

@ -23,6 +23,7 @@ if TYPE_CHECKING:
from modules.user_container_manager import ContainerHandle from modules.user_container_manager import ContainerHandle
@dataclass
class MCPToolBinding: class MCPToolBinding:
alias: str alias: str
server_id: str server_id: str
@ -30,6 +31,9 @@ class MCPToolBinding:
remote_name: str remote_name: str
transport: str transport: str
timeout_seconds: int timeout_seconds: int
@dataclass
class MCPClientPoolEntry: class MCPClientPoolEntry:
"""长连接 MCP 客户端池条目。""" """长连接 MCP 客户端池条目。"""