@velvetmonkey/flywheel-memory 2.0.47 → 2.0.48

package/dist/index.js

@@ -3594,6 +3594,19 @@ function updateSuppressionList(stateDb2) {
   transaction();
   return updated;
 }
+function suppressEntity(stateDb2, entity) {
+  stateDb2.db.prepare(`
+    INSERT INTO wikilink_suppressions (entity, false_positive_rate, updated_at)
+    VALUES (?, 1.0, datetime('now'))
+    ON CONFLICT(entity) DO UPDATE SET false_positive_rate = 1.0, updated_at = datetime('now')
+  `).run(entity);
+}
+function unsuppressEntity(stateDb2, entity) {
+  const result = stateDb2.db.prepare(
+    "DELETE FROM wikilink_suppressions WHERE entity = ? COLLATE NOCASE"
+  ).run(entity);
+  return result.changes > 0;
+}
 function isSuppressed(stateDb2, entity, folder) {
   const row = stateDb2.db.prepare(
     "SELECT entity FROM wikilink_suppressions WHERE entity = ? COLLATE NOCASE"
@@ -13062,7 +13075,7 @@ Example: vault_add_to_section({ path: "daily/2026-02-15.md", section: "Log", con
             throw new Error(validationResult.blockReason || "Output validation failed");
           }
           let workingReplacement = validationResult.content;
-          let { content: processedReplacement } = maybeApplyWikilinks(workingReplacement, skipWikilinks, notePath);
+          let { content: processedReplacement } = maybeApplyWikilinks(workingReplacement, skipWikilinks, notePath, ctx.content);
           if (suggestOutgoingLinks && !skipWikilinks && processedReplacement.length >= 100) {
             const result = await suggestRelatedLinks(processedReplacement, { maxSuggestions, notePath });
             if (result.suffix) {
@@ -15857,7 +15870,7 @@ function registerWikilinkFeedbackTools(server2, getStateDb) {
       title: "Wikilink Feedback",
       description: 'Report and query wikilink accuracy feedback. Modes: "report" (record feedback), "list" (view recent feedback), "stats" (entity accuracy statistics), "dashboard" (full feedback loop data), "entity_timeline" (score history for an entity), "layer_timeseries" (per-layer contribution over time), "snapshot_diff" (compare two graph snapshots).',
       inputSchema: {
-        mode: z21.enum(["report", "list", "stats", "dashboard", "entity_timeline", "layer_timeseries", "snapshot_diff"]).describe("Operation mode"),
+        mode: z21.enum(["report", "list", "stats", "dashboard", "entity_timeline", "layer_timeseries", "snapshot_diff", "suppress", "unsuppress"]).describe("Operation mode"),
         entity: z21.string().optional().describe("Entity name (required for report and entity_timeline modes, optional filter for list/stats)"),
         note_path: z21.string().optional().describe("Note path where the wikilink appeared (for report mode)"),
         context: z21.string().optional().describe("Surrounding text context (for report mode)"),
@@ -15904,6 +15917,9 @@ function registerWikilinkFeedbackTools(server2, getStateDb) {
             ).run(entity, note_path);
           }
           const suppressionUpdated = updateSuppressionList(stateDb2) > 0;
+          if (!correct) {
+            suppressEntity(stateDb2, entity);
+          }
           result = {
             mode: "report",
             reported: {
@@ -15982,6 +15998,38 @@ function registerWikilinkFeedbackTools(server2, getStateDb) {
           };
           break;
         }
+        case "suppress": {
+          if (!entity) {
+            return {
+              content: [{ type: "text", text: JSON.stringify({ error: "entity is required for suppress mode" }) }],
+              isError: true
+            };
+          }
+          suppressEntity(stateDb2, entity);
+          result = {
+            mode: "suppress",
+            entity,
+            suppressed: true,
+            total_suppressed: getSuppressedCount(stateDb2)
+          };
+          break;
+        }
+        case "unsuppress": {
+          if (!entity) {
+            return {
+              content: [{ type: "text", text: JSON.stringify({ error: "entity is required for unsuppress mode" }) }],
+              isError: true
+            };
+          }
+          const wasRemoved = unsuppressEntity(stateDb2, entity);
+          result = {
+            mode: "unsuppress",
+            entity,
+            was_suppressed: wasRemoved,
+            total_suppressed: getSuppressedCount(stateDb2)
+          };
+          break;
+        }
       }
       return {
         content: [
@@ -15995,8 +16043,142 @@ function registerWikilinkFeedbackTools(server2, getStateDb) {
   );
 }
 
-// src/tools/write/config.ts
+// 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;
+}
+
+// src/tools/write/corrections.ts
+function registerCorrectionTools(server2, getStateDb) {
+  server2.tool(
+    "vault_record_correction",
+    'Record a persistent correction (e.g., "that link is wrong", "undo that"). Survives across sessions.',
+    {
+      correction_type: z22.enum(["wrong_link", "wrong_entity", "wrong_category", "general"]).describe("Type of correction"),
+      description: z22.string().describe("What went wrong and what should be done"),
+      entity: z22.string().optional().describe("Entity name (if correction is about a specific entity)"),
+      note_path: z22.string().optional().describe("Note path (if correction is about a specific note)")
+    },
+    async ({ correction_type, description, entity, note_path }) => {
+      const stateDb2 = getStateDb();
+      if (!stateDb2) {
+        return {
+          content: [{ type: "text", text: JSON.stringify({ error: "StateDb not available" }) }],
+          isError: true
+        };
+      }
+      const correction = recordCorrection(stateDb2, correction_type, description, "user", entity, note_path);
+      return {
+        content: [{
+          type: "text",
+          text: JSON.stringify({
+            recorded: true,
+            correction
+          }, null, 2)
+        }]
+      };
+    }
+  );
+  server2.tool(
+    "vault_list_corrections",
+    "List recorded corrections, optionally filtered by status or entity.",
+    {
+      status: z22.enum(["pending", "applied", "dismissed"]).optional().describe("Filter by status"),
+      entity: z22.string().optional().describe("Filter by entity name"),
+      limit: z22.number().min(1).max(200).default(50).describe("Max entries to return")
+    },
+    async ({ status, entity, limit }) => {
+      const stateDb2 = getStateDb();
+      if (!stateDb2) {
+        return {
+          content: [{ type: "text", text: JSON.stringify({ error: "StateDb not available" }) }],
+          isError: true
+        };
+      }
+      const corrections = listCorrections(stateDb2, status, entity, limit);
+      return {
+        content: [{
+          type: "text",
+          text: JSON.stringify({
+            corrections,
+            count: corrections.length
+          }, null, 2)
+        }]
+      };
+    }
+  );
+  server2.tool(
+    "vault_resolve_correction",
+    "Resolve a correction by marking it as applied or dismissed.",
+    {
+      correction_id: z22.number().describe("ID of the correction to resolve"),
+      status: z22.enum(["applied", "dismissed"]).describe("New status")
+    },
+    async ({ correction_id, status }) => {
+      const stateDb2 = getStateDb();
+      if (!stateDb2) {
+        return {
+          content: [{ type: "text", text: JSON.stringify({ error: "StateDb not available" }) }],
+          isError: true
+        };
+      }
+      const resolved = resolveCorrection(stateDb2, correction_id, status);
+      if (!resolved) {
+        return {
+          content: [{ type: "text", text: JSON.stringify({ error: `Correction ${correction_id} not found` }) }],
+          isError: true
+        };
+      }
+      return {
+        content: [{
+          type: "text",
+          text: JSON.stringify({
+            resolved: true,
+            correction_id,
+            status
+          }, null, 2)
+        }]
+      };
+    }
+  );
+}
+
+// src/tools/write/config.ts
+import { z as z23 } from "zod";
 import { saveFlywheelConfigToDb as saveFlywheelConfigToDb2 } from "@velvetmonkey/vault-core";
 function registerConfigTools(server2, getConfig, setConfig, getStateDb) {
   server2.registerTool(
@@ -16005,9 +16187,9 @@ function registerConfigTools(server2, getConfig, setConfig, getStateDb) {
       title: "Flywheel Config",
       description: 'Read or update Flywheel configuration.\n- "get": Returns the current FlywheelConfig\n- "set": Updates a single config key and returns the updated config\n\nExample: flywheel_config({ mode: "get" })\nExample: flywheel_config({ mode: "set", key: "exclude_analysis_tags", value: ["habit", "daily"] })',
       inputSchema: {
-        mode: z22.enum(["get", "set"]).describe("Operation mode"),
-        key: z22.string().optional().describe("Config key to update (required for set mode)"),
-        value: z22.unknown().optional().describe("New value for the key (required for set mode)")
+        mode: z23.enum(["get", "set"]).describe("Operation mode"),
+        key: z23.string().optional().describe("Config key to update (required for set mode)"),
+        value: z23.unknown().optional().describe("New value for the key (required for set mode)")
       }
     },
     async ({ mode, key, value }) => {
@@ -16045,7 +16227,7 @@ function registerConfigTools(server2, getConfig, setConfig, getStateDb) {
 }
 
 // src/tools/write/enrich.ts
-import { z as z23 } from "zod";
+import { z as z24 } from "zod";
 import * as fs29 from "fs/promises";
 import * as path30 from "path";
 function hasSkipWikilinks(content) {
@@ -16100,9 +16282,9 @@ function registerInitTools(server2, vaultPath2, getStateDb) {
     "vault_init",
     "Initialize vault for Flywheel \u2014 scans legacy notes with zero wikilinks and applies entity links. Safe to re-run (idempotent). Use dry_run (default) to preview.",
     {
-      dry_run: z23.boolean().default(true).describe("If true (default), preview what would be linked without modifying files"),
-      batch_size: z23.number().default(50).describe("Maximum notes to process per invocation (default: 50)"),
-      offset: z23.number().default(0).describe("Skip this many eligible notes (for pagination across invocations)")
+      dry_run: z24.boolean().default(true).describe("If true (default), preview what would be linked without modifying files"),
+      batch_size: z24.number().default(50).describe("Maximum notes to process per invocation (default: 50)"),
+      offset: z24.number().default(0).describe("Skip this many eligible notes (for pagination across invocations)")
     },
     async ({ dry_run, batch_size, offset }) => {
       const startTime = Date.now();
@@ -16197,7 +16379,7 @@ function registerInitTools(server2, vaultPath2, getStateDb) {
 }
 
 // src/tools/read/metrics.ts
-import { z as z24 } from "zod";
+import { z as z25 } from "zod";
 
 // src/core/shared/metrics.ts
 var ALL_METRICS = [
@@ -16363,10 +16545,10 @@ function registerMetricsTools(server2, getIndex, getStateDb) {
       title: "Vault Growth",
       description: 'Track vault growth over time. Modes: "current" (live snapshot), "history" (time series), "trends" (deltas vs N days ago), "index_activity" (rebuild history). Tracks 11 metrics: note_count, link_count, orphan_count, tag_count, entity_count, avg_links_per_note, link_density, connected_ratio, wikilink_accuracy, wikilink_feedback_volume, wikilink_suppressed_count.',
       inputSchema: {
-        mode: z24.enum(["current", "history", "trends", "index_activity"]).describe("Query mode: current snapshot, historical time series, trend analysis, or index rebuild activity"),
-        metric: z24.string().optional().describe('Filter to specific metric (e.g., "note_count"). Omit for all metrics.'),
-        days_back: z24.number().optional().describe("Number of days to look back for history/trends (default: 30)"),
-        limit: z24.number().optional().describe("Number of recent events to return for index_activity mode (default: 20)")
+        mode: z25.enum(["current", "history", "trends", "index_activity"]).describe("Query mode: current snapshot, historical time series, trend analysis, or index rebuild activity"),
+        metric: z25.string().optional().describe('Filter to specific metric (e.g., "note_count"). Omit for all metrics.'),
+        days_back: z25.number().optional().describe("Number of days to look back for history/trends (default: 30)"),
+        limit: z25.number().optional().describe("Number of recent events to return for index_activity mode (default: 20)")
       }
     },
     async ({ mode, metric, days_back, limit: eventLimit }) => {
@@ -16439,7 +16621,7 @@ function registerMetricsTools(server2, getIndex, getStateDb) {
 }
 
 // src/tools/read/activity.ts
-import { z as z25 } from "zod";
+import { z as z26 } from "zod";
 
 // src/core/shared/toolTracking.ts
 function recordToolInvocation(stateDb2, event) {
@@ -16599,10 +16781,10 @@ function registerActivityTools(server2, getStateDb, getSessionId2) {
       title: "Vault Activity",
       description: 'Track tool usage patterns and session activity. Modes:\n- "session": Current session summary (tools called, notes accessed)\n- "sessions": List of recent sessions\n- "note_access": Notes ranked by query frequency\n- "tool_usage": Tool usage patterns (most-used tools, avg duration)',
       inputSchema: {
-        mode: z25.enum(["session", "sessions", "note_access", "tool_usage"]).describe("Activity query mode"),
-        session_id: z25.string().optional().describe("Specific session ID (for session mode, defaults to current)"),
-        days_back: z25.number().optional().describe("Number of days to look back (default: 30)"),
-        limit: z25.number().optional().describe("Maximum results to return (default: 20)")
+        mode: z26.enum(["session", "sessions", "note_access", "tool_usage"]).describe("Activity query mode"),
+        session_id: z26.string().optional().describe("Specific session ID (for session mode, defaults to current)"),
+        days_back: z26.number().optional().describe("Number of days to look back (default: 30)"),
+        limit: z26.number().optional().describe("Maximum results to return (default: 20)")
       }
     },
     async ({ mode, session_id, days_back, limit: resultLimit }) => {
@@ -16669,7 +16851,7 @@ function registerActivityTools(server2, getStateDb, getSessionId2) {
 }
 
 // src/tools/read/similarity.ts
-import { z as z26 } from "zod";
+import { z as z27 } from "zod";
 
 // src/core/read/similarity.ts
 import * as fs30 from "fs";
@@ -16933,9 +17115,9 @@ function registerSimilarityTools(server2, getIndex, getVaultPath, getStateDb) {
       title: "Find Similar Notes",
       description: "Find notes similar to a given note using FTS5 keyword matching. When embeddings have been built (via init_semantic), automatically uses hybrid ranking (BM25 + embedding similarity via Reciprocal Rank Fusion). Use exclude_linked to filter out notes already connected via wikilinks.",
       inputSchema: {
-        path: z26.string().describe('Path to the source note (relative to vault root, e.g. "projects/alpha.md")'),
-        limit: z26.number().optional().describe("Maximum number of similar notes to return (default: 10)"),
-        exclude_linked: z26.boolean().optional().describe("Exclude notes already linked to/from the source note (default: true)")
+        path: z27.string().describe('Path to the source note (relative to vault root, e.g. "projects/alpha.md")'),
+        limit: z27.number().optional().describe("Maximum number of similar notes to return (default: 10)"),
+        exclude_linked: z27.boolean().optional().describe("Exclude notes already linked to/from the source note (default: true)")
       }
     },
     async ({ path: path33, limit, exclude_linked }) => {
@@ -16979,7 +17161,7 @@ function registerSimilarityTools(server2, getIndex, getVaultPath, getStateDb) {
 }
 
 // src/tools/read/semantic.ts
-import { z as z27 } from "zod";
+import { z as z28 } from "zod";
 import { getAllEntitiesFromDb } from "@velvetmonkey/vault-core";
 function registerSemanticTools(server2, getVaultPath, getStateDb) {
   server2.registerTool(
@@ -16988,7 +17170,7 @@ function registerSemanticTools(server2, getVaultPath, getStateDb) {
       title: "Initialize Semantic Search",
       description: "Download the embedding model and build semantic search index for this vault. After running, search and find_similar automatically use hybrid ranking (BM25 + semantic). Run once per vault \u2014 subsequent calls skip already-embedded notes unless force=true.",
       inputSchema: {
-        force: z27.boolean().optional().describe(
+        force: z28.boolean().optional().describe(
           "Rebuild all embeddings even if they already exist (default: false)"
         )
       }
@@ -17068,7 +17250,7 @@ function registerSemanticTools(server2, getVaultPath, getStateDb) {
 
 // src/tools/read/merges.ts
 init_levenshtein();
-import { z as z28 } from "zod";
+import { z as z29 } from "zod";
 import { getAllEntitiesFromDb as getAllEntitiesFromDb2, getDismissedMergePairs, recordMergeDismissal } from "@velvetmonkey/vault-core";
 function normalizeName(name) {
   return name.toLowerCase().replace(/[.\-_]/g, "").replace(/js$/, "").replace(/ts$/, "");
@@ -17078,7 +17260,7 @@ function registerMergeTools2(server2, getStateDb) {
     "suggest_entity_merges",
     "Find potential duplicate entities that could be merged based on name similarity",
     {
-      limit: z28.number().optional().default(50).describe("Maximum number of suggestions to return")
+      limit: z29.number().optional().default(50).describe("Maximum number of suggestions to return")
     },
     async ({ limit }) => {
       const stateDb2 = getStateDb();
@@ -17180,11 +17362,11 @@ function registerMergeTools2(server2, getStateDb) {
     "dismiss_merge_suggestion",
     "Permanently dismiss a merge suggestion so it never reappears",
     {
-      source_path: z28.string().describe("Path of the source entity"),
-      target_path: z28.string().describe("Path of the target entity"),
-      source_name: z28.string().describe("Name of the source entity"),
-      target_name: z28.string().describe("Name of the target entity"),
-      reason: z28.string().describe("Original suggestion reason")
+      source_path: z29.string().describe("Path of the source entity"),
+      target_path: z29.string().describe("Path of the target entity"),
+      source_name: z29.string().describe("Name of the source entity"),
+      target_name: z29.string().describe("Name of the target entity"),
+      reason: z29.string().describe("Original suggestion reason")
     },
     async ({ source_path, target_path, source_name, target_name, reason }) => {
       const stateDb2 = getStateDb();
@@ -17583,6 +17765,7 @@ registerSystemTools2(server, vaultPath);
 registerPolicyTools(server, vaultPath);
 registerTagTools(server, () => vaultIndex, () => vaultPath);
 registerWikilinkFeedbackTools(server, () => stateDb);
+registerCorrectionTools(server, () => stateDb);
 registerInitTools(server, vaultPath, () => stateDb);
 registerConfigTools(
   server,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@velvetmonkey/flywheel-memory",
-  "version": "2.0.47",
+  "version": "2.0.48",
   "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.47",
+    "@velvetmonkey/vault-core": "^2.0.48",
     "better-sqlite3": "^11.0.0",
     "chokidar": "^4.0.0",
     "gray-matter": "^4.0.3",