lean_ctx/tool_defs/
mod.rs1use std::sync::Arc;
2
3use rmcp::model::Tool;
4use serde_json::{Map, Value};
5
6mod granular;
7pub use granular::{granular_tool_defs, list_all_tool_defs, unified_tool_defs};
8
9pub fn tool_def(name: &'static str, description: &'static str, schema_value: Value) -> Tool {
10 let schema: Map<String, Value> = match schema_value {
11 Value::Object(map) => map,
12 _ => Map::new(),
13 };
14 Tool::new(name, description, Arc::new(schema))
15}
16
17const CORE_TOOL_NAMES: &[&str] = &[
18 "ctx_read",
19 "ctx_search",
20 "ctx_shell",
21 "ctx_tree",
22 "ctx_edit",
23 "ctx_session",
24 "ctx_knowledge",
25 "ctx_overview",
26 "ctx_graph",
27 "ctx_call",
28];
29
30pub fn lazy_tool_defs() -> Vec<Tool> {
31 let all = granular_tool_defs();
32 all.into_iter()
33 .filter(|t| CORE_TOOL_NAMES.contains(&t.name.as_ref()))
34 .collect()
35}
36
37pub fn discover_tools(query: &str) -> String {
38 let all = list_all_tool_defs();
39 let query_lower = query.to_lowercase();
40 let matches: Vec<(&str, &str)> = all
41 .iter()
42 .filter(|(name, desc, _)| {
43 name.to_lowercase().contains(&query_lower) || desc.to_lowercase().contains(&query_lower)
44 })
45 .map(|(name, desc, _)| (*name, *desc))
46 .collect();
47
48 if matches.is_empty() {
49 return format!("No tools found matching '{query}'. Try broader terms like: graph, cost, session, search, compress, agent, workflow, gain.");
50 }
51
52 let mut out = format!("{} tools matching '{query}':\n", matches.len());
53 for (name, desc) in &matches {
54 let short = if desc.len() > 80 { &desc[..80] } else { desc };
55 out.push_str(&format!(" {name} — {short}\n"));
56 }
57 out.push_str(
58 "\nIf your MCP client registers tools only once at startup (static tools/list), \
59use ctx_call (available in lazy mode) to invoke discovered tools:\n\
60 ctx_call {\"name\":\"ctx_graph\",\"arguments\":{\"action\":\"status\"}}\n",
61 );
62 out
63}
64
65pub fn is_full_mode() -> bool {
66 std::env::var("LEAN_CTX_FULL_TOOLS").is_ok()
67 || std::env::var("LEAN_CTX_LAZY_TOOLS")
68 .is_ok_and(|v| v == "0" || v.eq_ignore_ascii_case("false"))
69}