@velvetmonkey/flywheel-memory 2.0.52 → 2.0.54

package/dist/index.js

@@ -4192,6 +4192,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";
@@ -5986,6 +6055,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 +6547,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 +6564,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) {
@@ -16352,62 +16434,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",

package/package.json

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