Merge pull request #4 from tidewave-ai/sd-streamable

support streamable http transport

Author: SteffenDE

.github/workflows/ci.yml

@@ -18,5 +18,7 @@ jobs:
     - uses: actions/checkout@v4
     - name: Build
       run: cargo build --verbose
+    - name: Build examples
+      run: cargo build --examples
     - name: Run tests
       run: cargo test --verbose

Cargo.lock

@@ -4,7 +4,16 @@ version = "0.1.1"
 edition = "2024"
 
 [dependencies]
-rmcp = { git = "https://github.com/modelcontextprotocol/rust-sdk.git", rev = "afb8a905e54b87c69e880f9377cfe8424aa6f13b", features = ["server", "client", "transport-sse", "transport-child-process"] }
+rmcp = { git = "https://github.com/modelcontextprotocol/rust-sdk.git", rev = "076dc2c2cd8910bee56bae13f29bbcff8c279666", features = [
+    "server",
+    "client",
+    "reqwest",
+    "client-side-sse",
+    "transport-sse-client",
+    "transport-streamable-http-client",
+    "transport-worker",
+    "transport-child-process"
+] }
 clap = { version = "4.5.37", features = ["derive"] }
 tokio = { version = "1", features = ["full"] }
 tracing = "0.1.41"
@@ -20,7 +29,17 @@ version = "0.9"
 features = ["vendored"]
 
 [dev-dependencies]
-rmcp = { git = "https://github.com/modelcontextprotocol/rust-sdk.git", rev = "afb8a905e54b87c69e880f9377cfe8424aa6f13b", features = ["server", "client", "transport-sse", "transport-sse-server", "transport-child-process", "macros"] }
+rmcp = { git = "https://github.com/modelcontextprotocol/rust-sdk.git", rev = "076dc2c2cd8910bee56bae13f29bbcff8c279666", features = [
+    "server",
+    "client",
+    "reqwest",
+    "client-side-sse",
+    "transport-sse-client",
+    "transport-sse-server",
+    "transport-child-process",
+    "transport-streamable-http-server",
+    "macros"
+] }
 axum = { version = "0.8", features = ["macros"] }
 serde = { version = "1.0", features = ["derive"] }
 serde_json = "1.0"

Cargo.toml

@@ -0,0 +1,59 @@
+use anyhow::Context;
+use clap::Parser;
+use rmcp::transport::SseServer;
+use tracing_subscriber::FmtSubscriber;
+
+use rmcp::{
+    ServerHandler,
+    model::{ServerCapabilities, ServerInfo},
+    schemars, tool,
+};
+#[derive(Debug, Clone, Default)]
+pub struct Echo;
+#[tool(tool_box)]
+impl Echo {
+    #[tool(description = "Echo a message")]
+    fn echo(&self, #[tool(param)] message: String) -> String {
+        message
+    }
+}
+
+#[tool(tool_box)]
+impl ServerHandler for Echo {
+    fn get_info(&self) -> ServerInfo {
+        ServerInfo {
+            instructions: Some("A simple echo server".into()),
+            capabilities: ServerCapabilities::builder().enable_tools().build(),
+            ..Default::default()
+        }
+    }
+}
+
+#[derive(Parser)]
+#[command(author, version, about, long_about = None)]
+struct Args {
+    /// Address to bind the server to
+    #[arg(short, long, default_value = "127.0.0.1:8080")]
+    address: std::net::SocketAddr,
+}
+
+#[tokio::main]
+async fn main() -> anyhow::Result<()> {
+    let subscriber = FmtSubscriber::builder()
+        .with_max_level(tracing::Level::DEBUG)
+        .with_writer(std::io::stderr)
+        .finish();
+
+    // Parse command line arguments
+    let args = Args::parse();
+
+    tracing::subscriber::set_global_default(subscriber).context("Failed to set up logging")?;
+
+    let ct = SseServer::serve(args.address)
+        .await?
+        .with_service(Echo::default);
+
+    tokio::signal::ctrl_c().await?;
+    ct.cancel();
+    Ok(())
+}

examples/echo.rs

@@ -0,0 +1,59 @@
+use anyhow::Context;
+use clap::Parser;
+use rmcp::transport::StreamableHttpServer;
+use tracing_subscriber::FmtSubscriber;
+
+use rmcp::{
+    ServerHandler,
+    model::{ServerCapabilities, ServerInfo},
+    schemars, tool,
+};
+#[derive(Debug, Clone, Default)]
+pub struct Echo;
+#[tool(tool_box)]
+impl Echo {
+    #[tool(description = "Echo a message")]
+    fn echo(&self, #[tool(param)] message: String) -> String {
+        message
+    }
+}
+
+#[tool(tool_box)]
+impl ServerHandler for Echo {
+    fn get_info(&self) -> ServerInfo {
+        ServerInfo {
+            instructions: Some("A simple echo server".into()),
+            capabilities: ServerCapabilities::builder().enable_tools().build(),
+            ..Default::default()
+        }
+    }
+}
+
+#[derive(Parser)]
+#[command(author, version, about, long_about = None)]
+struct Args {
+    /// Address to bind the server to
+    #[arg(short, long, default_value = "127.0.0.1:8080")]
+    address: std::net::SocketAddr,
+}
+
+#[tokio::main]
+async fn main() -> anyhow::Result<()> {
+    let subscriber = FmtSubscriber::builder()
+        .with_max_level(tracing::Level::DEBUG)
+        .with_writer(std::io::stderr)
+        .finish();
+
+    // Parse command line arguments
+    let args = Args::parse();
+
+    tracing::subscriber::set_global_default(subscriber).context("Failed to set up logging")?;
+
+    let ct = StreamableHttpServer::serve(args.address)
+        .await?
+        .with_service(Echo::default);
+
+    tokio::signal::ctrl_c().await?;
+    ct.cancel();
+    Ok(())
+}