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

feat(examples): Add func call support (#168)

Browse source 
Some works about simple-chat.
1. fix some bugs.
2. add func call optional.

Signed-off-by: jokemanfire <hu.dingyang@zte.com.cn>
This commit is contained in:
jokemanfire 2025-05-12 13:52:57 +08:00 • committed by GitHub
commit 6de444c62f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 160 additions and 108 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -3,7 +3,7 @@ use std::{collections::HashMap, sync::Arc};
use anyhow::Result;
use async_trait::async_trait;
use rmcp::{
model::{CallToolRequestParam, Tool as McpTool},
model::{CallToolRequestParam, CallToolResult, Tool as McpTool},
service::ServerSink,
};
use serde_json::Value;
@ -18,7 +18,7 @@ pub trait Tool: Send + Sync {
fn name(&self) -> String;
fn description(&self) -> String;
fn parameters(&self) -> Value;
async fn call(&self, args: Value) -> Result<String>;
async fn call(&self, args: Value) -> Result<CallToolResult>;
}
pub struct McpToolAdapter {
@ -50,12 +50,12 @@ impl Tool for McpToolAdapter {
serde_json::to_value(&self.tool.input_schema).unwrap_or(serde_json::json!({}))
}
async fn call(&self, args: Value) -> Result<String> {
async fn call(&self, args: Value) -> Result<CallToolResult> {
let arguments = match args {
Value::Object(map) => Some(map),
_ => None,
};
println!("arguments: {:?}", arguments);
let call_result = self
.server
.call_tool(CallToolRequestParam {
@ -63,9 +63,8 @@ impl Tool for McpToolAdapter {
arguments,
})
.await?;
let result = serde_json::to_string(&call_result).unwrap();
Ok(result)
Ok(call_result)
}
}
#[derive(Default)]