rust-zero-mcp 0.1.0-alpha.1

Streamable HTTP Model Context Protocol server runtime for rust-zero
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

use rust_zero_core::ServiceGroup;
use rust_zero_mcp::{McpServer, McpServerConfig, Prompt, PromptArgument, Resource, Tool};
use serde_json::json;
use std::{error::Error, time::Duration};

#[actix_web::main]
async fn main() -> Result<(), Box<dyn Error>> {
    let server = McpServer::new(McpServerConfig {
        stateful: true,
        ..McpServerConfig::default()
    })?;

    server.add_tool(
        Tool::new(
            "echo",
            json!({
                "type": "object",
                "properties": {"text": {"type": "string"}},
                "required": ["text"]
            }),
        )
        .with_description("Echo text supplied by the client"),
        |metadata, arguments| async move {
            Ok(json!({
                "content": [{"type": "text", "text": arguments["text"]}],
                "_meta": {"requestId": metadata.header("x-request-id")}
            }))
        },
    );
    server.add_resource(
        Resource {
            uri: "config://service".into(),
            name: "service-config".into(),
            description: Some("Public service configuration".into()),
            mime_type: Some("application/json".into()),
        },
        |_, params| async move {
            Ok(json!({"contents": [{
                "uri": params["uri"],
                "mimeType": "application/json",
                "text": "{\"environment\":\"development\"}"
            }]}))
        },
    );
    server.add_prompt(
        Prompt {
            name: "summarize".into(),
            description: Some("Summarize supplied text".into()),
            arguments: vec![PromptArgument {
                name: "text".into(),
                description: Some("Text to summarize".into()),
                required: true,
            }],
        },
        |_, params| async move {
            Ok(json!({"messages": [{
                "role": "user",
                "content": {
                    "type": "text",
                    "text": format!("Summarize: {}", params["arguments"]["text"])
                }
            }]}))
        },
    );

    let mut services = ServiceGroup::new().with_shutdown_timeout(Duration::from_secs(35));
    services.add("mcp", move |mut shutdown| async move {
        server
            .serve_until(async move { shutdown.requested().await })
            .await
    });
    services.start()?.wait_for_signal().await?;
    Ok(())
}