tooluniverse.mcp_client_tool module

MCP Client Tool for ToolUniverse

This module provides a tool that acts as a client to connect to an existing MCP server, supporting all MCP functionality including tools, resources, and prompts.

class tooluniverse.mcp_client_tool.BaseMCPClient

Bases: object

Base MCP client with common functionality shared between MCPClientTool and MCPAutoLoaderTool. Provides session management, request handling, and async cleanup patterns.

__init__(server_url, transport='http', timeout=30)

async _ensure_session()

Ensure HTTP session is available for HTTP transport

async _close_session()

Close HTTP session if exists

_get_mcp_endpoint(path)

Get the full MCP endpoint URL

async _initialize_mcp_session()

Initialize MCP session if needed (for compatibility with different MCP servers)

async _make_mcp_request(method, params=None)

Make an MCP JSON-RPC request

_run_with_cleanup(async_func)

Common async execution pattern with proper cleanup

class tooluniverse.mcp_client_tool.MCPClientTool

Bases: BaseTool, BaseMCPClient

A tool that acts as an MCP client to connect to existing MCP servers. Supports both HTTP and WebSocket transports.

__init__(tool_config)

async list_tools()

List available tools from the MCP server

async call_tool(name, arguments)

Call a tool on the MCP server

async list_resources()

List available resources from the MCP server

async read_resource(uri)

Read a resource from the MCP server

async list_prompts()

List available prompts from the MCP server

async get_prompt(name, arguments=None)

Get a prompt from the MCP server

run(arguments)

Main run method for the tool. Supports different operations based on the ‘operation’ argument.

async _run_list_tools()

Run list_tools operation

async _run_call_tool(arguments)

Run call_tool operation

async _run_list_resources()

Run list_resources operation

async _run_read_resource(arguments)

Run read_resource operation

async _run_list_prompts()

Run list_prompts operation

async _run_get_prompt(arguments)

Run get_prompt operation

class tooluniverse.mcp_client_tool.MCPProxyTool

Bases: MCPClientTool

A proxy tool that automatically forwards tool calls to an MCP server. This creates individual tools for each tool available on the MCP server.

__init__(tool_config)

run(arguments)

Forward the call directly to the target tool on the MCP server

class tooluniverse.mcp_client_tool.MCPServerDiscovery

Bases: object

Helper class to discover and create tool configurations for MCP servers.

async static discover_server_tools(server_url, transport='http')

Discover all tools available on an MCP server and return tool configurations.

static create_mcp_tools_config(server_configs)

Create tool configurations for multiple MCP servers.

Parameters:

server_configs (List[Dict[str, str]]) – List of server configurations, each containing: - server_url: URL of the MCP server - transport: ‘http’ or ‘websocket’ (optional, defaults to ‘http’) - server_name: Name prefix for tools (optional)

Returns:

List of tool configurations that can be loaded into ToolUniverse

Return type:

List[Dict[str, Any]]

class tooluniverse.mcp_client_tool.MCPAutoLoaderTool

Bases: BaseTool, BaseMCPClient

An advanced MCP tool that automatically discovers and loads all tools from an MCP server. It can register discovered tools as individual ToolUniverse tools for seamless usage.

__init__(tool_config)

async discover_tools()

Discover all available tools from the MCP server

async call_tool(tool_name, arguments)

Directly call an MCP tool by name

generate_proxy_tool_configs()

Generate proxy tool configurations for discovered tools

register_tools_in_engine(engine)

Register discovered tools directly in the ToolUniverse engine

async auto_load_and_register(engine)

Automatically discover, load and register all MCP tools

run(arguments)

Main run method for the auto-loader tool

__del__()

Cleanup when object is destroyed