lean_ctx/tools/registered/
ctx_knowledge.rs1use rmcp::model::Tool;
2use rmcp::ErrorData;
3use serde_json::{json, Map, Value};
4
5use crate::server::tool_trait::{get_str, get_str_array, McpTool, ToolContext, ToolOutput};
6use crate::tool_defs::tool_def;
7
8pub struct CtxKnowledgeTool;
9
10impl McpTool for CtxKnowledgeTool {
11 fn name(&self) -> &'static str {
12 "ctx_knowledge"
13 }
14
15 fn tool_def(&self) -> Tool {
16 tool_def(
17 "ctx_knowledge",
18 "Persistent project knowledge across sessions (facts, patterns, history). Supports recall modes, embeddings, feedback, and typed relations.",
19 json!({
20 "type": "object",
21 "properties": {
22 "action": {
23 "type": "string",
24 "enum": ["policy", "remember", "recall", "pattern", "feedback", "relate", "unrelate", "relations", "relations_diagram", "consolidate", "status", "health", "remove", "export", "timeline", "rooms", "search", "wakeup", "embeddings_status", "embeddings_reset", "embeddings_reindex"],
25 "description": "Knowledge operation to perform."
26 },
27 "trigger": {
28 "type": "string",
29 "description": "For gotcha action: what triggers the bug"
30 },
31 "resolution": {
32 "type": "string",
33 "description": "For gotcha action: how to fix/avoid it"
34 },
35 "severity": {
36 "type": "string",
37 "enum": ["critical", "warning", "info"],
38 "description": "For gotcha action: severity level (default: warning)"
39 },
40 "category": {
41 "type": "string",
42 "description": "Fact category (architecture, api, testing, deployment, conventions, dependencies)"
43 },
44 "key": {
45 "type": "string",
46 "description": "Fact key/identifier"
47 },
48 "value": {
49 "type": "string",
50 "description": "Value for action (fact value, pattern text, feedback up/down, relation kind)."
51 },
52 "query": {
53 "type": "string",
54 "description": "Query/target for recall/relate/relations."
55 },
56 "mode": {
57 "type": "string",
58 "enum": ["auto", "exact", "semantic", "hybrid"],
59 "description": "Recall mode (default: auto)."
60 },
61 "pattern_type": {
62 "type": "string",
63 "description": "Pattern type for pattern action"
64 },
65 "examples": {
66 "type": "array",
67 "items": { "type": "string" },
68 "description": "Examples for pattern action"
69 },
70 "confidence": {
71 "type": "number",
72 "description": "Confidence score 0.0-1.0 for remember action (default: 0.8)"
73 }
74 },
75 "required": ["action"]
76 }),
77 )
78 }
79
80 fn handle(
81 &self,
82 args: &Map<String, Value>,
83 ctx: &ToolContext,
84 ) -> Result<ToolOutput, ErrorData> {
85 let action = get_str(args, "action")
86 .ok_or_else(|| ErrorData::invalid_params("action is required", None))?;
87 let category = get_str(args, "category");
88 let key = get_str(args, "key");
89 let value = get_str(args, "value");
90 let query = get_str(args, "query");
91 let mode = get_str(args, "mode");
92 let pattern_type = get_str(args, "pattern_type");
93 let examples = get_str_array(args, "examples");
94 let confidence: Option<f32> = args
95 .get("confidence")
96 .and_then(serde_json::Value::as_f64)
97 .map(|v| v as f32);
98
99 let session_handle = ctx
100 .session
101 .as_ref()
102 .ok_or_else(|| ErrorData::internal_error("session not available", None))?;
103 let (session_id, project_root) = {
104 let timeout_dur =
105 crate::core::io_health::adaptive_timeout(std::time::Duration::from_secs(10));
106 let read_result = tokio::task::block_in_place(|| {
107 tokio::runtime::Handle::current()
108 .block_on(tokio::time::timeout(timeout_dur, session_handle.read()))
109 });
110 if let Ok(session) = read_result {
111 let sid = session.id.clone();
112 let root = session
113 .project_root
114 .clone()
115 .unwrap_or_else(|| ctx.project_root.clone());
116 (sid, root)
117 } else {
118 tracing::warn!("ctx_knowledge: session read-lock timeout, using fallback");
119 ("unknown".to_string(), ctx.project_root.clone())
120 }
121 };
122
123 if action == "gotcha" {
124 let trigger = get_str(args, "trigger").unwrap_or_default();
125 let resolution = get_str(args, "resolution").unwrap_or_default();
126 let severity = get_str(args, "severity").unwrap_or_default();
127 let cat = category.as_deref().unwrap_or("convention");
128
129 if trigger.is_empty() || resolution.is_empty() {
130 return Ok(ToolOutput {
131 text: "ERROR: trigger and resolution are required for gotcha action"
132 .to_string(),
133 original_tokens: 0,
134 saved_tokens: 0,
135 mode: Some(action),
136 path: None,
137 changed: false,
138 });
139 }
140
141 let mut store = crate::core::gotcha_tracker::GotchaStore::load(&project_root);
142 let msg = match store.report_gotcha(&trigger, &resolution, cat, &severity, &session_id)
143 {
144 Some(gotcha) => {
145 let conf = (gotcha.confidence * 100.0) as u32;
146 let label = gotcha.category.short_label();
147 format!("Gotcha recorded: [{label}] {trigger} (confidence: {conf}%)")
148 }
149 None => {
150 format!("Gotcha noted: {trigger} (evicted by higher-confidence entries)")
151 }
152 };
153 let _ = store.save(&project_root);
154 return Ok(ToolOutput {
155 text: msg,
156 original_tokens: 0,
157 saved_tokens: 0,
158 mode: Some(action),
159 path: None,
160 changed: false,
161 });
162 }
163
164 let result = crate::tools::ctx_knowledge::handle(
165 &project_root,
166 &action,
167 category.as_deref(),
168 key.as_deref(),
169 value.as_deref(),
170 query.as_deref(),
171 &session_id,
172 pattern_type.as_deref(),
173 examples,
174 confidence,
175 mode.as_deref(),
176 );
177
178 Ok(ToolOutput {
179 text: result,
180 original_tokens: 0,
181 saved_tokens: 0,
182 mode: Some(action),
183 path: None,
184 changed: false,
185 })
186 }
187}