linksee-memory 0.1.4 → 0.3.0

package/dist/mcp/server.js

@@ -4,7 +4,7 @@
 //        forget / consolidate / read_smart
 import { Server } from '@modelcontextprotocol/sdk/server/index.js';
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
-import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js';
+import { CallToolRequestSchema, ListToolsRequestSchema, ListResourcesRequestSchema, ListResourceTemplatesRequestSchema, ReadResourceRequestSchema, ListPromptsRequestSchema, GetPromptRequestSchema, } from '@modelcontextprotocol/sdk/types.js';
 import { openDb, runMigrations } from '../db/migrate.js';
 import { computeHeat } from '../lib/heat-index.js';
 import { decideForgetting } from '../lib/forgetting.js';
@@ -12,10 +12,24 @@ import { refreshMomentumForEntity } from '../lib/momentum.js';
 import { consolidate as runConsolidate } from '../lib/consolidate.js';
 import { isPastedExternalContent } from '../lib/session-parser.js';
 import { handleReadSmart as handleReadSmartImpl } from './read-smart.js';
-const SERVER_VERSION = '0.1.1';
+import { STATIC_RESOURCES, RESOURCE_TEMPLATES, readResource } from './resources.js';
+import { PROMPTS, getPrompt } from './prompts.js';
+import { fetchRoots, isInsideRoots } from './roots.js';
+import { sampleConsolidation } from './sampling.js';
+import { confirmForget } from './elicitation.js';
+const SERVER_VERSION = '0.3.0';
 const db = openDb();
 runMigrations(db);
-const server = new Server({ name: 'linksee-memory', version: SERVER_VERSION }, { capabilities: { tools: {} } });
+const server = new Server({ name: 'linksee-memory', version: SERVER_VERSION }, {
+    capabilities: {
+        tools: {},
+        resources: { subscribe: false, listChanged: false },
+        prompts: { listChanged: false },
+        // Sampling, Roots, Elicitation are CLIENT capabilities the server consumes.
+        // We don't declare them under "capabilities" — we just call them via server.request
+        // and gracefully degrade when the client doesn't support them.
+    },
+});
 // ============================================================
 // Layer alias map — natural language → canonical layer
 // (agents can say layer="decisions" and we resolve to "learning")
@@ -233,14 +247,14 @@ function toFtsQuery(raw) {
     return tokens.map((t) => `"${t}"`).join(' OR ');
 }
 function runFtsQuery(query, layer, limit) {
-    let sql = `
-    SELECT m.id, m.entity_id, e.name as entity_name, e.kind as entity_kind, e.momentum_score,
-           m.layer, m.content, m.importance, m.created_at, m.last_accessed_at, m.access_count,
-           bm25(memories_fts) as bm25_score
-    FROM memories_fts
-    JOIN memories m ON m.id = memories_fts.rowid
-    JOIN entities e ON e.id = m.entity_id
-    WHERE memories_fts MATCH ?
+    let sql = `
+    SELECT m.id, m.entity_id, e.name as entity_name, e.kind as entity_kind, e.momentum_score,
+           m.layer, m.content, m.importance, m.created_at, m.last_accessed_at, m.access_count,
+           bm25(memories_fts) as bm25_score
+    FROM memories_fts
+    JOIN memories m ON m.id = memories_fts.rowid
+    JOIN entities e ON e.id = m.entity_id
+    WHERE memories_fts MATCH ?
   `;
     const params = [query];
     if (layer) {
@@ -252,13 +266,13 @@ function runFtsQuery(query, layer, limit) {
     return db.prepare(sql).all(...params);
 }
 function runLikeQuery(query, entityName, layer, limit) {
-    let sql = `
-    SELECT m.id, m.entity_id, e.name as entity_name, e.kind as entity_kind, e.momentum_score,
-           m.layer, m.content, m.importance, m.created_at, m.last_accessed_at, m.access_count,
-           0 as bm25_score
-    FROM memories m
-    JOIN entities e ON e.id = m.entity_id
-    WHERE 1=1
+    let sql = `
+    SELECT m.id, m.entity_id, e.name as entity_name, e.kind as entity_kind, e.momentum_score,
+           m.layer, m.content, m.importance, m.created_at, m.last_accessed_at, m.access_count,
+           0 as bm25_score
+    FROM memories m
+    JOIN entities e ON e.id = m.entity_id
+    WHERE 1=1
   `;
     const params = [];
     if (entityName) {
@@ -479,11 +493,14 @@ function handleForget(args) {
             return JSON.stringify({ ok: false, error: `memory_id ${args.memory_id} not found` });
         }
         if (target.protected === 1 || target.importance >= 0.9) {
+            const isLayerProtected = target.protected === 1;
             return JSON.stringify({
                 ok: false,
                 preserved: true,
-                reason: target.protected === 1 ? `${target.layer}-layer is auto-protected` : 'pinned (importance>=0.9)',
-                hint: 'Use update_memory to lower importance below 0.9 first, then forget.',
+                reason: isLayerProtected ? `${target.layer}-layer is auto-protected` : 'pinned (importance>=0.9)',
+                hint: isLayerProtected
+                    ? `${target.layer} memories are permanently protected (the whole point — pain lessons must not be lost). If you truly need to delete, copy its content to another layer via remember() first, then drop the DB row manually via a SQLite client.`
+                    : 'Use update_memory to lower importance below 0.9 first, then forget.',
             });
         }
         const res = db.prepare('DELETE FROM memories WHERE id = ?').run(args.memory_id);
@@ -491,8 +508,8 @@ function handleForget(args) {
     }
     // Auto-sweep — also respect pin (importance >= 0.9) as protection
     const rows = db
-        .prepare(`SELECT id, layer, importance, access_count, last_accessed_at, protected
-              FROM memories
+        .prepare(`SELECT id, layer, importance, access_count, last_accessed_at, protected
+              FROM memories
               WHERE protected = 0 AND importance < 0.9`)
         .all();
     const now = Math.floor(Date.now() / 1000);
@@ -540,17 +557,17 @@ function handleConsolidate(args) {
         // read-only audit: count candidates using the same rules.
         const now = Math.floor(Date.now() / 1000);
         const ageCutoff = now - (args.min_age_days ?? 7) * 86400;
-        const candidates = db.prepare(`
-      SELECT m.entity_id, e.name as entity_name, m.layer, COUNT(*) as c
-      FROM memories m
-      JOIN entities e ON e.id = m.entity_id
-      WHERE m.protected = 0
-        AND m.importance < 0.9
-        AND m.layer IN ('context', 'emotion', 'implementation')
-        AND m.created_at <= ?
-      GROUP BY m.entity_id, m.layer
-      HAVING c >= 2
-      ORDER BY c DESC
+        const candidates = db.prepare(`
+      SELECT m.entity_id, e.name as entity_name, m.layer, COUNT(*) as c
+      FROM memories m
+      JOIN entities e ON e.id = m.entity_id
+      WHERE m.protected = 0
+        AND m.importance < 0.9
+        AND m.layer IN ('context', 'emotion', 'implementation')
+        AND m.created_at <= ?
+      GROUP BY m.entity_id, m.layer
+      HAVING c >= 2
+      ORDER BY c DESC
     `).all(ageCutoff);
         const totalReplaced = candidates.reduce((s, c) => s + c.c, 0);
         return JSON.stringify({
@@ -624,19 +641,19 @@ function handleListEntities(args) {
     const minMemories = Math.max(1, Number(args?.min_memories ?? 1));
     const limit = Math.max(1, Math.min(200, Number(args?.limit ?? 30)));
     const offset = Math.max(0, Number(args?.offset ?? 0));
-    let sql = `
-    SELECT e.id, e.name, e.kind, e.canonical_key, e.momentum_score,
-           e.updated_at, e.created_at,
-           COUNT(m.id) as memory_count,
-           MAX(m.last_accessed_at) as last_memory_access,
-           SUM(CASE WHEN m.layer = 'goal' THEN 1 ELSE 0 END) as goal_count,
-           SUM(CASE WHEN m.layer = 'caveat' THEN 1 ELSE 0 END) as caveat_count,
-           SUM(CASE WHEN m.layer = 'learning' THEN 1 ELSE 0 END) as learning_count,
-           SUM(CASE WHEN m.layer = 'implementation' THEN 1 ELSE 0 END) as impl_count,
-           SUM(CASE WHEN m.importance >= 0.9 THEN 1 ELSE 0 END) as pinned_count
-    FROM entities e
-    LEFT JOIN memories m ON m.entity_id = e.id
-    WHERE 1=1
+    let sql = `
+    SELECT e.id, e.name, e.kind, e.canonical_key, e.momentum_score,
+           e.updated_at, e.created_at,
+           COUNT(m.id) as memory_count,
+           MAX(m.last_accessed_at) as last_memory_access,
+           SUM(CASE WHEN m.layer = 'goal' THEN 1 ELSE 0 END) as goal_count,
+           SUM(CASE WHEN m.layer = 'caveat' THEN 1 ELSE 0 END) as caveat_count,
+           SUM(CASE WHEN m.layer = 'learning' THEN 1 ELSE 0 END) as learning_count,
+           SUM(CASE WHEN m.layer = 'implementation' THEN 1 ELSE 0 END) as impl_count,
+           SUM(CASE WHEN m.importance >= 0.9 THEN 1 ELSE 0 END) as pinned_count
+    FROM entities e
+    LEFT JOIN memories m ON m.entity_id = e.id
+    WHERE 1=1
   `;
     const params = [];
     if (kind) {
@@ -688,29 +705,29 @@ function handleRecallFile(args) {
         return JSON.stringify({ ok: true, count: 0, note: 'No edits found for that path substring.' });
     }
     // Daily breakdown
-    const daily = db.prepare(`
-    SELECT DATE(occurred_at, 'unixepoch') as day, operation, COUNT(*) as edits
-    FROM session_file_edits WHERE file_path LIKE ?
-    GROUP BY day, operation ORDER BY day
+    const daily = db.prepare(`
+    SELECT DATE(occurred_at, 'unixepoch') as day, operation, COUNT(*) as edits
+    FROM session_file_edits WHERE file_path LIKE ?
+    GROUP BY day, operation ORDER BY day
   `).all(`%${sub}%`);
     // Distinct context_snippets (intents) — deduped, ordered by recency
-    const intents = db.prepare(`
-    SELECT DISTINCT context_snippet, MAX(occurred_at) as last_at, COUNT(*) as freq
-    FROM session_file_edits
-    WHERE file_path LIKE ? AND context_snippet IS NOT NULL AND LENGTH(context_snippet) > 20
-    GROUP BY context_snippet
-    ORDER BY last_at DESC
-    LIMIT ?
+    const intents = db.prepare(`
+    SELECT DISTINCT context_snippet, MAX(occurred_at) as last_at, COUNT(*) as freq
+    FROM session_file_edits
+    WHERE file_path LIKE ? AND context_snippet IS NOT NULL AND LENGTH(context_snippet) > 20
+    GROUP BY context_snippet
+    ORDER BY last_at DESC
+    LIMIT ?
   `).all(`%${sub}%`, maxIntents);
     // Linked memories
-    const memories = db.prepare(`
-    SELECT DISTINCT m.id, m.layer, m.content, m.importance, e.name as entity_name
-    FROM session_file_edits sfe
-    JOIN memories m ON m.id = sfe.memory_id
-    JOIN entities e ON e.id = m.entity_id
-    WHERE sfe.file_path LIKE ?
-    ORDER BY m.importance DESC
-    LIMIT 20
+    const memories = db.prepare(`
+    SELECT DISTINCT m.id, m.layer, m.content, m.importance, e.name as entity_name
+    FROM session_file_edits sfe
+    JOIN memories m ON m.id = sfe.memory_id
+    JOIN entities e ON e.id = m.entity_id
+    WHERE sfe.file_path LIKE ?
+    ORDER BY m.importance DESC
+    LIMIT 20
   `).all(`%${sub}%`);
     // Distinct file paths matched (the substring may match multiple files)
     const paths = db.prepare(`SELECT file_path, COUNT(*) as edits FROM session_file_edits WHERE file_path LIKE ? GROUP BY file_path ORDER BY edits DESC`).all(`%${sub}%`);
@@ -743,6 +760,158 @@ function handleReadSmart(args) {
     return handleReadSmartImpl(db, { path: args.path, force: args.force });
 }
 // ============================================================
+// v0.3.0 — five-blocks helpers (sampling / roots / elicitation in handlers)
+// ============================================================
+async function handleRecallFileWithRoots(args) {
+    const baseJson = handleRecallFile(args);
+    if (!args?.scope_to_roots)
+        return baseJson;
+    let parsed;
+    try {
+        parsed = JSON.parse(baseJson);
+    }
+    catch {
+        return baseJson;
+    }
+    if (!parsed?.ok || !Array.isArray(parsed.paths_matched))
+        return baseJson;
+    const roots = await fetchRoots(server);
+    if (roots.length === 0) {
+        parsed.roots_filter = { applied: false, reason: 'client provided no roots' };
+        return JSON.stringify(parsed);
+    }
+    const filtered = parsed.paths_matched.filter((p) => isInsideRoots(p.file_path, roots));
+    parsed.roots_filter = { applied: true, root_count: roots.length, before: parsed.paths_matched.length, after: filtered.length };
+    parsed.paths_matched = filtered;
+    return JSON.stringify(parsed);
+}
+async function handleConsolidateWithSampling(args) {
+    // Sampling only applies on a real run (not dry-run) and only when explicitly opted in.
+    if (!args?.use_llm || args?.dry_run)
+        return handleConsolidate(args);
+    // 1. Snapshot all candidate memories BEFORE consolidate runs so we can recover
+    //    their content (the originals are deleted by consolidate).
+    const ageCutoff = Math.floor(Date.now() / 1000) - (typeof args?.min_age_days === 'number' ? args.min_age_days : 7) * 86400;
+    const snapshot = new Map();
+    const candidateRows = db
+        .prepare(`SELECT m.id, m.content, e.name as entity_name
+       FROM memories m JOIN entities e ON e.id = m.entity_id
+       WHERE m.protected = 0
+         AND m.layer IN ('context','emotion','implementation')
+         AND m.created_at <= ?`)
+        .all(ageCutoff);
+    for (const r of candidateRows)
+        snapshot.set(r.id, r);
+    // 2. Run the normal heuristic consolidate (creates learning entries, deletes source).
+    const baseJson = handleConsolidate(args);
+    let parsed;
+    try {
+        parsed = JSON.parse(baseJson);
+    }
+    catch {
+        return baseJson;
+    }
+    if (!parsed?.ok || !Array.isArray(parsed.learningIdsCreated)) {
+        parsed = parsed ?? {};
+        parsed.sampling = { applied: false, reason: 'consolidate returned no learning entries' };
+        return JSON.stringify(parsed);
+    }
+    // 3. For each new learning entry, look up its replaced_ids from the audit table,
+    //    gather source contents from the snapshot, and request a sampled summary.
+    let upgraded = 0;
+    let declined = 0;
+    const declineReasons = [];
+    for (const learningId of parsed.learningIdsCreated) {
+        const audit = db.prepare('SELECT replaced_ids FROM consolidations WHERE learning_id = ?').get(learningId);
+        if (!audit) {
+            declined++;
+            continue;
+        }
+        let replaced;
+        try {
+            replaced = JSON.parse(audit.replaced_ids);
+        }
+        catch {
+            declined++;
+            continue;
+        }
+        if (!Array.isArray(replaced) || replaced.length < 2) {
+            declined++;
+            continue;
+        }
+        const sources = replaced
+            .map((id) => snapshot.get(id))
+            .filter((s) => Boolean(s));
+        if (sources.length < 2) {
+            declined++;
+            continue;
+        }
+        const entityName = sources[0]?.entity_name ?? '<entity>';
+        const result = await sampleConsolidation(server, sources.map((s) => s.content), entityName);
+        if (result.ok && result.text) {
+            db.prepare('UPDATE memories SET content = ? WHERE id = ?').run(result.text.trim(), learningId);
+            upgraded++;
+        }
+        else {
+            declined++;
+            if (result.reason && declineReasons.length < 3)
+                declineReasons.push(result.reason);
+        }
+    }
+    parsed.sampling = {
+        applied: true,
+        upgraded,
+        declined,
+        ...(declineReasons.length ? { decline_reasons: declineReasons } : {}),
+    };
+    return JSON.stringify(parsed);
+}
+async function handleForgetInteractive(args) {
+    if (!args?.interactive || !args?.memory_id)
+        return handleForget(args);
+    const id = Number(args.memory_id);
+    const row = db
+        .prepare(`SELECT m.id, m.layer, m.content, m.importance, e.name as entity FROM memories m JOIN entities e ON e.id = m.entity_id WHERE m.id = ?`)
+        .get(id);
+    if (!row)
+        return JSON.stringify({ ok: false, error: `memory ${id} not found` });
+    const ok = await confirmForget(server, {
+        id: row.id,
+        entity: row.entity,
+        layer: row.layer,
+        importance: row.importance,
+        preview: row.content,
+    });
+    if (!ok)
+        return JSON.stringify({ ok: false, declined: true, memory_id: id, reason: 'user declined elicitation' });
+    return handleForget({ memory_id: id });
+}
+// Append new optional flags to existing tools (backward-compatible).
+const RECALL_FILE_TOOL = TOOLS.find((t) => t.name === 'recall_file');
+if (RECALL_FILE_TOOL && RECALL_FILE_TOOL.inputSchema.properties) {
+    RECALL_FILE_TOOL.inputSchema.properties.scope_to_roots = {
+        type: 'boolean',
+        default: false,
+        description: 'If true, filter results to files inside the client-provided roots (Roots block). Skip silently when client provides no roots.',
+    };
+}
+const CONSOLIDATE_TOOL = TOOLS.find((t) => t.name === 'consolidate');
+if (CONSOLIDATE_TOOL && CONSOLIDATE_TOOL.inputSchema.properties) {
+    CONSOLIDATE_TOOL.inputSchema.properties.use_llm = {
+        type: 'boolean',
+        default: false,
+        description: 'If true, request the client LLM (Sampling block) to write the consolidated summary instead of the heuristic. Falls back gracefully if the client refuses.',
+    };
+}
+const FORGET_TOOL = TOOLS.find((t) => t.name === 'forget');
+if (FORGET_TOOL && FORGET_TOOL.inputSchema.properties) {
+    FORGET_TOOL.inputSchema.properties.interactive = {
+        type: 'boolean',
+        default: false,
+        description: 'If true, ask the user to confirm via Elicitation before deleting. Only applies when memory_id is set.',
+    };
+}
+// ============================================================
 // MCP wiring
 // ============================================================
 server.setRequestHandler(ListToolsRequestSchema, async () => ({ tools: TOOLS }));
@@ -764,13 +933,13 @@ server.setRequestHandler(CallToolRequestSchema, async (req) => {
                 text = handleListEntities(args);
                 break;
             case 'forget':
-                text = handleForget(args);
+                text = await handleForgetInteractive(args);
                 break;
             case 'consolidate':
-                text = handleConsolidate(args);
+                text = await handleConsolidateWithSampling(args);
                 break;
             case 'recall_file':
-                text = handleRecallFile(args);
+                text = await handleRecallFileWithRoots(args);
                 break;
             case 'read_smart':
                 text = handleReadSmart(args);
@@ -786,6 +955,24 @@ server.setRequestHandler(CallToolRequestSchema, async (req) => {
         };
     }
 });
+// ============================================================
+// Resources block
+// ============================================================
+server.setRequestHandler(ListResourcesRequestSchema, async () => ({ resources: STATIC_RESOURCES }));
+server.setRequestHandler(ListResourceTemplatesRequestSchema, async () => ({ resourceTemplates: RESOURCE_TEMPLATES }));
+server.setRequestHandler(ReadResourceRequestSchema, async (req) => {
+    const { uri } = req.params;
+    const result = readResource(db, uri);
+    return { contents: [result] };
+});
+// ============================================================
+// Prompts block
+// ============================================================
+server.setRequestHandler(ListPromptsRequestSchema, async () => ({ prompts: PROMPTS }));
+server.setRequestHandler(GetPromptRequestSchema, async (req) => {
+    const { name, arguments: promptArgs } = req.params;
+    return getPrompt(name, promptArgs);
+});
 const transport = new StdioServerTransport();
 await server.connect(transport);
 process.stderr.write(`[linksee-memory] MCP server ready on stdio (v${SERVER_VERSION})\n`);