Struct McpClient

pub struct McpClient { /* private fields */ }

MCP client that manages connections to multiple MCP servers

The main interface for interacting with Model Context Protocol servers. Handles connection lifecycle, tool discovery, and provides integration with tool calling systems.

Features

• Multi-server Management: Connects to and manages multiple MCP servers simultaneously
• Automatic Tool Discovery: Discovers available tools from connected servers
• Tool Registration: Converts MCP tools to internal Tool format for seamless integration
• Connection Pooling: Maintains persistent connections for efficient tool execution
• Error Handling: Robust error handling with proper cleanup and reconnection logic

Example

use mistralrs_mcp::{McpClient, McpClientConfig};

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    let config = McpClientConfig::default();
    let mut client = McpClient::new(config);
     
    // Initialize all configured server connections
    client.initialize().await?;
     
    // Get tool callbacks for model integration
    let callbacks = client.get_tool_callbacks_with_tools();
     
    Ok(())
}

Implementations

impl McpClient

pub fn new(config: McpClientConfig) -> McpClient

Create a new MCP client with the given configuration

pub async fn initialize(&mut self) -> Result<(), Error>

Initialize connections to all configured servers

pub fn get_tool_callbacks( &self, ) -> &HashMap<String, Arc<dyn Fn(&CalledFunction) -> Result<String, Error> + Send + Sync>>

Get tool callbacks that can be used with the existing tool calling system

pub fn get_tool_callbacks_with_tools( &self, ) -> &HashMap<String, ToolCallbackWithTool>

Get tool callbacks with their associated Tool definitions

pub fn get_tools(&self) -> &HashMap<String, McpToolInfo>

Get discovered tools information