modelcontextprotocol/rust-sdk
Watch
1
0
Fork
You've already forked rust-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2

fix(examples): Prevent transport from closing due to premature client drop (#324)

Browse source
This commit is contained in:
kah 2025-07-21 10:40:47 +09:00 • committed by GitHub
commit 8967daf87f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 9 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
examples/simple-chat-client/src/bin/simple_chat.rs
View file

@ -74,7 +74,7 @@ async fn main() -> Result<()> {
if config.mcp.is_some() {
let mcp_clients = config.create_mcp_clients().await?;
for (name, client) in mcp_clients {
for (name, client) in mcp_clients.iter() {
println!("load MCP tool: {}", name);
let server = client.peer().clone();
let tools = get_mcp_tools(server).await?;
@ -84,6 +84,7 @@ async fn main() -> Result<()> {
tool_set.add_tool(tool);
}
}
tool_set.set_clients(mcp_clients);
}
// create chat session

8
examples/simple-chat-client/src/tool.rs
View file

@ -3,8 +3,9 @@ use std::{collections::HashMap, sync::Arc};
use anyhow::Result;
use async_trait::async_trait;
use rmcp::{
RoleClient,
model::{CallToolRequestParam, CallToolResult, Tool as McpTool},
service::ServerSink,
service::{RunningService, ServerSink},
};
use serde_json::Value;
@ -70,9 +71,14 @@ impl Tool for McpToolAdapter {
#[derive(Default)]
pub struct ToolSet {
tools: HashMap<String, Arc<dyn Tool>>,
clients: HashMap<String, RunningService<RoleClient, ()>>,
}
impl ToolSet {
pub fn set_clients(&mut self, clients: HashMap<String, RunningService<RoleClient, ()>>) {
self.clients = clients;
}
pub fn add_tool<T: Tool + 'static>(&mut self, tool: T) {
self.tools.insert(tool.name(), Arc::new(tool));
}