Skip to content

Commit 2449d19

Browse files
committed
refactor!(kwsearch): remove --search-tool-prompt CLI option
Signed-off-by: Xin Liu <sam@secondstate.io>
1 parent a84cbfc commit 2449d19

File tree

2 files changed

+1
-19
lines changed

2 files changed

+1
-19
lines changed

cardea-kwsearch/cardea-kwsearch-mcp-server/src/main.rs

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ use rmcp::transport::{
66
sse_server::SseServer,
77
streamable_http_server::{StreamableHttpService, session::local::LocalSessionManager},
88
};
9-
use search::{ConnectionConfig, KeywordSearchServer, set_search_tool_prompt};
9+
use search::{ConnectionConfig, KeywordSearchServer};
1010
use tokio::sync::RwLock as TokioRwLock;
1111
use tracing_subscriber::{self, layer::SubscriberExt, util::SubscriberInitExt};
1212

@@ -32,12 +32,6 @@ struct Args {
3232
/// Maximum number of query results to return
3333
#[arg(long, default_value = "10")]
3434
limit: usize,
35-
/// The prompt for the `search` mcp tool
36-
#[arg(
37-
long,
38-
default_value = "Please extract 3 to 5 keywords from my question, separated by spaces. Then, try to return a tool call that invokes the keyword search tool.\n\nMy question is: {query}"
39-
)]
40-
search_tool_prompt: String,
4135
}
4236

4337
#[derive(Debug, Clone, ValueEnum)]
@@ -58,9 +52,6 @@ async fn main() -> anyhow::Result<()> {
5852

5953
let args = Args::parse();
6054

61-
// Set the search tool prompt from CLI
62-
set_search_tool_prompt(args.search_tool_prompt);
63-
6455
let connection_config = ConnectionConfig {
6556
base_url: args.base_url,
6657
api_key: None,

cardea-kwsearch/cardea-kwsearch-mcp-server/src/search.rs

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@ use rmcp::{
1010
service::RequestContext,
1111
tool, tool_handler, tool_router,
1212
};
13-
use std::sync::OnceLock;
1413
use tracing::{error, info};
1514

1615
const PROMPT_SEARCH_TOOL: &str = r#"
@@ -49,12 +48,6 @@ Examples:
4948
</tool_call>
5049
"#;
5150

52-
static SEARCH_TOOL_PROMPT: OnceLock<String> = OnceLock::new();
53-
54-
pub fn set_search_tool_prompt(prompt: String) {
55-
SEARCH_TOOL_PROMPT.set(prompt).unwrap_or_default();
56-
}
57-
5851
#[derive(Debug, Clone)]
5952
pub struct KeywordSearchServer {
6053
tool_router: ToolRouter<Self>,
@@ -238,8 +231,6 @@ impl ServerHandler for KeywordSearchServer {
238231
McpError::invalid_params("No query provided to `search` tool", None)
239232
})?;
240233

241-
// let prompt = SEARCH_TOOL_PROMPT.get().unwrap();
242-
// let prompt = prompt.replace("{query}", &query);
243234
let prompt = format!("{PROMPT_SEARCH_TOOL}\n\n### Input Query\n{query:#?}");
244235

245236
Ok(GetPromptResult {

0 commit comments

Comments
 (0)