feat: add memory management tools and utility commands

2026-07-11 20:44:15 +07:00
parent 08490532d2
commit fe82840c03
13 changed files with 535 additions and 0 deletions
@@ -0,0 +1,79 @@
+use serde_json::{json, Value};
+use anyhow::{Result, anyhow};
+use super::super::Tool;
+use super::super::ToolCtx;
+use crate::model::memory::Memory;
+
+pub struct Remember;
+
+impl Tool for Remember {
+    fn name(&self) -> &'static str {
+        "remember"
+    }
+
+    fn description(&self) -> &'static str {
+        "Save a piece of information to persistent project memory. Memory entries are injected into future conversations via the system prompt, so use this to record conventions, preferences, and important context."
+    }
+
+    fn parameters(&self) -> Value {
+        json!({
+            "type": "object",
+            "properties": {
+                "name": {
+                    "type": "string",
+                    "description": "Short unique name for the memory (kebab-case, e.g. 'testing-conventions')"
+                },
+                "description": {
+                    "type": "string",
+                    "description": "One-line summary shown in the memory index"
+                },
+                "content": {
+                    "type": "string",
+                    "description": "The memory content body"
+                },
+                "kind": {
+                    "type": "string",
+                    "description": "Type of memory: 'project', 'reference', 'lesson', or 'feedback'",
+                    "enum": ["project", "reference", "lesson", "feedback"]
+                }
+            },
+            "required": ["name", "description", "content", "kind"]
+        })
+    }
+
+    fn run(&self, ctx: &ToolCtx, args: &Value) -> Result<String> {
+        let name = args.get("name")
+            .and_then(|v| v.as_str())
+            .ok_or_else(|| anyhow!("missing required argument: name"))?;
+        let description = args.get("description")
+            .and_then(|v| v.as_str())
+            .ok_or_else(|| anyhow!("missing required argument: description"))?;
+        let content = args.get("content")
+            .and_then(|v| v.as_str())
+            .ok_or_else(|| anyhow!("missing required argument: content"))?;
+        let kind = args.get("kind")
+            .and_then(|v| v.as_str())
+            .ok_or_else(|| anyhow!("missing required argument: kind"))?;
+
+        if Memory::slugify(name).is_none() {
+            anyhow::bail!("invalid memory name: must produce a valid slug (alphanumeric + hyphens, 1-80 chars)");
+        }
+
+        let now = chrono::Utc::now().timestamp_millis();
+        let memory = Memory {
+            name: name.to_string(),
+            description: description.to_string(),
+            content: content.to_string(),
+            kind: kind.to_string(),
+            created_at: now,
+            updated_at: now,
+            outcome: None,
+            lifecycle: "new".to_string(),
+        };
+
+        memory.write(&ctx.memory_dir)
+            .map_err(|e| anyhow!("failed to write memory '{}': {}", name, e))?;
+
+        Ok(format!("saved memory '{}' ({})", name, kind))
+    }
+}