Skip to main content

mcp

MCP (Model Context Protocol) tools integration.

class MCPClient(server_url: str)​

Client for communicating with MCP servers.

__init__(server_url: str)​

Initialize MCP client.

Args: server_url: URL of the MCP server

list_tools() -> List[Dict[str, Any]]​

List available tools from MCP server.

Returns: List of tool definitions

call_tool(tool_name: str, arguments: Dict[str, Any]) -> Any​

Call a tool on the MCP server.

Args: tool_name: Name of the tool to call arguments: Tool arguments

Returns: Tool execution result

close() -> None​

Close the HTTP client.

class MCPToolManager() -> None​

Manage MCP tools and servers.

__init__() -> None​

Initialize MCP tool manager.

add_server(server: datalayer_core.agents.models.MCPServer) -> None​

Add an MCP server.

Args: server: MCP server configuration

remove_server(server_id: str) -> None​

Remove an MCP server.

Args: server_id: ID of the server to remove

update_server(server_id: str, server: datalayer_core.agents.models.MCPServer) -> None​

Update an MCP server configuration.

Args: server_id: ID of the server to update server: New server configuration

get_servers() -> List[datalayer_core.agents.models.MCPServer]​

Get all MCP servers.

Returns: List of MCP server configurations

get_available_tools() -> List[Dict[str, Any]]​

Get all available tools from enabled MCP servers.

Returns: List of tool definitions with server information

register_with_agent(agent: Any) -> None​

Register MCP tools with Pydantic AI agent.

Args: agent: The Pydantic AI agent

close_all() -> None​

Close all MCP clients.