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.session.as_ref().unwrap();
100 let (session_id, project_root) = {
101 let timeout_dur =
102 crate::core::io_health::adaptive_timeout(std::time::Duration::from_secs(10));
103 let read_result = tokio::task::block_in_place(|| {
104 tokio::runtime::Handle::current()
105 .block_on(tokio::time::timeout(timeout_dur, session_handle.read()))
106 });
107 if let Ok(session) = read_result {
108 let sid = session.id.clone();
109 let root = session
110 .project_root
111 .clone()
112 .unwrap_or_else(|| ctx.project_root.clone());
113 (sid, root)
114 } else {
115 tracing::warn!("ctx_knowledge: session read-lock timeout, using fallback");
116 ("unknown".to_string(), ctx.project_root.clone())
117 }
118 };
119
120 if action == "gotcha" {
121 let trigger = get_str(args, "trigger").unwrap_or_default();
122 let resolution = get_str(args, "resolution").unwrap_or_default();
123 let severity = get_str(args, "severity").unwrap_or_default();
124 let cat = category.as_deref().unwrap_or("convention");
125
126 if trigger.is_empty() || resolution.is_empty() {
127 return Ok(ToolOutput {
128 text: "ERROR: trigger and resolution are required for gotcha action"
129 .to_string(),
130 original_tokens: 0,
131 saved_tokens: 0,
132 mode: Some(action),
133 path: None,
134 changed: false,
135 });
136 }
137
138 let mut store = crate::core::gotcha_tracker::GotchaStore::load(&project_root);
139 let msg = match store.report_gotcha(&trigger, &resolution, cat, &severity, &session_id)
140 {
141 Some(gotcha) => {
142 let conf = (gotcha.confidence * 100.0) as u32;
143 let label = gotcha.category.short_label();
144 format!("Gotcha recorded: [{label}] {trigger} (confidence: {conf}%)")
145 }
146 None => {
147 format!("Gotcha noted: {trigger} (evicted by higher-confidence entries)")
148 }
149 };
150 let _ = store.save(&project_root);
151 return Ok(ToolOutput {
152 text: msg,
153 original_tokens: 0,
154 saved_tokens: 0,
155 mode: Some(action),
156 path: None,
157 changed: false,
158 });
159 }
160
161 let result = crate::tools::ctx_knowledge::handle(
162 &project_root,
163 &action,
164 category.as_deref(),
165 key.as_deref(),
166 value.as_deref(),
167 query.as_deref(),
168 &session_id,
169 pattern_type.as_deref(),
170 examples,
171 confidence,
172 mode.as_deref(),
173 );
174
175 Ok(ToolOutput {
176 text: result,
177 original_tokens: 0,
178 saved_tokens: 0,
179 mode: Some(action),
180 path: None,
181 changed: false,
182 })
183 }
184}