3075dc9152* fix(model): deprecate ServerInfo and ClientInfo aliases Those names collide with the protocol's serverInfo/clientInfo fields, which are only Implementation identity. Prefer InitializeResult and InitializeRequestParams. Handler trait signatures, the tool_handler macro output, tests, examples and docs now use the canonical types; the aliases remain as deprecated shims for the rest of 3.x. Fixes #1082 * fix(model): keep public signatures on the deprecated aliases The public API check compares rendered signatures, and rustdoc records a type alias by the name it was written with. Respelling `ServerHandler::get_info`, `ClientHandler::get_info`, `DiscoverResult::from_server_info` and `impl ClientHandler for ClientInfo` onto `InitializeResult`/`InitializeRequestParams` therefore reads as changed public items even though the aliases resolve to those exact types and nothing downstream breaks. Put those four signatures back on the aliases so the published API is byte-identical, and note why at the alias definitions. The deprecation still steers new code, and the macro output, tests, examples and docs keep using the canonical names. * fix(model): migrate leftover ClientInfo test sites Two tests landed on main after the alias deprecation and still named ClientInfo. Clippy -D warnings and the no-local feature test job both failed on them.
122 lines
3.4 KiB
Rust
122 lines
3.4 KiB
Rust
#![cfg(not(feature = "local"))]
|
|
#![cfg(feature = "client")]
|
|
|
|
use rmcp::{
|
|
ClientHandler, RoleClient, RoleServer, ServerHandler,
|
|
handler::server::router::{prompt::PromptRouter, tool::ToolRouter},
|
|
model::{
|
|
CacheScope, InitializeRequestParams, InitializeResult, ListPromptsResult, ListToolsResult,
|
|
ProtocolVersion,
|
|
},
|
|
prompt_handler,
|
|
service::serve_directly,
|
|
tool_handler,
|
|
};
|
|
|
|
#[derive(Debug, Clone)]
|
|
struct CacheHintServer {
|
|
tool_router: ToolRouter<Self>,
|
|
prompt_router: PromptRouter<Self>,
|
|
}
|
|
|
|
impl CacheHintServer {
|
|
fn new() -> Self {
|
|
Self {
|
|
tool_router: ToolRouter::new(),
|
|
prompt_router: PromptRouter::new(),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[tool_handler(router = self.tool_router)]
|
|
#[prompt_handler(router = self.prompt_router)]
|
|
impl ServerHandler for CacheHintServer {}
|
|
|
|
#[derive(Debug, Clone)]
|
|
struct VersionedClient {
|
|
protocol_version: ProtocolVersion,
|
|
}
|
|
|
|
impl ClientHandler for VersionedClient {
|
|
fn get_info(&self) -> InitializeRequestParams {
|
|
let mut info = InitializeRequestParams::default();
|
|
info.protocol_version = self.protocol_version.clone();
|
|
info
|
|
}
|
|
}
|
|
|
|
/// Wires the pair up directly on `protocol_version`. `2026-07-28` removed the
|
|
/// `initialize` handshake, so a peer on that revision is reached the way the
|
|
/// discover lifecycle leaves one: with the version already agreed.
|
|
async fn list_results(protocol_version: ProtocolVersion) -> (ListToolsResult, ListPromptsResult) {
|
|
let (server_transport, client_transport) = tokio::io::duplex(4096);
|
|
|
|
let client_handler = VersionedClient {
|
|
protocol_version: protocol_version.clone(),
|
|
};
|
|
let mut server_peer_info = InitializeResult::default();
|
|
server_peer_info.protocol_version = protocol_version;
|
|
|
|
let server = serve_directly::<RoleServer, _, _, _, _>(
|
|
CacheHintServer::new(),
|
|
server_transport,
|
|
Some(client_handler.get_info()),
|
|
);
|
|
let server_handle = tokio::spawn(async move {
|
|
server.waiting().await?;
|
|
anyhow::Ok(())
|
|
});
|
|
|
|
let client = serve_directly::<RoleClient, _, _, _, _>(
|
|
client_handler,
|
|
client_transport,
|
|
Some(server_peer_info.into()),
|
|
);
|
|
let tools = client
|
|
.list_tools(None)
|
|
.await
|
|
.expect("tools/list should succeed");
|
|
let prompts = client
|
|
.list_prompts(None)
|
|
.await
|
|
.expect("prompts/list should succeed");
|
|
|
|
client.cancel().await.expect("client should cancel");
|
|
server_handle.await.expect("server task").expect("server");
|
|
(tools, prompts)
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn handler_macros_should_emit_required_cache_hints_for_2026_07_28() {
|
|
let (tools, prompts) = list_results(ProtocolVersion::V_2026_07_28).await;
|
|
|
|
assert_eq!(
|
|
(
|
|
tools.ttl_ms,
|
|
tools.cache_scope,
|
|
prompts.ttl_ms,
|
|
prompts.cache_scope,
|
|
),
|
|
(
|
|
Some(0),
|
|
Some(CacheScope::Public),
|
|
Some(0),
|
|
Some(CacheScope::Public),
|
|
)
|
|
);
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn handler_macros_should_omit_cache_hints_for_legacy_versions() {
|
|
let (tools, prompts) = list_results(ProtocolVersion::V_2025_11_25).await;
|
|
|
|
assert_eq!(
|
|
(
|
|
tools.ttl_ms,
|
|
tools.cache_scope,
|
|
prompts.ttl_ms,
|
|
prompts.cache_scope,
|
|
),
|
|
(None, None, None, None)
|
|
);
|
|
}
|