9e881a645b* feat: implement SEP-1319 Decouple Request Payload from RPC Methods * test: update tests * fix: update handler trait methods to use new types * fix: update examples * fix: correct deprecation version * fix: update wrapper macros to use new *Params type names
104 lines
3.1 KiB
Rust
104 lines
3.1 KiB
Rust
use anyhow::Result;
|
|
use rmcp::{
|
|
ServiceExt,
|
|
model::{CallToolRequestParams, GetPromptRequestParams, ReadResourceRequestParams},
|
|
object,
|
|
transport::{ConfigureCommandExt, TokioChildProcess},
|
|
};
|
|
use tokio::process::Command;
|
|
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<()> {
|
|
// Initialize logging
|
|
tracing_subscriber::registry()
|
|
.with(
|
|
tracing_subscriber::EnvFilter::try_from_default_env()
|
|
.unwrap_or_else(|_| format!("info,{}=debug", env!("CARGO_CRATE_NAME")).into()),
|
|
)
|
|
.with(tracing_subscriber::fmt::layer())
|
|
.init();
|
|
|
|
let client = ()
|
|
.serve(TokioChildProcess::new(Command::new("npx").configure(
|
|
|cmd| {
|
|
cmd.arg("-y").arg("@modelcontextprotocol/server-everything");
|
|
},
|
|
))?)
|
|
.await?;
|
|
|
|
// Initialize
|
|
let server_info = client.peer_info();
|
|
tracing::info!("Connected to server: {server_info:#?}");
|
|
|
|
// List tools
|
|
let tools = client.list_all_tools().await?;
|
|
tracing::info!("Available tools: {tools:#?}");
|
|
|
|
// Call tool echo
|
|
let tool_result = client
|
|
.call_tool(CallToolRequestParams {
|
|
meta: None,
|
|
name: "echo".into(),
|
|
arguments: Some(object!({ "message": "hi from rmcp" })),
|
|
task: None,
|
|
})
|
|
.await?;
|
|
tracing::info!("Tool result for echo: {tool_result:#?}");
|
|
|
|
// Call tool longRunningOperation
|
|
let tool_result = client
|
|
.call_tool(CallToolRequestParams {
|
|
meta: None,
|
|
name: "longRunningOperation".into(),
|
|
arguments: Some(object!({ "duration": 3, "steps": 1 })),
|
|
task: None,
|
|
})
|
|
.await?;
|
|
tracing::info!("Tool result for longRunningOperation: {tool_result:#?}");
|
|
|
|
// List resources
|
|
let resources = client.list_all_resources().await?;
|
|
tracing::info!("Available resources: {resources:#?}");
|
|
|
|
// Read resource
|
|
let resource = client
|
|
.read_resource(ReadResourceRequestParams {
|
|
meta: None,
|
|
uri: "test://static/resource/3".into(),
|
|
})
|
|
.await?;
|
|
tracing::info!("Resource: {resource:#?}");
|
|
|
|
// List prompts
|
|
let prompts = client.list_all_prompts().await?;
|
|
tracing::info!("Available prompts: {prompts:#?}");
|
|
|
|
// Get simple prompt
|
|
let prompt = client
|
|
.get_prompt(GetPromptRequestParams {
|
|
meta: None,
|
|
name: "simple_prompt".into(),
|
|
arguments: None,
|
|
})
|
|
.await?;
|
|
tracing::info!("Prompt - simple: {prompt:#?}");
|
|
|
|
// Get complex prompt (returns text & image)
|
|
let prompt = client
|
|
.get_prompt(GetPromptRequestParams {
|
|
meta: None,
|
|
name: "complex_prompt".into(),
|
|
arguments: Some(object!({ "temperature": "0.5", "style": "formal" })),
|
|
})
|
|
.await?;
|
|
tracing::info!("Prompt - complex: {prompt:#?}");
|
|
|
|
// List resource templates
|
|
let resource_templates = client.list_all_resource_templates().await?;
|
|
tracing::info!("Available resource templates: {resource_templates:#?}");
|
|
|
|
client.cancel().await?;
|
|
|
|
Ok(())
|
|
}
|