zesdex/src/tool/memory/remember.rs

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(),
            scope: None,
            before_snippet: None,
            after_snippet: None,
            provenances: vec![],
        };

        memory.write(&ctx.memory_dir)
            .map_err(|e| anyhow!("failed to write memory '{}': {}", name, e))?;

        Ok(format!("saved memory '{}' ({})", name, kind))
    }
}