joshuacoles/mcp-rb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

It can call tools!

Browse Source
This commit is contained in:
Joshua Coles 2025-03-14 21:38:13 +00:00
parent 212de0524e
commit 24a1ae6ef9
1 changed files with 21 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
ext/mcp/src/internal-test.rs
View File

@ -9,12 +9,12 @@ use jsonrpsee::core::traits::ToRpcParams;
use tokio::process::{Child, ChildStdin, ChildStdout}; use tokio::process::{Child, ChildStdin, ChildStdout};
use tokio::io::{AsyncBufReadExt, AsyncReadExt, AsyncWriteExt, BufReader}; use tokio::io::{AsyncBufReadExt, AsyncReadExt, AsyncWriteExt, BufReader};
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use serde_json::Error; use serde_json::{json, Error};
use serde_json::value::RawValue; use serde_json::value::RawValue;
use tokio::time::sleep; use tokio::time::sleep;
use tracing::debug; use tracing::debug;
use types::{Implementation, InitializeRequestParams, InitializeResult}; use types::{Implementation, InitializeRequestParams, InitializeResult};
use crate::types::{ClientCapabilities, ListToolsRequestParams, ListToolsResult}; use crate::types::{CallToolRequestParams, ClientCapabilities, ListToolsRequestParams, ListToolsResult};
mod types; mod types;
@ -69,6 +69,16 @@ impl<T: Serialize> ToRpcParams for RpcArg<T> {
} }
} }
trait ToRpcArg: Sized {
fn to_rpc(self) -> RpcArg<Self>;
}
impl<T: Serialize> ToRpcArg for &T {
fn to_rpc(self) -> RpcArg<Self> {
RpcArg(self)
}
}
struct NoParams; struct NoParams;
impl ToRpcParams for NoParams { impl ToRpcParams for NoParams {
@ -101,17 +111,21 @@ async fn main() -> anyhow::Result<()> {
transport.clone(), transport.clone(),
); );
let response: InitializeResult = client.request("initialize", RpcArg(InitializeRequestParams { let response: InitializeResult = client.request("initialize", InitializeRequestParams {
capabilities: ClientCapabilities::default(), capabilities: ClientCapabilities::default(),
client_info: Implementation { name: "Rust MCP".to_string(), version: "0.1.0".to_string() }, client_info: Implementation { name: "Rust MCP".to_string(), version: "0.1.0".to_string() },
protocol_version: "2024-11-05".to_string(), protocol_version: "2024-11-05".to_string(),
})).await?; }.to_rpc()).await?;
println!("Response: {:?}", response);
client.notification("notifications/initialized", NoParams).await?; client.notification("notifications/initialized", NoParams).await?;
let response: ListToolsResult = client.request("tools/list", RpcArg(ListToolsRequestParams::default())).await?; let response: ListToolsResult = client.request("tools/list", ListToolsRequestParams::default().to_rpc()).await?;
let response: serde_json::Value = client.request("tools/call", CallToolRequestParams {
arguments: json!({ "url": "http://example.com" }).as_object().unwrap().clone(),
name: "fetch".to_string(),
}.to_rpc()).await?;
println!("Response: {:#?}", response); println!("Response: {:#?}", response);
Ok(()) Ok(())