test: add impl of list/get prompt to main.rs and stdio_integration to
test both new methods
This commit is contained in:
parent
7afab42c61
commit
ef9309cc78
2 changed files with 45 additions and 1 deletions
|
|
@ -82,5 +82,16 @@ async fn main() -> Result<(), ClientError> {
|
||||||
let resource = client.read_resource("memo://insights").await?;
|
let resource = client.read_resource("memo://insights").await?;
|
||||||
println!("Resource: {resource:?}\n");
|
println!("Resource: {resource:?}\n");
|
||||||
|
|
||||||
|
let prompts = client.list_prompts(None).await?;
|
||||||
|
println!("Prompts: {prompts:?}\n");
|
||||||
|
|
||||||
|
let prompt = client
|
||||||
|
.get_prompt(
|
||||||
|
"example_prompt",
|
||||||
|
serde_json::json!({"message": "hello there!"}),
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
println!("Prompt: {prompt:?}\n");
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,6 +1,7 @@
|
||||||
use anyhow::Result;
|
use anyhow::Result;
|
||||||
use mcp_core::content::Content;
|
use mcp_core::content::Content;
|
||||||
use mcp_core::handler::ResourceError;
|
use mcp_core::handler::{PromptError, ResourceError};
|
||||||
|
use mcp_core::prompt::{Prompt, PromptArgument};
|
||||||
use mcp_core::{handler::ToolError, protocol::ServerCapabilities, resource::Resource, tool::Tool};
|
use mcp_core::{handler::ToolError, protocol::ServerCapabilities, resource::Resource, tool::Tool};
|
||||||
use mcp_server::router::{CapabilitiesBuilder, RouterService};
|
use mcp_server::router::{CapabilitiesBuilder, RouterService};
|
||||||
use mcp_server::{ByteTransport, Router, Server};
|
use mcp_server::{ByteTransport, Router, Server};
|
||||||
|
|
@ -61,6 +62,7 @@ impl Router for CounterRouter {
|
||||||
CapabilitiesBuilder::new()
|
CapabilitiesBuilder::new()
|
||||||
.with_tools(false)
|
.with_tools(false)
|
||||||
.with_resources(false, false)
|
.with_resources(false, false)
|
||||||
|
.with_prompts(false)
|
||||||
.build()
|
.build()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -153,6 +155,37 @@ impl Router for CounterRouter {
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn list_prompts(&self) -> Vec<Prompt> {
|
||||||
|
vec![Prompt::new(
|
||||||
|
"example_prompt",
|
||||||
|
Some("This is an example prompt that takes one required agrument, message"),
|
||||||
|
Some(vec![PromptArgument {
|
||||||
|
name: "message".to_string(),
|
||||||
|
description: Some("A message to put in the prompt".to_string()),
|
||||||
|
required: Some(true),
|
||||||
|
}]),
|
||||||
|
)]
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_prompt(
|
||||||
|
&self,
|
||||||
|
prompt_name: &str,
|
||||||
|
) -> Pin<Box<dyn Future<Output = Result<String, PromptError>> + Send + 'static>> {
|
||||||
|
let prompt_name = prompt_name.to_string();
|
||||||
|
Box::pin(async move {
|
||||||
|
match prompt_name.as_str() {
|
||||||
|
"example_prompt" => {
|
||||||
|
let prompt = "This is an example prompt with your message here: '{message}'";
|
||||||
|
Ok(prompt.to_string())
|
||||||
|
}
|
||||||
|
_ => Err(PromptError::NotFound(format!(
|
||||||
|
"Prompt {} not found",
|
||||||
|
prompt_name
|
||||||
|
))),
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tokio::main]
|
#[tokio::main]
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue