@velvetmonkey/flywheel-memory 2.0.53 → 2.0.55

package/dist/index.js

@@ -3832,9 +3832,14 @@ function trackWikilinkApplications(stateDb2, notePath, entities) {
       applied_at = datetime('now'),
       status = 'applied'
   `);
+  const lookupCanonical = stateDb2.db.prepare(
+    `SELECT name FROM entities WHERE LOWER(name) = LOWER(?) LIMIT 1`
+  );
   const transaction = stateDb2.db.transaction(() => {
     for (const entity of entities) {
-      upsert.run(entity, notePath);
+      const row = lookupCanonical.get(entity);
+      const canonicalName = row?.name ?? entity;
+      upsert.run(canonicalName, notePath);
     }
   });
   transaction();
@@ -4192,6 +4197,75 @@ function getExtendedDashboardData(stateDb2) {
   };
 }
 
+// src/core/write/corrections.ts
+function recordCorrection(stateDb2, type, description, source = "user", entity, notePath) {
+  const result = stateDb2.db.prepare(`
+    INSERT INTO corrections (entity, note_path, correction_type, description, source)
+    VALUES (?, ?, ?, ?, ?)
+  `).run(entity ?? null, notePath ?? null, type, description, source);
+  return stateDb2.db.prepare(
+    "SELECT * FROM corrections WHERE id = ?"
+  ).get(result.lastInsertRowid);
+}
+function listCorrections(stateDb2, status, entity, limit = 50) {
+  const conditions = [];
+  const params = [];
+  if (status) {
+    conditions.push("status = ?");
+    params.push(status);
+  }
+  if (entity) {
+    conditions.push("entity = ? COLLATE NOCASE");
+    params.push(entity);
+  }
+  const where = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
+  params.push(limit);
+  return stateDb2.db.prepare(
+    `SELECT * FROM corrections ${where} ORDER BY created_at DESC LIMIT ?`
+  ).all(...params);
+}
+function resolveCorrection(stateDb2, id, newStatus) {
+  const result = stateDb2.db.prepare(`
+    UPDATE corrections
+    SET status = ?, resolved_at = datetime('now')
+    WHERE id = ?
+  `).run(newStatus, id);
+  return result.changes > 0;
+}
+function getCorrectedEntityNotePairs(stateDb2) {
+  const rows = stateDb2.db.prepare(`
+    SELECT entity, note_path FROM corrections
+    WHERE correction_type = 'wrong_link'
+      AND entity IS NOT NULL AND note_path IS NOT NULL
+      AND status IN ('pending', 'applied')
+  `).all();
+  const map = /* @__PURE__ */ new Map();
+  for (const row of rows) {
+    const key = row.entity.toLowerCase();
+    if (!map.has(key)) map.set(key, /* @__PURE__ */ new Set());
+    map.get(key).add(row.note_path);
+  }
+  return map;
+}
+function processPendingCorrections(stateDb2) {
+  const pending = listCorrections(stateDb2, "pending");
+  let processed = 0;
+  for (const correction of pending) {
+    if (!correction.entity) {
+      resolveCorrection(stateDb2, correction.id, "dismissed");
+      continue;
+    }
+    if (correction.correction_type === "wrong_link") {
+      recordFeedback(stateDb2, correction.entity, "correction:wrong_link", correction.note_path || "", false, 1);
+    } else if (correction.correction_type === "wrong_category") {
+      recordFeedback(stateDb2, correction.entity, "correction:wrong_category", "", false, 0.5);
+    }
+    resolveCorrection(stateDb2, correction.id, "applied");
+    processed++;
+  }
+  return processed;
+}
+
 // src/core/write/git.ts
 import { simpleGit, CheckRepoActions } from "simple-git";
 import path8 from "path";
@@ -5427,13 +5501,13 @@ var EXCLUDED_FOLDERS2 = /* @__PURE__ */ new Set([
   ".claude",
   ".git"
 ]);
-function noteContainsEntity(content, entityName) {
+function noteContainsEntity(content, entityName, contentTokens) {
   const entityTokens = tokenize(entityName);
   if (entityTokens.length === 0) return false;
-  const contentTokens = new Set(tokenize(content));
+  const tokens = contentTokens ?? new Set(tokenize(content));
   let matchCount = 0;
   for (const token of entityTokens) {
-    if (contentTokens.has(token)) {
+    if (tokens.has(token)) {
       matchCount++;
     }
   }
@@ -5478,9 +5552,10 @@ async function mineCooccurrences(vaultPath2, entities, options = {}) {
     try {
       const content = await readFile2(file.path, "utf-8");
       notesScanned++;
+      const contentTokens = new Set(tokenize(content));
       const mentionedEntities = [];
       for (const entity of validEntities) {
-        if (noteContainsEntity(content, entity)) {
+        if (noteContainsEntity(content, entity, contentTokens)) {
           mentionedEntities.push(entity);
         }
       }
@@ -5986,6 +6061,14 @@ function processWikilinks(content, notePath, existingContent) {
       return !isSuppressed(moduleStateDb5, name, folder);
     });
   }
+  if (moduleStateDb5 && notePath) {
+    const correctedPairs = getCorrectedEntityNotePairs(moduleStateDb5);
+    entities = entities.filter((e) => {
+      const name = getEntityName2(e).toLowerCase();
+      const paths = correctedPairs.get(name);
+      return !paths || !paths.has(notePath);
+    });
+  }
   const sortedEntities = sortEntitiesByPriority(entities, notePath);
   const resolved = resolveAliasWikilinks(content, sortedEntities, {
     caseInsensitive: true
@@ -6470,6 +6553,7 @@ async function suggestRelatedLinks(content, options = {}) {
   const noteFolder = notePath ? notePath.split("/")[0] : void 0;
   const feedbackBoosts = moduleStateDb5 ? getAllFeedbackBoosts(moduleStateDb5, noteFolder) : /* @__PURE__ */ new Map();
   const suppressionPenalties = moduleStateDb5 ? getAllSuppressionPenalties(moduleStateDb5) : /* @__PURE__ */ new Map();
+  const correctedPairs = moduleStateDb5 ? getCorrectedEntityNotePairs(moduleStateDb5) : /* @__PURE__ */ new Map();
   const edgeWeightMap = moduleStateDb5 ? getEntityEdgeWeightMap(moduleStateDb5) : /* @__PURE__ */ new Map();
   const scoredEntities = [];
   const directlyMatchedEntities = /* @__PURE__ */ new Set();
@@ -6486,6 +6570,10 @@ async function suggestRelatedLinks(content, options = {}) {
     if (linkedEntities.has(entityName.toLowerCase())) {
       continue;
     }
+    if (notePath && correctedPairs.has(entityName.toLowerCase())) {
+      const paths = correctedPairs.get(entityName.toLowerCase());
+      if (paths.has(notePath)) continue;
+    }
     const contentScore = disabled.has("exact_match") && disabled.has("stem_match") ? 0 : scoreEntity(entity, contentTokens, contentStems, config, cooccurrenceIndex);
     let score = contentScore;
     if (contentScore > 0) {
@@ -8804,6 +8892,20 @@ function purgeOldIndexEvents(stateDb2, retentionDays = 90) {
   ).run(cutoff);
   return result.changes;
 }
+function purgeOldSuggestionEvents(stateDb2, retentionDays = 30) {
+  const cutoff = Date.now() - retentionDays * 24 * 60 * 60 * 1e3;
+  const result = stateDb2.db.prepare(
+    "DELETE FROM suggestion_events WHERE timestamp < ?"
+  ).run(cutoff);
+  return result.changes;
+}
+function purgeOldNoteLinkHistory(stateDb2, retentionDays = 90) {
+  const cutoff = new Date(Date.now() - retentionDays * 24 * 60 * 60 * 1e3).toISOString();
+  const result = stateDb2.db.prepare(
+    "DELETE FROM note_link_history WHERE last_positive_at < ?"
+  ).run(cutoff);
+  return result.changes;
+}
 
 // src/core/read/sweep.ts
 var DEFAULT_SWEEP_INTERVAL_MS = 5 * 60 * 1e3;
@@ -16352,62 +16454,6 @@ function registerWikilinkFeedbackTools(server2, getStateDb) {
 
 // src/tools/write/corrections.ts
 import { z as z22 } from "zod";
-
-// src/core/write/corrections.ts
-function recordCorrection(stateDb2, type, description, source = "user", entity, notePath) {
-  const result = stateDb2.db.prepare(`
-    INSERT INTO corrections (entity, note_path, correction_type, description, source)
-    VALUES (?, ?, ?, ?, ?)
-  `).run(entity ?? null, notePath ?? null, type, description, source);
-  return stateDb2.db.prepare(
-    "SELECT * FROM corrections WHERE id = ?"
-  ).get(result.lastInsertRowid);
-}
-function listCorrections(stateDb2, status, entity, limit = 50) {
-  const conditions = [];
-  const params = [];
-  if (status) {
-    conditions.push("status = ?");
-    params.push(status);
-  }
-  if (entity) {
-    conditions.push("entity = ? COLLATE NOCASE");
-    params.push(entity);
-  }
-  const where = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
-  params.push(limit);
-  return stateDb2.db.prepare(
-    `SELECT * FROM corrections ${where} ORDER BY created_at DESC LIMIT ?`
-  ).all(...params);
-}
-function resolveCorrection(stateDb2, id, newStatus) {
-  const result = stateDb2.db.prepare(`
-    UPDATE corrections
-    SET status = ?, resolved_at = datetime('now')
-    WHERE id = ?
-  `).run(newStatus, id);
-  return result.changes > 0;
-}
-function processPendingCorrections(stateDb2) {
-  const pending = listCorrections(stateDb2, "pending");
-  let processed = 0;
-  for (const correction of pending) {
-    if (!correction.entity) {
-      resolveCorrection(stateDb2, correction.id, "dismissed");
-      continue;
-    }
-    if (correction.correction_type === "wrong_link") {
-      recordFeedback(stateDb2, correction.entity, "correction:wrong_link", correction.note_path || "", false, 1);
-    } else if (correction.correction_type === "wrong_category") {
-      recordFeedback(stateDb2, correction.entity, "correction:wrong_category", "", false, 0.5);
-    }
-    resolveCorrection(stateDb2, correction.id, "applied");
-    processed++;
-  }
-  return processed;
-}
-
-// src/tools/write/corrections.ts
 function registerCorrectionTools(server2, getStateDb) {
   server2.tool(
     "vault_record_correction",
@@ -16724,6 +16770,60 @@ function getRecentSessionSummaries(stateDb2, limit = 5, agent_id) {
     "SELECT * FROM session_summaries ORDER BY ended_at DESC LIMIT ?"
   ).all(limit);
 }
+function sweepExpiredMemories(stateDb2) {
+  const now = Date.now();
+  const msPerDay = 864e5;
+  const expired = stateDb2.db.prepare(`
+    SELECT key FROM memories
+    WHERE ttl_days IS NOT NULL
+    AND superseded_by IS NULL
+    AND (created_at + (ttl_days * ?)) < ?
+  `).all(msPerDay, now);
+  for (const { key } of expired) {
+    removeGraphSignals(stateDb2, key);
+  }
+  const result = stateDb2.db.prepare(`
+    DELETE FROM memories
+    WHERE ttl_days IS NOT NULL
+    AND superseded_by IS NULL
+    AND (created_at + (ttl_days * ?)) < ?
+  `).run(msPerDay, now);
+  return result.changes;
+}
+function decayMemoryConfidence(stateDb2) {
+  const now = Date.now();
+  const msPerDay = 864e5;
+  const halfLifeDays = 30;
+  const lambda = Math.LN2 / (halfLifeDays * msPerDay);
+  const staleThreshold = now - 7 * msPerDay;
+  const staleMemories = stateDb2.db.prepare(`
+    SELECT id, accessed_at, confidence FROM memories
+    WHERE accessed_at < ? AND superseded_by IS NULL AND confidence > 0.1
+  `).all(staleThreshold);
+  let updated = 0;
+  const updateStmt = stateDb2.db.prepare(
+    "UPDATE memories SET confidence = ? WHERE id = ?"
+  );
+  for (const mem of staleMemories) {
+    const ageDays = (now - mem.accessed_at) / msPerDay;
+    const decayFactor = Math.exp(-lambda * ageDays * msPerDay);
+    const newConfidence = Math.max(0.1, mem.confidence * decayFactor);
+    if (Math.abs(newConfidence - mem.confidence) > 0.01) {
+      updateStmt.run(newConfidence, mem.id);
+      updated++;
+    }
+  }
+  return updated;
+}
+function pruneSupersededMemories(stateDb2, retentionDays = 90) {
+  const cutoff = Date.now() - retentionDays * 864e5;
+  const result = stateDb2.db.prepare(`
+    DELETE FROM memories
+    WHERE superseded_by IS NOT NULL
+    AND updated_at < ?
+  `).run(cutoff);
+  return result.changes;
+}
 function findContradictions2(stateDb2, entity) {
   const conditions = ["superseded_by IS NULL"];
   const params = [];
@@ -19207,6 +19307,11 @@ async function runPostIndexWork(index) {
       purgeOldMetrics(stateDb, 90);
       purgeOldIndexEvents(stateDb, 90);
       purgeOldInvocations(stateDb, 90);
+      purgeOldSuggestionEvents(stateDb, 30);
+      purgeOldNoteLinkHistory(stateDb, 90);
+      sweepExpiredMemories(stateDb);
+      decayMemoryConfidence(stateDb);
+      pruneSupersededMemories(stateDb, 90);
       serverLog("server", "Growth metrics recorded");
     } catch (err) {
       serverLog("server", `Failed to record metrics: ${err instanceof Error ? err.message : err}`, "error");
@@ -19801,6 +19906,9 @@ async function runPostIndexWork(index) {
         tracker.end({ tracked: trackedLinks, mentions: mentionResults });
         serverLog("watcher", `Wikilink check: ${trackedLinks.reduce((s, t) => s + t.entities.length, 0)} tracked links in ${trackedLinks.length} files, ${mentionResults.reduce((s, m) => s + m.entities.length, 0)} unwikified mentions`);
         tracker.start("implicit_feedback", { files: filteredEvents.length });
+        const deletedFiles = new Set(
+          filteredEvents.filter((e) => e.type === "delete").map((e) => e.path)
+        );
         const preSuppressed = stateDb ? new Set(getAllSuppressionPenalties(stateDb).keys()) : /* @__PURE__ */ new Set();
         const feedbackResults = [];
         if (stateDb) {
@@ -19816,6 +19924,7 @@ async function runPostIndexWork(index) {
         }
         if (stateDb && linkDiffs.length > 0) {
           for (const diff of linkDiffs) {
+            if (deletedFiles.has(diff.file)) continue;
             for (const target of diff.removed) {
               if (feedbackResults.some((r) => r.entity === target && r.file === diff.file)) continue;
               const entity = entitiesAfter.find(
@@ -19834,6 +19943,7 @@ async function runPostIndexWork(index) {
             `SELECT 1 FROM wikilink_applications WHERE LOWER(entity) = LOWER(?) AND note_path = ? AND status = 'applied'`
           );
           for (const diff of linkDiffs) {
+            if (deletedFiles.has(diff.file)) continue;
             for (const target of diff.added) {
               if (checkApplication.get(target, diff.file)) continue;
               const entity = entitiesAfter.find(

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@velvetmonkey/flywheel-memory",
-  "version": "2.0.53",
+  "version": "2.0.55",
   "description": "MCP server that gives Claude full read/write access to your Obsidian vault. Select from 42 tools for search, backlinks, graph queries, mutations, and hybrid semantic search.",
   "type": "module",
   "main": "dist/index.js",
@@ -52,7 +52,7 @@
   },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.25.1",
-    "@velvetmonkey/vault-core": "^2.0.53",
+    "@velvetmonkey/vault-core": "^2.0.55",
     "better-sqlite3": "^11.0.0",
     "chokidar": "^4.0.0",
     "gray-matter": "^4.0.3",