181341db48
- 新增 HumanApprovalModal 组件，用于处理用户对关键操作的审批。 - 引入 useApproval 可组合项，用于管理审批状态和逻辑。 - 更新 AgentChatComponent 以显示审批模态框并处理审批操作。 - 增强消息处理功能，支持工具调用合并并改进对 AI 消息块的处理。 - 重构各种组件和 API，以整合新的审批流程，确保代理交互期间的流畅用户体验。
112 lines
3.7 KiB
Python
112 lines
3.7 KiB
Python
"""MCP Client setup and management for LangGraph ReAct Agent."""
|
|
|
|
from collections.abc import Callable
|
|
from typing import Any, cast
|
|
|
|
from langchain_mcp_adapters.client import ( # type: ignore[import-untyped]
|
|
MultiServerMCPClient,
|
|
)
|
|
|
|
from src.utils import logger
|
|
|
|
# Global MCP tools cache
|
|
_mcp_tools_cache: dict[str, list[Callable[..., Any]]] = {}
|
|
|
|
# MCP Server configurations
|
|
MCP_SERVERS = {
|
|
"sequentialthinking": {
|
|
"url": "https://remote.mcpservers.org/sequentialthinking/mcp",
|
|
"transport": "streamable_http",
|
|
},
|
|
# 这些 stdio 的 MCP server 需要在本地启动，启动的时候需要安装对应的包，需要时间
|
|
# "time": {
|
|
# "command": "uvx",
|
|
# "args": ["mcp-server-time"],
|
|
# "transport": "stdio",
|
|
# },
|
|
# "mcp-server-chart": {
|
|
# "url": "https://mcp.api-inference.modelscope.net/9993ae42524c4c/mcp",
|
|
# "transport": "streamable_http",
|
|
# },
|
|
# 需要在 docker 内安装 npx
|
|
# "mysql": {
|
|
# "command": "npx",
|
|
# "args": ["-y", "@benborla29/mcp-server-mysql@2.0.2"],
|
|
# "env": {
|
|
# "MYSQL_HOST": "172.19.13.6",
|
|
# "MYSQL_PORT": "3306",
|
|
# "MYSQL_USER": "read-only",
|
|
# "MYSQL_PASS": "password123",
|
|
# "MYSQL_DB": "feed"
|
|
# },
|
|
# "transport": "stdio"
|
|
# }
|
|
}
|
|
|
|
|
|
async def get_mcp_client(
|
|
server_configs: dict[str, Any] | None = None,
|
|
) -> MultiServerMCPClient | None:
|
|
"""Initializes an MCP client with the given server configurations."""
|
|
configs = server_configs or MCP_SERVERS
|
|
try:
|
|
client = MultiServerMCPClient(configs) # pyright: ignore[reportArgumentType]
|
|
logger.info(f"Initialized MCP client with servers: {list(configs.keys())}")
|
|
return client
|
|
except Exception as e:
|
|
logger.error("Failed to initialize MCP client: {}", e)
|
|
return None
|
|
|
|
|
|
async def get_mcp_tools(server_name: str, additional_servers: dict[str, dict] = None) -> list[Callable[..., Any]]:
|
|
"""Get MCP tools for a specific server, initializing client if needed."""
|
|
global _mcp_tools_cache
|
|
|
|
# Return cached tools if available
|
|
if server_name in _mcp_tools_cache:
|
|
return _mcp_tools_cache[server_name]
|
|
|
|
mcp_servers = MCP_SERVERS | (additional_servers or {})
|
|
|
|
try:
|
|
assert server_name in mcp_servers, f"Server {server_name} not found in MCP_SERVERS"
|
|
client = await get_mcp_client({server_name: mcp_servers[server_name]})
|
|
if client is None:
|
|
return []
|
|
|
|
# Get all tools and filter by server (if tools have server metadata)
|
|
all_tools = await client.get_tools()
|
|
tools = cast(list[Callable[..., Any]], all_tools)
|
|
|
|
_mcp_tools_cache[server_name] = tools
|
|
logger.info(f"Loaded {len(tools)} tools from MCP server '{server_name}'")
|
|
return tools
|
|
except AssertionError as e:
|
|
logger.warning(f"Failed to load tools from MCP server '{server_name}': {e}")
|
|
return []
|
|
except Exception as e:
|
|
logger.error(f"Failed to load tools from MCP server '{server_name}': {e}")
|
|
return []
|
|
|
|
|
|
async def get_all_mcp_tools() -> list[Callable[..., Any]]:
|
|
"""Get all tools from all configured MCP servers."""
|
|
all_tools = []
|
|
for server_name in MCP_SERVERS.keys():
|
|
tools = await get_mcp_tools(server_name)
|
|
all_tools.extend(tools)
|
|
return all_tools
|
|
|
|
|
|
def add_mcp_server(name: str, config: dict[str, Any]) -> None:
|
|
"""Add a new MCP server configuration."""
|
|
MCP_SERVERS[name] = config
|
|
# Clear client to force reinitialization with new config
|
|
clear_mcp_cache()
|
|
|
|
|
|
def clear_mcp_cache() -> None:
|
|
"""Clear the MCP tools cache (useful for testing)."""
|
|
global _mcp_tools_cache
|
|
_mcp_tools_cache = {}
|