fix: fill missing fully qualified syntax in prompt_handler macros (#866)
* fix: fill missing fully qualified syntax in prompt_handler macros * fix: remove unused prompt handler test imports --------- Co-authored-by: Dale Seo <5466341+DaleSeo@users.noreply.github.com>
This commit is contained in:
parent
d1cabb458f
commit
415852806d
2 changed files with 8 additions and 14 deletions
|
|
@ -33,9 +33,9 @@ pub fn prompt_handler(attr: TokenStream, input: TokenStream) -> syn::Result<Toke
|
||||||
let get_prompt_impl: ImplItem = parse_quote! {
|
let get_prompt_impl: ImplItem = parse_quote! {
|
||||||
async fn get_prompt(
|
async fn get_prompt(
|
||||||
&self,
|
&self,
|
||||||
request: GetPromptRequestParams,
|
request: rmcp::model::GetPromptRequestParams,
|
||||||
context: RequestContext<RoleServer>,
|
context: rmcp::service::RequestContext<rmcp::RoleServer>,
|
||||||
) -> Result<GetPromptResult, rmcp::ErrorData> {
|
) -> Result<rmcp::model::GetPromptResult, rmcp::ErrorData> {
|
||||||
let prompt_context = rmcp::handler::server::prompt::PromptContext::new(
|
let prompt_context = rmcp::handler::server::prompt::PromptContext::new(
|
||||||
self,
|
self,
|
||||||
request.name,
|
request.name,
|
||||||
|
|
@ -56,11 +56,11 @@ pub fn prompt_handler(attr: TokenStream, input: TokenStream) -> syn::Result<Toke
|
||||||
let list_prompts_impl: ImplItem = parse_quote! {
|
let list_prompts_impl: ImplItem = parse_quote! {
|
||||||
async fn list_prompts(
|
async fn list_prompts(
|
||||||
&self,
|
&self,
|
||||||
_request: Option<PaginatedRequestParams>,
|
_request: Option<rmcp::model::PaginatedRequestParams>,
|
||||||
_context: RequestContext<RoleServer>,
|
_context: rmcp::service::RequestContext<rmcp::RoleServer>,
|
||||||
) -> Result<ListPromptsResult, rmcp::ErrorData> {
|
) -> Result<rmcp::model::ListPromptsResult, rmcp::ErrorData> {
|
||||||
let prompts = #router_expr.list_all();
|
let prompts = #router_expr.list_all();
|
||||||
Ok(ListPromptsResult {
|
Ok(rmcp::model::ListPromptsResult {
|
||||||
prompts,
|
prompts,
|
||||||
meta: #meta,
|
meta: #meta,
|
||||||
next_cursor: None,
|
next_cursor: None,
|
||||||
|
|
|
||||||
|
|
@ -3,13 +3,7 @@
|
||||||
// the ServerHandler trait implementation methods.
|
// the ServerHandler trait implementation methods.
|
||||||
#![allow(dead_code)]
|
#![allow(dead_code)]
|
||||||
|
|
||||||
use rmcp::{
|
use rmcp::{ServerHandler, handler::server::router::prompt::PromptRouter, prompt_handler};
|
||||||
RoleServer, ServerHandler,
|
|
||||||
handler::server::router::prompt::PromptRouter,
|
|
||||||
model::{GetPromptRequestParams, GetPromptResult, ListPromptsResult, PaginatedRequestParams},
|
|
||||||
prompt_handler,
|
|
||||||
service::RequestContext,
|
|
||||||
};
|
|
||||||
|
|
||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
pub struct TestPromptServer {
|
pub struct TestPromptServer {
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue