portugalfuturista/rust-sdk
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(examples): add streamable HTTP section in configs (#296)

Browse source
This commit is contained in:
Tyress K 2025-07-03 12:07:28 +09:00 committed by GitHub
parent ee80903091
commit 31155a440c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 18 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
examples/rig-integration/Cargo.toml
View file

@ -20,6 +20,7 @@ rmcp = { workspace = true, features = [
"reqwest",
"transport-child-process",
"transport-sse-client",
"transport-streamable-http-client"
] }
anyhow = "1.0"
serde_json = "1"

8
examples/rig-integration/src/config/mcp.rs
View file

@ -14,6 +14,9 @@ pub struct McpServerConfig {
#[derive(Debug, Serialize, Deserialize, Clone)]
#[serde(tag = "protocol", rename_all = "lowercase")]
pub enum McpServerTransportConfig {
Streamable {
url: String,
},
Sse {
url: String,
},
@ -60,6 +63,11 @@ impl McpConfig {
impl McpServerTransportConfig {
pub async fn start(&self) -> anyhow::Result<RunningService<RoleClient, ()>> {
let client = match self {
McpServerTransportConfig::Streamable { url } => {
let transport =
rmcp::transport::StreamableHttpClientTransport::from_uri(url.to_string());
().serve(transport).await?
}
McpServerTransportConfig::Sse { url } => {
let transport = rmcp::transport::SseClientTransport::start(url.to_string()).await?;
().serve(transport).await?

1
examples/simple-chat-client/Cargo.toml
View file

@ -18,6 +18,7 @@ rmcp = { workspace = true, features = [
"client",
"transport-child-process",
"transport-sse-client",
"transport-streamable-http-client",
"reqwest"
], no-default-features = true }
clap = { version = "4.0", features = ["derive"] }

8
examples/simple-chat-client/src/config.rs
View file

@ -29,6 +29,9 @@ pub struct McpServerConfig {
#[derive(Debug, Serialize, Deserialize, Clone)]
#[serde(tag = "protocol", rename_all = "lowercase")]
pub enum McpServerTransportConfig {
Streamable {
url: String,
},
Sse {
url: String,
},
@ -44,6 +47,11 @@ pub enum McpServerTransportConfig {
impl McpServerTransportConfig {
pub async fn start(&self) -> Result<RunningService<RoleClient, ()>> {
let client = match self {
McpServerTransportConfig::Streamable { url } => {
let transport =
rmcp::transport::StreamableHttpClientTransport::from_uri(url.to_string());
().serve(transport).await?
}
McpServerTransportConfig::Sse { url } => {
let transport =
rmcp::transport::sse_client::SseClientTransport::start(url.to_owned()).await?;