modelcontextprotocol/rust-sdk
Watch
1
0
Fork
You've already forked rust-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

feat: provide more context information (#236)

Browse source 
* feat: provide more context information

1. provide context for notification
2. allow extract more info from tool call
3. inject http request part for streamable http server

* docs: add document for getting peer from context
This commit is contained in:
4t145 2025-05-29 11:52:54 +08:00 • committed by GitHub
commit 969fad26cc
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 194 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

25
crates/rmcp/src/transport/streamable_http_server/tower.rs
View file

@ -10,7 +10,7 @@ use tokio_stream::wrappers::ReceiverStream;
use super::session::SessionManager;
use crate::{
RoleServer,
model::ClientJsonRpcMessage,
model::{ClientJsonRpcMessage, GetExtensions},
serve_server,
service::serve_directly,
transport::{
@ -243,7 +243,7 @@ where
// json deserialize request body
let (part, body) = request.into_parts();
let message = match expect_json(body).await {
let mut message = match expect_json(body).await {
Ok(message) => message,
Err(response) => return Ok(response),
};
@ -271,6 +271,20 @@ where
)
.expect("valid response"));
}
// inject request part to extensions
match &mut message {
ClientJsonRpcMessage::Request(req) => {
req.request.extensions_mut().insert(part);
}
ClientJsonRpcMessage::Notification(not) => {
not.notification.extensions_mut().insert(part);
}
_ => {
// skip
}
}
match message {
ClientJsonRpcMessage::Request(_) => {
let stream = self
@ -379,11 +393,8 @@ where
self.config.sse_keep_alive,
))
}
ClientJsonRpcMessage::Notification(notification) => {
service
.handle_notification(notification.notification)
.await
.map_err(internal_error_response("handle notification"))?;
ClientJsonRpcMessage::Notification(_notification) => {
// ignore
Ok(accepted_response())
}
ClientJsonRpcMessage::Response(_json_rpc_response) => Ok(accepted_response()),