refactor!(tidb): remove --search-tool-prompt CLI option

Signed-off-by: Xin Liu <sam@secondstate.io>

Author: apepkuss

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

@@ -11,7 +11,7 @@ use rmcp::transport::{
 };
 use rustls::crypto::{CryptoProvider, ring::default_provider};
 use std::{env, path::PathBuf};
-use tidb::{TidbServer, set_search_tool_prompt};
+use tidb::TidbServer;
 use tokio::sync::RwLock as TokioRwLock;
 use tracing::{error, info};
 use tracing_subscriber::{self, layer::SubscriberExt, util::SubscriberInitExt};
@@ -41,12 +41,6 @@ struct Args {
     /// Maximum number of query results to return
     #[arg(long, default_value = "10")]
     limit: u64,
-    /// The prompt for the `search` mcp tool
-    #[arg(
-        long,
-        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}"
-    )]
-    search_tool_prompt: String,
 }
 
 #[derive(Debug, Clone, ValueEnum)]
@@ -125,9 +119,6 @@ async fn main() -> anyhow::Result<()> {
         .set(TokioRwLock::new(config))
         .map_err(|_| anyhow::anyhow!("Failed to set TIDB_ACCESS_CONFIG"))?;
 
-    // Set the search tool prompt from CLI
-    set_search_tool_prompt(args.search_tool_prompt);
-
     info!("Starting Cardea TiDB MCP server on {}", args.socket_addr);
 
     match args.transport {

cardea-tidb/cardea-tidb-mcp-server/src/tidb.rs

@@ -8,7 +8,7 @@ use rmcp::{
     service::RequestContext,
     tool, tool_handler, tool_router,
 };
-use std::{result::Result, sync::OnceLock};
+use std::result::Result;
 use tracing::{error, info};
 
 const PROMPT_SEARCH_TOOL: &str = r#"
@@ -47,12 +47,6 @@ Examples:
     </tool_call>
 "#;
 
-static SEARCH_TOOL_PROMPT: OnceLock<String> = OnceLock::new();
-
-pub fn set_search_tool_prompt(prompt: String) {
-    SEARCH_TOOL_PROMPT.set(prompt).unwrap_or_default();
-}
-
 #[derive(Debug, Clone)]
 pub struct TidbServer {
     tool_router: ToolRouter<Self>,
@@ -257,8 +251,6 @@ impl ServerHandler for TidbServer {
                         McpError::invalid_params("No query provided to `search` tool", None)
                     })?;
 
-                // let prompt = SEARCH_TOOL_PROMPT.get().unwrap();
-                // let prompt = prompt.replace("{query}", &query);
                 let prompt = format!("{PROMPT_SEARCH_TOOL}\n\n### Input Query\n{query:#?}");
 
                 Ok(GetPromptResult {