lean_ctx/tools/registered/
ctx_knowledge.rs

1use 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 session = session_handle.blocking_read();
102            let sid = session.id.clone();
103            let root = session.project_root.clone().unwrap_or_else(|| {
104                std::env::current_dir().map_or_else(
105                    |_| "unknown".to_string(),
106                    |p| p.to_string_lossy().to_string(),
107                )
108            });
109            (sid, root)
110        };
111
112        if action == "gotcha" {
113            let trigger = get_str(args, "trigger").unwrap_or_default();
114            let resolution = get_str(args, "resolution").unwrap_or_default();
115            let severity = get_str(args, "severity").unwrap_or_default();
116            let cat = category.as_deref().unwrap_or("convention");
117
118            if trigger.is_empty() || resolution.is_empty() {
119                return Ok(ToolOutput {
120                    text: "ERROR: trigger and resolution are required for gotcha action"
121                        .to_string(),
122                    original_tokens: 0,
123                    saved_tokens: 0,
124                    mode: Some(action),
125                    path: None,
126                    changed: false,
127                });
128            }
129
130            let mut store = crate::core::gotcha_tracker::GotchaStore::load(&project_root);
131            let msg = match store.report_gotcha(&trigger, &resolution, cat, &severity, &session_id)
132            {
133                Some(gotcha) => {
134                    let conf = (gotcha.confidence * 100.0) as u32;
135                    let label = gotcha.category.short_label();
136                    format!("Gotcha recorded: [{label}] {trigger} (confidence: {conf}%)")
137                }
138                None => {
139                    format!("Gotcha noted: {trigger} (evicted by higher-confidence entries)")
140                }
141            };
142            let _ = store.save(&project_root);
143            return Ok(ToolOutput {
144                text: msg,
145                original_tokens: 0,
146                saved_tokens: 0,
147                mode: Some(action),
148                path: None,
149                changed: false,
150            });
151        }
152
153        let result = crate::tools::ctx_knowledge::handle(
154            &project_root,
155            &action,
156            category.as_deref(),
157            key.as_deref(),
158            value.as_deref(),
159            query.as_deref(),
160            &session_id,
161            pattern_type.as_deref(),
162            examples,
163            confidence,
164            mode.as_deref(),
165        );
166
167        Ok(ToolOutput {
168            text: result,
169            original_tokens: 0,
170            saved_tokens: 0,
171            mode: Some(action),
172            path: None,
173            changed: false,
174        })
175    }
176}
lean_ctx/tools/registered/ctx_knowledge.rs

lean_ctx/tools/registered/
ctx_knowledge.rs
