@velvetmonkey/flywheel-memory 2.0.148 → 2.0.150

package/dist/index.js

@@ -642,9 +642,6 @@ function loadNoteEmbeddingsForPaths(paths) {
   }
   return result;
 }
-function getEntityEmbedding(entityName) {
-  return getEmbMap().get(entityName) ?? null;
-}
 function getEntityEmbeddingsCount() {
   const db4 = getDb();
   if (!db4) return 0;
@@ -3396,7 +3393,8 @@ async function rebuildIndex(vaultPath2) {
   console.error(`[Flywheel] Scanning vault for entities...`);
   const startTime = Date.now();
   entityIndex = await scanVaultEntities(vaultPath2, {
-    excludeFolders: DEFAULT_EXCLUDE_FOLDERS
+    excludeFolders: DEFAULT_EXCLUDE_FOLDERS,
+    customCategories: getConfig()?.custom_categories
   });
   indexReady = true;
   lastLoadedAt = Date.now();
@@ -4921,7 +4919,7 @@ function formatContent(content, format) {
       const lines = trimmed.split("\n");
       return lines.map((line, i) => {
         if (i === 0) return `- ${line}`;
-        if (line === "") return "";
+        if (line === "") return "  ";
         return `  ${line}`;
       }).join("\n");
     }
@@ -4932,7 +4930,7 @@ function formatContent(content, format) {
       const lines = trimmed.split("\n");
       return lines.map((line, i) => {
         if (i === 0) return `- [ ] ${line}`;
-        if (line === "") return "";
+        if (line === "") return "      ";
         return `      ${line}`;
       }).join("\n");
     }
@@ -4943,7 +4941,7 @@ function formatContent(content, format) {
       const lines = trimmed.split("\n");
       return lines.map((line, i) => {
         if (i === 0) return `1. ${line}`;
-        if (line === "") return "";
+        if (line === "") return "   ";
         return `   ${line}`;
       }).join("\n");
     }
@@ -4959,7 +4957,7 @@ function formatContent(content, format) {
       const indent = "  ";
       return lines.map((line, i) => {
         if (i === 0) return `${prefix}${line}`;
-        if (line === "") return "";
+        if (line === "") return indent;
         return `${indent}${line}`;
       }).join("\n");
     }
@@ -5036,7 +5034,7 @@ function insertInSection(content, section, newContent, position, options) {
       if (indent) {
         const contentLines = formattedContent.split("\n");
         const indentedContent = contentLines.map((line) => {
-          if (line === "") return line;
+          if (line === "") return indent || line;
           return indent + line;
         }).join("\n");
         lines.splice(section.contentStartLine, 0, indentedContent);
@@ -5059,7 +5057,7 @@ function insertInSection(content, section, newContent, position, options) {
         const indent = detectSectionBaseIndentation(lines, section.contentStartLine, section.endLine);
         const contentLines = formattedContent.split("\n");
         const indentedContent = contentLines.map((line) => {
-          if (line === "") return line;
+          if (line === "") return indent || line;
           return indent + line;
         }).join("\n");
         lines[lastContentLineIdx] = indentedContent;
@@ -5082,7 +5080,7 @@ function insertInSection(content, section, newContent, position, options) {
         const indent = detectSectionBaseIndentation(lines, section.contentStartLine, section.endLine);
         const contentLines = formattedContent.split("\n");
         const indentedContent = contentLines.map((line) => {
-          if (line === "") return line;
+          if (line === "") return indent || line;
           return indent + line;
         }).join("\n");
         lines.splice(insertLine, 0, indentedContent);
@@ -10718,9 +10716,8 @@ var ALL_CATEGORIES = [
 ];
 var PRESETS = {
   // Presets
-  default: ["search", "read", "write", "tasks"],
-  agent: ["search", "read", "write", "memory"],
-  full: ALL_CATEGORIES.filter((c) => c !== "memory"),
+  default: ["search", "read", "write", "tasks", "memory"],
+  full: [...ALL_CATEGORIES],
   // Composable bundles (one per category)
   graph: ["graph"],
   schema: ["schema"],
@@ -10734,6 +10731,8 @@ var PRESETS = {
 };
 var DEFAULT_PRESET = "default";
 var DEPRECATED_ALIASES = {
+  agent: "default",
+  // agent merged into default — memory now included
   minimal: "default",
   writer: "default",
   // writer was default+tasks, now default includes tasks
@@ -10848,9 +10847,8 @@ var TOOL_CATEGORY = {
   tasks: "tasks",
   vault_toggle_task: "tasks",
   vault_add_task: "tasks",
-  // memory (3 tools) -- agent working memory
+  // memory (2 tools) -- session memory
   memory: "memory",
-  recall: "memory",
   brief: "memory",
   // note-ops (4 tools) -- file management
   vault_delete_note: "note-ops",
@@ -10889,10 +10887,12 @@ function generateInstructions(categories, registry) {
   parts.push(`Flywheel provides tools to search, read, and write an Obsidian vault's knowledge graph.
 
 Tool selection:
-  1. "search" is the primary tool. Each result includes: frontmatter, tags, aliases,
-     backlinks (with line numbers), outlinks (with line numbers and existence check),
-     headings, content snippet or preview, entity category, hub score, and timestamps.
-     This is usually enough to answer without reading any files.
+  1. "search" is the primary tool. One call searches notes, entities, and memories.
+     Each result carries: type (note/entity/memory), frontmatter, tags, aliases,
+     backlinks (ranked by edge weight \xD7 recency), outlinks (existence-checked),
+     section provenance, extracted dates, entity bridges, confidence scores,
+     content snippet or preview, entity category, hub score, and timestamps.
+     This is a decision surface \u2014 usually enough to answer without reading any files.
   2. Escalate to "get_note_structure" only when you need the full markdown content
      or word count. Use "get_section_content" to read one section by heading name.
   3. Start with a broad search: just query text, no filters. Only add folder, tag,
@@ -10900,7 +10900,7 @@ Tool selection:
   if (!hasEmbeddingsIndex()) {
     parts.push(`
 **Setup:** Run \`init_semantic\` once to build embeddings. This unlocks hybrid search (BM25 + semantic),
-improves recall results, and enables similarity-based tools. Without it, search is keyword-only.`);
+improves search results, and enables similarity-based tools. Without it, search is keyword-only.`);
   }
   if (registry?.isMultiVault) {
     parts.push(`
@@ -10915,9 +10915,9 @@ This server manages multiple vaults. Every tool has an optional "vault" paramete
 **Frontmatter matters more than content** for Flywheel's intelligence. When creating or updating notes, always set:
   - \`type:\` \u2014 drives entity categorization (person, project, technology). Without it, the category is guessed from the name alone and is often wrong.
   - \`aliases:\` \u2014 alternative names so the entity is found when referred to differently. Without it, the entity is invisible to searches using alternate names.
-  - \`description:\` \u2014 one-line summary shown in search results and used by recall. Without it, search results and recall are degraded.
+  - \`description:\` \u2014 one-line summary shown in search results and used for entity ranking. Without it, search quality is degraded.
   - Tags \u2014 used for filtering, suggestion scoring, and schema analysis.
-Good frontmatter is the highest-leverage action for improving suggestions, recall, and link quality.`);
+Good frontmatter is the highest-leverage action for improving suggestions, search, and link quality.`);
   if (categories.has("read")) {
     parts.push(`
 ## Read
@@ -10968,7 +10968,10 @@ you say "run the weekly review for this week".`);
     parts.push(`
 ## Memory
 
-Session workflow: call "brief" at conversation start for vault context (recent sessions, active entities, stored memories). Use "recall" before answering questions \u2014 it searches entities, notes, and memories with graph-boosted ranking. Use "memory" with action "store" to save observations, facts, or context that should persist across sessions (e.g. key decisions, user preferences, project status).`);
+"brief" delivers startup context (recent sessions, active entities, stored memories) \u2014 call it at
+conversation start. "search" finds everything \u2014 notes, entities, and memories in one call. "memory"
+with action "store" persists observations, facts, or preferences across sessions (e.g. key decisions,
+user preferences, project status).`);
   }
   if (categories.has("graph")) {
     parts.push(`
@@ -11013,7 +11016,7 @@ import * as path37 from "path";
 import { dirname as dirname5, join as join19 } from "path";
 import { statSync as statSync6, readFileSync as readFileSync5 } from "fs";
 import { fileURLToPath } from "url";
-import { z as z40 } from "zod";
+import { z as z39 } from "zod";
 import { getSessionId } from "@velvetmonkey/vault-core";
 init_vault_scope();
 
@@ -13851,64 +13854,6 @@ function enrichResultLight(result, index, stateDb2) {
   }
   return enriched;
 }
-function enrichEntityCompact(entityName, stateDb2, index) {
-  const enriched = {};
-  if (stateDb2) {
-    try {
-      const entity = getEntityByName2(stateDb2, entityName);
-      if (entity) {
-        enriched.category = entity.category;
-        enriched.hub_score = entity.hubScore;
-        if (entity.aliases.length > 0) enriched.aliases = entity.aliases;
-        enriched.path = entity.path;
-      }
-    } catch {
-    }
-  }
-  if (index) {
-    const entityPath = enriched.path ?? index.entities.get(entityName.toLowerCase());
-    if (entityPath) {
-      const note = index.notes.get(entityPath);
-      const normalizedPath = entityPath.toLowerCase().replace(/\.md$/, "");
-      const backlinks = index.backlinks.get(normalizedPath) || [];
-      enriched.backlink_count = backlinks.length;
-      if (note) {
-        if (Object.keys(note.frontmatter).length > 0) enriched.frontmatter = note.frontmatter;
-        if (note.tags.length > 0) enriched.tags = note.tags;
-        if (note.outlinks.length > 0) {
-          enriched.outlink_names = getOutlinkNames(note.outlinks, entityPath, index, stateDb2, COMPACT_OUTLINK_NAMES);
-        }
-      }
-    }
-  }
-  return enriched;
-}
-function enrichNoteCompact(notePath, stateDb2, index) {
-  const enriched = {};
-  if (!index) return enriched;
-  const note = index.notes.get(notePath);
-  if (!note) return enriched;
-  const normalizedPath = notePath.toLowerCase().replace(/\.md$/, "");
-  const backlinks = index.backlinks.get(normalizedPath) || [];
-  if (Object.keys(note.frontmatter).length > 0) enriched.frontmatter = note.frontmatter;
-  if (note.tags.length > 0) enriched.tags = note.tags;
-  enriched.backlink_count = backlinks.length;
-  enriched.modified = note.modified.toISOString();
-  if (note.outlinks.length > 0) {
-    enriched.outlink_names = getOutlinkNames(note.outlinks, notePath, index, stateDb2, COMPACT_OUTLINK_NAMES);
-  }
-  if (stateDb2) {
-    try {
-      const entity = getEntityByName2(stateDb2, note.title);
-      if (entity) {
-        enriched.category = entity.category;
-        enriched.hub_score = entity.hubScore;
-      }
-    } catch {
-    }
-  }
-  return enriched;
-}
 
 // src/core/read/multihop.ts
 import { getEntityByName as getEntityByName3, searchEntities } from "@velvetmonkey/vault-core";
@@ -14381,7 +14326,7 @@ function sortNotes(notes, sortBy, order) {
 function registerQueryTools(server2, getIndex, getVaultPath, getStateDb3) {
   server2.tool(
     "search",
-    'Search the vault \u2014 always start with just a query, no filters. Top results get full metadata (frontmatter, top backlinks/outlinks ranked by edge weight + recency); remaining results get lightweight summaries. Narrow with filters only if the broad search returns too many irrelevant results. Use get_note_structure for headings/full structure, get_backlinks for complete backlink lists.\n\nSearches across content (FTS5 full-text + hybrid semantic), entities (people/projects/technologies), and metadata (frontmatter/tags/folders). Hybrid semantic results are automatically included when embeddings have been built (via init_semantic).\n\nExample: search({ query: "quarterly review", limit: 5 })\nExample: search({ where: { type: "project", status: "active" } })\n\nMulti-vault: when configured with multiple vaults, omitting the `vault` parameter searches all vaults and merges results (each result includes a `vault` field). Pass `vault` to search a specific vault.',
+    'Search everything \u2014 notes, entities, and memories \u2014 in one call. Returns a decision surface with three sections: note results (with section provenance, dates, bridges, confidence), matching entity profiles, and relevant memories.\n\nNote results carry full metadata (frontmatter, scored backlinks/outlinks, snippets). Start with just a query, no filters. Narrow with filters only if needed. Use get_note_structure for full content, get_section_content to read one section.\n\nSearches note content (FTS5 + hybrid semantic), entity profiles (people, projects, technologies), and stored memories. Hybrid results included automatically when embeddings are built (via init_semantic).\n\nExample: search({ query: "quarterly review", limit: 5 })\nExample: search({ where: { type: "project", status: "active" } })\n\nMulti-vault: omitting `vault` searches all vaults and merges results. Pass `vault` to search a specific vault.',
     {
       query: z5.string().optional().describe("Search query text. Required unless using metadata filters (where, has_tag, folder, etc.)"),
       // Metadata filters
@@ -14765,6 +14710,7 @@ function registerSystemTools(server2, getIndex, setIndex, getVaultPath, setConfi
         const stateDb2 = getStateDb3?.();
         if (stateDb2) {
           try {
+            const config = loadConfig(stateDb2);
             const entityIndex2 = await scanVaultEntities2(vaultPath2, {
               excludeFolders: [
                 "daily-notes",
@@ -14786,7 +14732,8 @@ function registerSystemTools(server2, getIndex, setIndex, getVaultPath, setConfi
                 "articles",
                 "bookmarks",
                 "web-clips"
-              ]
+              ],
+              customCategories: config.custom_categories
             });
             stateDb2.replaceAllEntities(entityIndex2);
             console.error(`[Flywheel] Updated ${entityIndex2._metadata.total_entities} entities in StateDb`);
@@ -21768,410 +21715,8 @@ function registerMemoryTools(server2, getStateDb3) {
   );
 }
 
-// src/tools/read/recall.ts
-import { z as z26 } from "zod";
-import { searchEntities as searchEntitiesDb2 } from "@velvetmonkey/vault-core";
-init_recency();
-init_cooccurrence();
-init_wikilinks();
-init_edgeWeights();
-init_wikilinkFeedback();
-init_embeddings();
-init_stemmer();
-
-// src/core/read/mmr.ts
-init_embeddings();
-function selectByMmr(candidates, limit, lambda = 0.7) {
-  if (candidates.length <= limit) return candidates;
-  if (candidates.length === 0) return [];
-  const maxScore = Math.max(...candidates.map((c) => c.score));
-  if (maxScore === 0) return candidates.slice(0, limit);
-  const normScores = /* @__PURE__ */ new Map();
-  for (const c of candidates) {
-    normScores.set(c.id, c.score / maxScore);
-  }
-  const selected = [];
-  const remaining = new Set(candidates.map((_, i) => i));
-  let bestIdx = 0;
-  let bestScore = -Infinity;
-  for (const idx of remaining) {
-    if (candidates[idx].score > bestScore) {
-      bestScore = candidates[idx].score;
-      bestIdx = idx;
-    }
-  }
-  selected.push(candidates[bestIdx]);
-  remaining.delete(bestIdx);
-  while (selected.length < limit && remaining.size > 0) {
-    let bestMmr = -Infinity;
-    let bestCandidate = -1;
-    for (const idx of remaining) {
-      const candidate = candidates[idx];
-      const relevance = normScores.get(candidate.id) || 0;
-      let maxSim = 0;
-      if (candidate.embedding !== null) {
-        for (const sel of selected) {
-          if (sel.embedding !== null) {
-            const sim = cosineSimilarity(candidate.embedding, sel.embedding);
-            if (sim > maxSim) maxSim = sim;
-          }
-        }
-      }
-      const mmr = lambda * relevance - (1 - lambda) * maxSim;
-      if (mmr > bestMmr) {
-        bestMmr = mmr;
-        bestCandidate = idx;
-      }
-    }
-    if (bestCandidate === -1) break;
-    selected.push(candidates[bestCandidate]);
-    remaining.delete(bestCandidate);
-  }
-  return selected;
-}
-
-// src/tools/read/recall.ts
-function scoreTextRelevance(query, content) {
-  const queryTokens = tokenize(query).map((t) => t.toLowerCase());
-  const queryStems = queryTokens.map((t) => stem(t));
-  const contentLower = content.toLowerCase();
-  const contentTokens = new Set(tokenize(contentLower));
-  const contentStems = new Set([...contentTokens].map((t) => stem(t)));
-  let score = 0;
-  for (let i = 0; i < queryTokens.length; i++) {
-    const token = queryTokens[i];
-    const stemmed = queryStems[i];
-    if (contentTokens.has(token)) {
-      score += 10;
-    } else if (contentStems.has(stemmed)) {
-      score += 5;
-    }
-  }
-  if (contentLower.includes(query.toLowerCase())) {
-    score += 15;
-  }
-  return score;
-}
-function getEdgeWeightBoost(entityName, edgeWeightMap) {
-  const avgWeight = edgeWeightMap.get(entityName.toLowerCase());
-  if (!avgWeight || avgWeight <= 1) return 0;
-  return Math.min((avgWeight - 1) * 3, 6);
-}
-async function performRecall(stateDb2, query, options = {}) {
-  const {
-    max_results = 20,
-    focus,
-    entity,
-    max_tokens,
-    diversity = 1,
-    vaultPath: vaultPath2
-  } = options;
-  const results = [];
-  const recencyIndex2 = loadRecencyFromStateDb();
-  const edgeWeightMap = getEntityEdgeWeightMap(stateDb2);
-  const feedbackBoosts = getAllFeedbackBoosts(stateDb2);
-  const cooccurrenceIndex2 = getCooccurrenceIndex();
-  if (!focus || focus === "entities") {
-    try {
-      const entityResults = searchEntitiesDb2(stateDb2, query, max_results);
-      for (const e of entityResults) {
-        const textScore = scoreTextRelevance(query, `${e.name} ${e.description || ""}`);
-        const recency = recencyIndex2 ? getRecencyBoost(e.name, recencyIndex2) : 0;
-        const feedback = feedbackBoosts.get(e.name) ?? 0;
-        const edgeWeight = getEdgeWeightBoost(e.name, edgeWeightMap);
-        const total = textScore + recency + feedback + edgeWeight;
-        if (total > 0) {
-          results.push({
-            type: "entity",
-            id: e.name,
-            content: e.description || `Entity: ${e.name} (${e.category})`,
-            score: total,
-            breakdown: {
-              textRelevance: textScore,
-              recencyBoost: recency,
-              cooccurrenceBoost: 0,
-              feedbackBoost: feedback,
-              edgeWeightBoost: edgeWeight,
-              semanticBoost: 0
-            }
-          });
-        }
-      }
-    } catch {
-    }
-  }
-  if (!focus || focus === "notes") {
-    try {
-      const noteResults = searchFTS5("", query, max_results);
-      for (const n of noteResults) {
-        const textScore = Math.max(10, scoreTextRelevance(query, `${n.title || ""} ${n.snippet || ""}`));
-        const entityName = n.title || n.path.replace(/\.md$/, "").split("/").pop() || "";
-        const recency = recencyIndex2 ? getRecencyBoost(entityName, recencyIndex2) : 0;
-        const feedback = feedbackBoosts.get(entityName) ?? 0;
-        const edgeWeight = getEdgeWeightBoost(entityName, edgeWeightMap);
-        const total = textScore + recency + feedback + edgeWeight;
-        results.push({
-          type: "note",
-          id: n.path,
-          content: n.snippet || n.title || n.path,
-          score: total,
-          breakdown: {
-            textRelevance: textScore,
-            recencyBoost: recency,
-            cooccurrenceBoost: 0,
-            // applied in post-pass below
-            feedbackBoost: feedback,
-            edgeWeightBoost: edgeWeight,
-            semanticBoost: 0
-          }
-        });
-      }
-    } catch {
-    }
-  }
-  if (!focus || focus === "memories") {
-    try {
-      const memResults = searchMemories(stateDb2, {
-        query,
-        entity,
-        limit: max_results
-      });
-      const now = Date.now();
-      for (const m of memResults) {
-        const textScore = scoreTextRelevance(query, `${m.key} ${m.value}`);
-        const confidenceBoost = m.confidence * 5;
-        let typeBoost = 0;
-        switch (m.memory_type) {
-          case "fact":
-            typeBoost = 3;
-            break;
-          case "preference":
-            typeBoost = 2;
-            break;
-          case "observation": {
-            const ageDays = (now - m.updated_at) / 864e5;
-            const recencyFactor = Math.max(0.2, 1 - ageDays / 7);
-            typeBoost = 1 + 4 * recencyFactor;
-            break;
-          }
-          case "summary":
-            typeBoost = 1;
-            break;
-        }
-        const memScore = textScore + confidenceBoost + typeBoost;
-        results.push({
-          type: "memory",
-          id: m.key,
-          content: m.value,
-          score: memScore,
-          breakdown: {
-            textRelevance: textScore,
-            recencyBoost: typeBoost,
-            // type-aware boost reported as recency
-            cooccurrenceBoost: 0,
-            feedbackBoost: confidenceBoost,
-            edgeWeightBoost: 0,
-            semanticBoost: 0
-          }
-        });
-      }
-    } catch {
-    }
-  }
-  if (cooccurrenceIndex2) {
-    const seedEntities = /* @__PURE__ */ new Set();
-    for (const r of results) {
-      if (r.type === "entity") {
-        seedEntities.add(r.id);
-      } else if (r.type === "note") {
-        const name = r.id.replace(/\.md$/, "").split("/").pop() || "";
-        if (name) seedEntities.add(name);
-      }
-    }
-    for (const r of results) {
-      if (r.type === "entity" || r.type === "note") {
-        const name = r.type === "entity" ? r.id : r.id.replace(/\.md$/, "").split("/").pop() || "";
-        const boost = getCooccurrenceBoost(name, seedEntities, cooccurrenceIndex2, recencyIndex2);
-        if (boost > 0) {
-          r.breakdown.cooccurrenceBoost = boost;
-          r.score += boost;
-        }
-      }
-    }
-  }
-  if ((!focus || focus === "entities") && query.length >= 20 && hasEntityEmbeddingsIndex()) {
-    try {
-      const embedding = await embedTextCached(query);
-      const semanticMatches = findSemanticallySimilarEntities(embedding, max_results);
-      for (const match of semanticMatches) {
-        if (match.similarity < 0.3) continue;
-        const boost = match.similarity * 15;
-        const existing = results.find((r) => r.type === "entity" && r.id === match.entityName);
-        if (existing) {
-          existing.score += boost;
-          existing.breakdown.semanticBoost = boost;
-        } else {
-          results.push({
-            type: "entity",
-            id: match.entityName,
-            content: `Semantically similar to: "${query}"`,
-            score: boost,
-            breakdown: {
-              textRelevance: 0,
-              recencyBoost: 0,
-              cooccurrenceBoost: 0,
-              feedbackBoost: 0,
-              edgeWeightBoost: 0,
-              semanticBoost: boost
-            }
-          });
-        }
-      }
-    } catch {
-    }
-  }
-  results.sort((a, b) => b.score - a.score);
-  const seen = /* @__PURE__ */ new Set();
-  const deduped = results.filter((r) => {
-    const key = `${r.type}:${r.id}`;
-    if (seen.has(key)) return false;
-    seen.add(key);
-    return true;
-  });
-  let selected;
-  if (hasEmbeddingsIndex() && deduped.length > max_results) {
-    const notePaths = deduped.filter((r) => r.type === "note").map((r) => r.id);
-    const noteEmbeddings = loadNoteEmbeddingsForPaths(notePaths);
-    let queryEmbedding = null;
-    const mmrCandidates = [];
-    for (const r of deduped) {
-      let embedding = null;
-      if (r.type === "entity") {
-        embedding = getEntityEmbedding(r.id);
-      } else if (r.type === "note") {
-        embedding = noteEmbeddings.get(r.id) ?? null;
-      } else if (r.type === "memory") {
-        try {
-          if (!queryEmbedding) queryEmbedding = await embedTextCached(query);
-          embedding = await embedTextCached(r.content);
-        } catch {
-        }
-      }
-      mmrCandidates.push({ id: `${r.type}:${r.id}`, score: r.score, embedding });
-    }
-    const mmrSelected = selectByMmr(mmrCandidates, max_results, diversity);
-    const selectedIds = new Set(mmrSelected.map((m) => m.id));
-    selected = deduped.filter((r) => selectedIds.has(`${r.type}:${r.id}`));
-    const orderMap = new Map(mmrSelected.map((m, i) => [m.id, i]));
-    selected.sort((a, b) => (orderMap.get(`${a.type}:${a.id}`) ?? 0) - (orderMap.get(`${b.type}:${b.id}`) ?? 0));
-  } else {
-    selected = deduped.slice(0, max_results);
-  }
-  if (vaultPath2) {
-    const queryTokens = tokenize(query).map((t) => t.toLowerCase());
-    let queryEmb = null;
-    if (hasEmbeddingsIndex()) {
-      try {
-        queryEmb = await embedTextCached(query);
-      } catch {
-      }
-    }
-    for (const r of selected) {
-      if (r.type !== "note") continue;
-      try {
-        const absPath = vaultPath2 + "/" + r.id;
-        const snippets = await extractBestSnippets(absPath, queryEmb, queryTokens);
-        if (snippets.length > 0 && snippets[0].text.length > 0) {
-          r.content = snippets[0].text;
-        }
-      } catch {
-      }
-    }
-  }
-  if (max_tokens) {
-    let tokenBudget = max_tokens;
-    const budgeted = [];
-    for (const r of selected) {
-      const estimatedTokens = Math.ceil(r.content.length / 4);
-      if (tokenBudget - estimatedTokens < 0 && budgeted.length > 0) break;
-      tokenBudget -= estimatedTokens;
-      budgeted.push(r);
-    }
-    return budgeted;
-  }
-  return selected;
-}
-function registerRecallTools(server2, getStateDb3, getVaultPath, getIndex) {
-  server2.tool(
-    "recall",
-    "Query everything the system knows about a topic. Searches across entities, notes, and memories with graph-boosted ranking.",
-    {
-      query: z26.string().describe('What to recall (e.g., "Project X", "meetings about auth")'),
-      max_results: z26.number().min(1).max(100).optional().describe("Max results (default: 20)"),
-      focus: z26.enum(["entities", "notes", "memories"]).optional().describe("Limit to a specific result type. Omit for best results (searches everything)."),
-      entity: z26.string().optional().describe("Filter memories by entity association"),
-      max_tokens: z26.number().optional().describe("Token budget for response (truncates lower-ranked results)"),
-      diversity: z26.number().min(0).max(1).optional().describe("Relevance vs diversity tradeoff (0=max diversity, 1=pure relevance, default: 0.7)")
-    },
-    async (args) => {
-      const stateDb2 = getStateDb3();
-      if (!stateDb2) {
-        return {
-          content: [{ type: "text", text: JSON.stringify({ error: "StateDb not available" }) }],
-          isError: true
-        };
-      }
-      const results = await performRecall(stateDb2, args.query, {
-        max_results: args.max_results,
-        focus: args.focus,
-        entity: args.entity,
-        max_tokens: args.max_tokens,
-        diversity: args.diversity,
-        vaultPath: getVaultPath?.()
-      });
-      const entities = results.filter((r) => r.type === "entity");
-      const notes = results.filter((r) => r.type === "note");
-      const memories = results.filter((r) => r.type === "memory");
-      const index = getIndex?.() ?? null;
-      const enrichedNotes = notes.map((n) => ({
-        path: n.id,
-        snippet: n.content,
-        ...enrichNoteCompact(n.id, stateDb2, index)
-      }));
-      const hopResults = index ? multiHopBackfill(enrichedNotes, index, stateDb2, {
-        maxBackfill: Math.max(5, (args.max_results ?? 10) - notes.length)
-      }) : [];
-      if (index) {
-        const allNoteResults = [...enrichedNotes, ...hopResults];
-        const expansionTerms = extractExpansionTerms(allNoteResults, args.query, index);
-        const expansionResults = expandQuery(expansionTerms, allNoteResults, index, stateDb2);
-        hopResults.push(...expansionResults);
-      }
-      return {
-        content: [{
-          type: "text",
-          text: JSON.stringify({
-            query: args.query,
-            total: results.length,
-            entities: entities.map((e) => ({
-              name: e.id,
-              description: e.content,
-              ...enrichEntityCompact(e.id, stateDb2, index)
-            })),
-            notes: [...enrichedNotes, ...hopResults],
-            memories: memories.map((m) => ({
-              key: m.id,
-              value: m.content
-            }))
-          }, null, 2)
-        }]
-      };
-    }
-  );
-}
-
 // src/tools/read/brief.ts
-import { z as z27 } from "zod";
+import { z as z26 } from "zod";
 init_corrections();
 function estimateTokens2(value) {
   const str = JSON.stringify(value);
@@ -22313,8 +21858,8 @@ function registerBriefTools(server2, getStateDb3) {
     "brief",
     "Get a startup context briefing: recent sessions, active entities, memories, pending corrections, and vault stats. Call at conversation start.",
     {
-      max_tokens: z27.number().optional().describe("Token budget (lower-priority sections truncated first)"),
-      focus: z27.string().optional().describe("Focus entity or topic (filters content)")
+      max_tokens: z26.number().optional().describe("Token budget (lower-priority sections truncated first)"),
+      focus: z26.string().optional().describe("Focus entity or topic (filters content)")
     },
     async (args) => {
       const stateDb2 = getStateDb3();
@@ -22365,24 +21910,24 @@ function registerBriefTools(server2, getStateDb3) {
 }
 
 // src/tools/write/config.ts
-import { z as z28 } from "zod";
+import { z as z27 } from "zod";
 import { saveFlywheelConfigToDb as saveFlywheelConfigToDb2 } from "@velvetmonkey/vault-core";
 var VALID_CONFIG_KEYS = {
-  vault_name: z28.string(),
-  exclude_task_tags: z28.array(z28.string()),
-  exclude_analysis_tags: z28.array(z28.string()),
-  exclude_entities: z28.array(z28.string()),
-  exclude_entity_folders: z28.array(z28.string()),
-  wikilink_strictness: z28.enum(["conservative", "balanced", "aggressive"]),
-  implicit_detection: z28.boolean(),
-  implicit_patterns: z28.array(z28.string()),
-  adaptive_strictness: z28.boolean(),
-  proactive_linking: z28.boolean(),
-  proactive_min_score: z28.number(),
-  proactive_max_per_file: z28.number(),
-  proactive_max_per_day: z28.number(),
-  custom_categories: z28.record(z28.string(), z28.object({
-    type_boost: z28.number().optional()
+  vault_name: z27.string(),
+  exclude_task_tags: z27.array(z27.string()),
+  exclude_analysis_tags: z27.array(z27.string()),
+  exclude_entities: z27.array(z27.string()),
+  exclude_entity_folders: z27.array(z27.string()),
+  wikilink_strictness: z27.enum(["conservative", "balanced", "aggressive"]),
+  implicit_detection: z27.boolean(),
+  implicit_patterns: z27.array(z27.string()),
+  adaptive_strictness: z27.boolean(),
+  proactive_linking: z27.boolean(),
+  proactive_min_score: z27.number(),
+  proactive_max_per_file: z27.number(),
+  proactive_max_per_day: z27.number(),
+  custom_categories: z27.record(z27.string(), z27.object({
+    type_boost: z27.number().optional()
   }))
 };
 function registerConfigTools(server2, getConfig2, setConfig, getStateDb3) {
@@ -22392,9 +21937,9 @@ function registerConfigTools(server2, getConfig2, setConfig, getStateDb3) {
       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: z28.enum(["get", "set"]).describe("Operation mode"),
-        key: z28.string().optional().describe("Config key to update (required for set mode)"),
-        value: z28.unknown().optional().describe("New value for the key (required for set mode)")
+        mode: z27.enum(["get", "set"]).describe("Operation mode"),
+        key: z27.string().optional().describe("Config key to update (required for set mode)"),
+        value: z27.unknown().optional().describe("New value for the key (required for set mode)")
       }
     },
     async ({ mode, key, value }) => {
@@ -22450,7 +21995,7 @@ function registerConfigTools(server2, getConfig2, setConfig, getStateDb3) {
 // src/tools/write/enrich.ts
 init_wikilinks();
 init_wikilinkFeedback();
-import { z as z29 } from "zod";
+import { z as z28 } from "zod";
 import * as fs32 from "fs/promises";
 import * as path35 from "path";
 import { scanVaultEntities as scanVaultEntities3, SCHEMA_VERSION as SCHEMA_VERSION2 } from "@velvetmonkey/vault-core";
@@ -22571,7 +22116,8 @@ async function executeRun(stateDb2, vaultPath2) {
   if (entityCount === 0) {
     const start = Date.now();
     try {
-      const entityIndex2 = await scanVaultEntities3(vaultPath2, { excludeFolders: EXCLUDE_FOLDERS });
+      const config = loadConfig(stateDb2);
+      const entityIndex2 = await scanVaultEntities3(vaultPath2, { excludeFolders: EXCLUDE_FOLDERS, customCategories: config.custom_categories });
       stateDb2.replaceAllEntities(entityIndex2);
       const newCount = entityIndex2._metadata.total_entities;
       steps.push({
@@ -22754,10 +22300,10 @@ function registerInitTools(server2, getVaultPath, getStateDb3) {
     "vault_init",
     `Initialize vault for Flywheel. Modes: "status" (check what's ready/missing), "run" (execute missing init steps), "enrich" (scan notes with zero wikilinks and apply entity links).`,
     {
-      mode: z29.enum(["status", "run", "enrich"]).default("status").describe("Operation mode (default: status)"),
-      dry_run: z29.boolean().default(true).describe("For enrich mode: preview without modifying files (default: true)"),
-      batch_size: z29.number().default(50).describe("For enrich mode: max notes per invocation (default: 50)"),
-      offset: z29.number().default(0).describe("For enrich mode: skip this many eligible notes (for pagination)")
+      mode: z28.enum(["status", "run", "enrich"]).default("status").describe("Operation mode (default: status)"),
+      dry_run: z28.boolean().default(true).describe("For enrich mode: preview without modifying files (default: true)"),
+      batch_size: z28.number().default(50).describe("For enrich mode: max notes per invocation (default: 50)"),
+      offset: z28.number().default(0).describe("For enrich mode: skip this many eligible notes (for pagination)")
     },
     async ({ mode, dry_run, batch_size, offset }) => {
       const stateDb2 = getStateDb3();
@@ -22784,7 +22330,7 @@ function registerInitTools(server2, getVaultPath, getStateDb3) {
 }
 
 // src/tools/read/metrics.ts
-import { z as z30 } from "zod";
+import { z as z29 } from "zod";
 function registerMetricsTools(server2, getIndex, getStateDb3) {
   server2.registerTool(
     "vault_growth",
@@ -22792,10 +22338,10 @@ function registerMetricsTools(server2, getIndex, getStateDb3) {
       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: z30.enum(["current", "history", "trends", "index_activity"]).describe("Query mode: current snapshot, historical time series, trend analysis, or index rebuild activity"),
-        metric: z30.string().optional().describe('Filter to specific metric (e.g., "note_count"). Omit for all metrics.'),
-        days_back: z30.number().optional().describe("Number of days to look back for history/trends (default: 30)"),
-        limit: z30.number().optional().describe("Number of recent events to return for index_activity mode (default: 20)")
+        mode: z29.enum(["current", "history", "trends", "index_activity"]).describe("Query mode: current snapshot, historical time series, trend analysis, or index rebuild activity"),
+        metric: z29.string().optional().describe('Filter to specific metric (e.g., "note_count"). Omit for all metrics.'),
+        days_back: z29.number().optional().describe("Number of days to look back for history/trends (default: 30)"),
+        limit: z29.number().optional().describe("Number of recent events to return for index_activity mode (default: 20)")
       }
     },
     async ({ mode, metric, days_back, limit: eventLimit }) => {
@@ -22868,7 +22414,7 @@ function registerMetricsTools(server2, getIndex, getStateDb3) {
 }
 
 // src/tools/read/activity.ts
-import { z as z31 } from "zod";
+import { z as z30 } from "zod";
 function registerActivityTools(server2, getStateDb3, getSessionId2) {
   server2.registerTool(
     "vault_activity",
@@ -22876,10 +22422,10 @@ function registerActivityTools(server2, getStateDb3, 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: z31.enum(["session", "sessions", "note_access", "tool_usage"]).describe("Activity query mode"),
-        session_id: z31.string().optional().describe("Specific session ID (for session mode, defaults to current)"),
-        days_back: z31.number().optional().describe("Number of days to look back (default: 30)"),
-        limit: z31.number().optional().describe("Maximum results to return (default: 20)")
+        mode: z30.enum(["session", "sessions", "note_access", "tool_usage"]).describe("Activity query mode"),
+        session_id: z30.string().optional().describe("Specific session ID (for session mode, defaults to current)"),
+        days_back: z30.number().optional().describe("Number of days to look back (default: 30)"),
+        limit: z30.number().optional().describe("Maximum results to return (default: 20)")
       }
     },
     async ({ mode, session_id, days_back, limit: resultLimit }) => {
@@ -22946,12 +22492,65 @@ function registerActivityTools(server2, getStateDb3, getSessionId2) {
 }
 
 // src/tools/read/similarity.ts
-import { z as z32 } from "zod";
+import { z as z31 } from "zod";
 
 // src/core/read/similarity.ts
 init_embeddings();
 import * as fs33 from "fs";
 import * as path36 from "path";
+
+// src/core/read/mmr.ts
+init_embeddings();
+function selectByMmr(candidates, limit, lambda = 0.7) {
+  if (candidates.length <= limit) return candidates;
+  if (candidates.length === 0) return [];
+  const maxScore = Math.max(...candidates.map((c) => c.score));
+  if (maxScore === 0) return candidates.slice(0, limit);
+  const normScores = /* @__PURE__ */ new Map();
+  for (const c of candidates) {
+    normScores.set(c.id, c.score / maxScore);
+  }
+  const selected = [];
+  const remaining = new Set(candidates.map((_, i) => i));
+  let bestIdx = 0;
+  let bestScore = -Infinity;
+  for (const idx of remaining) {
+    if (candidates[idx].score > bestScore) {
+      bestScore = candidates[idx].score;
+      bestIdx = idx;
+    }
+  }
+  selected.push(candidates[bestIdx]);
+  remaining.delete(bestIdx);
+  while (selected.length < limit && remaining.size > 0) {
+    let bestMmr = -Infinity;
+    let bestCandidate = -1;
+    for (const idx of remaining) {
+      const candidate = candidates[idx];
+      const relevance = normScores.get(candidate.id) || 0;
+      let maxSim = 0;
+      if (candidate.embedding !== null) {
+        for (const sel of selected) {
+          if (sel.embedding !== null) {
+            const sim = cosineSimilarity(candidate.embedding, sel.embedding);
+            if (sim > maxSim) maxSim = sim;
+          }
+        }
+      }
+      const mmr = lambda * relevance - (1 - lambda) * maxSim;
+      if (mmr > bestMmr) {
+        bestMmr = mmr;
+        bestCandidate = idx;
+      }
+    }
+    if (bestCandidate === -1) break;
+    selected.push(candidates[bestCandidate]);
+    remaining.delete(bestCandidate);
+  }
+  return selected;
+}
+
+// src/core/read/similarity.ts
 var STOP_WORDS = /* @__PURE__ */ new Set([
   "the",
   "be",
@@ -23225,9 +22824,9 @@ function registerSimilarityTools(server2, getIndex, getVaultPath, getStateDb3) {
       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). Already-linked notes are automatically excluded.",
       inputSchema: {
-        path: z32.string().describe('Path to the source note (relative to vault root, e.g. "projects/alpha.md")'),
-        limit: z32.number().optional().describe("Maximum number of similar notes to return (default: 10)"),
-        diversity: z32.number().min(0).max(1).optional().describe("Relevance vs diversity tradeoff (0=max diversity, 1=pure relevance, default: 0.7)")
+        path: z31.string().describe('Path to the source note (relative to vault root, e.g. "projects/alpha.md")'),
+        limit: z31.number().optional().describe("Maximum number of similar notes to return (default: 10)"),
+        diversity: z31.number().min(0).max(1).optional().describe("Relevance vs diversity tradeoff (0=max diversity, 1=pure relevance, default: 0.7)")
       }
     },
     async ({ path: path39, limit, diversity }) => {
@@ -23272,7 +22871,7 @@ function registerSimilarityTools(server2, getIndex, getVaultPath, getStateDb3) {
 
 // src/tools/read/semantic.ts
 init_embeddings();
-import { z as z33 } from "zod";
+import { z as z32 } from "zod";
 import { getAllEntitiesFromDb as getAllEntitiesFromDb3 } from "@velvetmonkey/vault-core";
 function registerSemanticTools(server2, getVaultPath, getStateDb3) {
   server2.registerTool(
@@ -23281,7 +22880,7 @@ function registerSemanticTools(server2, getVaultPath, getStateDb3) {
       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: z33.boolean().optional().describe(
+        force: z32.boolean().optional().describe(
           "Rebuild all embeddings even if they already exist (default: false)"
         )
       }
@@ -23376,7 +22975,7 @@ function registerSemanticTools(server2, getVaultPath, getStateDb3) {
 
 // src/tools/read/merges.ts
 init_levenshtein();
-import { z as z34 } from "zod";
+import { z as z33 } from "zod";
 import { getAllEntitiesFromDb as getAllEntitiesFromDb4, getDismissedMergePairs, recordMergeDismissal } from "@velvetmonkey/vault-core";
 function normalizeName(name) {
   return name.toLowerCase().replace(/[.\-_]/g, "").replace(/js$/, "").replace(/ts$/, "");
@@ -23386,7 +22985,7 @@ function registerMergeTools2(server2, getStateDb3) {
     "suggest_entity_merges",
     "Find potential duplicate entities that could be merged based on name similarity",
     {
-      limit: z34.number().optional().default(50).describe("Maximum number of suggestions to return")
+      limit: z33.number().optional().default(50).describe("Maximum number of suggestions to return")
     },
     async ({ limit }) => {
       const stateDb2 = getStateDb3();
@@ -23488,11 +23087,11 @@ function registerMergeTools2(server2, getStateDb3) {
     "dismiss_merge_suggestion",
     "Permanently dismiss a merge suggestion so it never reappears",
     {
-      source_path: z34.string().describe("Path of the source entity"),
-      target_path: z34.string().describe("Path of the target entity"),
-      source_name: z34.string().describe("Name of the source entity"),
-      target_name: z34.string().describe("Name of the target entity"),
-      reason: z34.string().describe("Original suggestion reason")
+      source_path: z33.string().describe("Path of the source entity"),
+      target_path: z33.string().describe("Path of the target entity"),
+      source_name: z33.string().describe("Name of the source entity"),
+      target_name: z33.string().describe("Name of the target entity"),
+      reason: z33.string().describe("Original suggestion reason")
     },
     async ({ source_path, target_path, source_name, target_name, reason }) => {
       const stateDb2 = getStateDb3();
@@ -23511,7 +23110,7 @@ function registerMergeTools2(server2, getStateDb3) {
 }
 
 // src/tools/read/temporalAnalysis.ts
-import { z as z35 } from "zod";
+import { z as z34 } from "zod";
 init_wikilinks();
 function formatDate3(d) {
   return d.toISOString().split("T")[0];
@@ -24037,9 +23636,9 @@ function registerTemporalAnalysisTools(server2, getIndex, getVaultPath, getState
       title: "Context Around Date",
       description: "Reconstruct what was happening in the vault around a specific date. Shows modified/created notes, active entities, wikilink activity, and file moves within a time window.",
       inputSchema: {
-        date: z35.string().describe("Center date in YYYY-MM-DD format"),
-        window_days: z35.coerce.number().default(3).describe("Days before and after the center date (default 3 = 7-day window)"),
-        limit: z35.coerce.number().default(50).describe("Maximum number of notes to return")
+        date: z34.string().describe("Center date in YYYY-MM-DD format"),
+        window_days: z34.coerce.number().default(3).describe("Days before and after the center date (default 3 = 7-day window)"),
+        limit: z34.coerce.number().default(50).describe("Maximum number of notes to return")
       }
     },
     async ({ date, window_days, limit: requestedLimit }) => {
@@ -24054,12 +23653,12 @@ function registerTemporalAnalysisTools(server2, getIndex, getVaultPath, getState
       title: "Predict Stale Notes",
       description: "Multi-signal staleness prediction. Scores notes by importance (backlinks, hub score, tasks, status) and staleness risk (age, entity disconnect, task urgency). Returns concrete recommendations: archive, update, review, or low_priority.",
       inputSchema: {
-        days: z35.coerce.number().default(30).describe("Notes not modified in this many days (default 30)"),
-        min_importance: z35.coerce.number().default(0).describe("Filter by minimum importance score 0-100 (default 0)"),
-        include_recommendations: z35.boolean().default(true).describe("Include action recommendations (default true)"),
-        folder: z35.string().optional().describe("Limit to notes in this folder"),
-        limit: z35.coerce.number().default(30).describe("Maximum results to return (default 30)"),
-        offset: z35.coerce.number().default(0).describe("Results to skip for pagination (default 0)")
+        days: z34.coerce.number().default(30).describe("Notes not modified in this many days (default 30)"),
+        min_importance: z34.coerce.number().default(0).describe("Filter by minimum importance score 0-100 (default 0)"),
+        include_recommendations: z34.boolean().default(true).describe("Include action recommendations (default true)"),
+        folder: z34.string().optional().describe("Limit to notes in this folder"),
+        limit: z34.coerce.number().default(30).describe("Maximum results to return (default 30)"),
+        offset: z34.coerce.number().default(0).describe("Results to skip for pagination (default 0)")
       }
     },
     async ({ days, min_importance, include_recommendations, folder, limit: requestedLimit, offset }) => {
@@ -24083,9 +23682,9 @@ function registerTemporalAnalysisTools(server2, getIndex, getVaultPath, getState
       title: "Track Concept Evolution",
       description: "Timeline of how an entity has evolved: link additions/removals, feedback events, category changes, co-occurrence shifts. Shows current state, chronological event history, link durability stats, and top co-occurrence neighbors.",
       inputSchema: {
-        entity: z35.string().describe("Entity name (case-insensitive)"),
-        days_back: z35.coerce.number().default(90).describe("How far back to look (default 90 days)"),
-        include_cooccurrence: z35.boolean().default(true).describe("Include co-occurrence neighbors (default true)")
+        entity: z34.string().describe("Entity name (case-insensitive)"),
+        days_back: z34.coerce.number().default(90).describe("How far back to look (default 90 days)"),
+        include_cooccurrence: z34.boolean().default(true).describe("Include co-occurrence neighbors (default true)")
       }
     },
     async ({ entity, days_back, include_cooccurrence }) => {
@@ -24105,10 +23704,10 @@ function registerTemporalAnalysisTools(server2, getIndex, getVaultPath, getState
       title: "Temporal Summary",
       description: "Generate a vault pulse report for a time period. Composes context, staleness prediction, and concept evolution into a single summary. Shows activity snapshot, entity momentum, and maintenance alerts. Use for weekly/monthly/quarterly reviews.",
       inputSchema: {
-        start_date: z35.string().describe("Start of period in YYYY-MM-DD format"),
-        end_date: z35.string().describe("End of period in YYYY-MM-DD format"),
-        focus_entities: z35.array(z35.string()).optional().describe("Specific entities to track evolution for (default: top 5 active entities in period)"),
-        limit: z35.coerce.number().default(50).describe("Maximum notes to include in context snapshot")
+        start_date: z34.string().describe("Start of period in YYYY-MM-DD format"),
+        end_date: z34.string().describe("End of period in YYYY-MM-DD format"),
+        focus_entities: z34.array(z34.string()).optional().describe("Specific entities to track evolution for (default: top 5 active entities in period)"),
+        limit: z34.coerce.number().default(50).describe("Maximum notes to include in context snapshot")
       }
     },
     async ({ start_date, end_date, focus_entities, limit: requestedLimit }) => {
@@ -24127,15 +23726,15 @@ function registerTemporalAnalysisTools(server2, getIndex, getVaultPath, getState
 }
 
 // src/tools/read/sessionHistory.ts
-import { z as z36 } from "zod";
+import { z as z35 } from "zod";
 function registerSessionHistoryTools(server2, getStateDb3) {
   server2.tool(
     "vault_session_history",
     "View session history. Without session_id: lists recent sessions. With session_id: returns chronological tool invocations, notes accessed, and timing. Hierarchical sessions supported (parent ID includes child sessions).",
     {
-      session_id: z36.string().optional().describe("Session ID for detail view. Omit for recent sessions list."),
-      include_children: z36.boolean().optional().describe("Include child sessions (default: true)"),
-      limit: z36.number().min(1).max(500).optional().describe("Max invocations to return in detail view (default: 200)")
+      session_id: z35.string().optional().describe("Session ID for detail view. Omit for recent sessions list."),
+      include_children: z35.boolean().optional().describe("Include child sessions (default: true)"),
+      limit: z35.number().min(1).max(500).optional().describe("Max invocations to return in detail view (default: 200)")
     },
     async (args) => {
       const stateDb2 = getStateDb3();
@@ -24183,7 +23782,7 @@ function registerSessionHistoryTools(server2, getStateDb3) {
 }
 
 // src/tools/read/entityHistory.ts
-import { z as z37 } from "zod";
+import { z as z36 } from "zod";
 
 // src/core/read/entityHistory.ts
 function normalizeTimestamp(ts) {
@@ -24341,8 +23940,8 @@ function registerEntityHistoryTools(server2, getStateDb3) {
     "vault_entity_history",
     "Get a unified timeline of everything about an entity: when it was linked, feedback received, suggestion scores, edge weight changes, metadata mutations, memories, and corrections. Sorted chronologically with pagination.",
     {
-      entity_name: z37.string().describe("Entity name to query (case-insensitive)"),
-      event_types: z37.array(z37.enum([
+      entity_name: z36.string().describe("Entity name to query (case-insensitive)"),
+      event_types: z36.array(z36.enum([
         "application",
         "feedback",
         "suggestion",
@@ -24351,10 +23950,10 @@ function registerEntityHistoryTools(server2, getStateDb3) {
         "memory",
         "correction"
       ])).optional().describe("Filter to specific event types. Omit for all types."),
-      start_date: z37.string().optional().describe("Start date (YYYY-MM-DD) for date range filter"),
-      end_date: z37.string().optional().describe("End date (YYYY-MM-DD) for date range filter"),
-      limit: z37.number().min(1).max(200).optional().describe("Max events to return (default: 50)"),
-      offset: z37.number().min(0).optional().describe("Offset for pagination (default: 0)")
+      start_date: z36.string().optional().describe("Start date (YYYY-MM-DD) for date range filter"),
+      end_date: z36.string().optional().describe("End date (YYYY-MM-DD) for date range filter"),
+      limit: z36.number().min(1).max(200).optional().describe("Max events to return (default: 50)"),
+      offset: z36.number().min(0).optional().describe("Offset for pagination (default: 0)")
     },
     async (args) => {
       const stateDb2 = getStateDb3();
@@ -24386,7 +23985,7 @@ function registerEntityHistoryTools(server2, getStateDb3) {
 }
 
 // src/tools/read/learningReport.ts
-import { z as z38 } from "zod";
+import { z as z37 } from "zod";
 
 // src/core/read/learningReport.ts
 function isoDate(d) {
@@ -24522,8 +24121,8 @@ function registerLearningReportTools(server2, getIndex, getStateDb3) {
     "flywheel_learning_report",
     "Get a narrative report of the flywheel auto-linking system's learning progress. Shows: applications by day, feedback (positive/negative), survival rate, top rejected entities, suggestion funnel (evaluations \u2192 applications \u2192 survivals), and graph growth. Use compare=true for period-over-period deltas.",
     {
-      days_back: z38.number().min(1).max(365).optional().describe("Analysis window in days (default: 7). Use 1 for today, 2 for last 48h, etc."),
-      compare: z38.boolean().optional().describe("Include comparison with the preceding equal-length period (default: false)")
+      days_back: z37.number().min(1).max(365).optional().describe("Analysis window in days (default: 7). Use 1 for today, 2 for last 48h, etc."),
+      compare: z37.boolean().optional().describe("Include comparison with the preceding equal-length period (default: false)")
     },
     async (args) => {
       const stateDb2 = getStateDb3();
@@ -24550,7 +24149,7 @@ function registerLearningReportTools(server2, getIndex, getStateDb3) {
 }
 
 // src/tools/read/calibrationExport.ts
-import { z as z39 } from "zod";
+import { z as z38 } from "zod";
 
 // src/core/read/calibrationExport.ts
 init_wikilinkFeedback();
@@ -24843,8 +24442,8 @@ function registerCalibrationExportTools(server2, getIndex, getStateDb3, getConfi
     "flywheel_calibration_export",
     "Export anonymized aggregate scoring data for cross-vault algorithm calibration. No entity names, note paths, or content \u2014 safe to share. Includes: suggestion funnel, per-layer contribution averages, survival rates by entity category, score distribution, suppression stats, recency/co-occurrence effectiveness, and threshold sweep.",
     {
-      days_back: z39.number().min(1).max(365).optional().describe("Analysis window in days (default: 30)"),
-      include_vault_id: z39.boolean().optional().describe("Include anonymous vault ID for longitudinal tracking (default: true)")
+      days_back: z38.number().min(1).max(365).optional().describe("Analysis window in days (default: 30)"),
+      include_vault_id: z38.boolean().optional().describe("Include anonymous vault ID for longitudinal tracking (default: true)")
     },
     async (args) => {
       const stateDb2 = getStateDb3();
@@ -25131,7 +24730,7 @@ function applyToolGating(targetServer, categories, getDb4, registry, getVaultPat
     const schemaIdx = handlerIdx - 1;
     const schema = args[schemaIdx];
     if (schema && typeof schema === "object" && !Array.isArray(schema)) {
-      schema.vault = z40.string().optional().describe(
+      schema.vault = z39.string().optional().describe(
         `Vault name for multi-vault mode. Available: ${registry.getVaultNames().join(", ")}. Default: ${registry.primaryName}`
       );
     }
@@ -25257,7 +24856,6 @@ function registerAllTools(targetServer, ctx) {
   registerLearningReportTools(targetServer, gvi, gsd);
   registerCalibrationExportTools(targetServer, gvi, gsd, gcf);
   registerMemoryTools(targetServer, gsd);
-  registerRecallTools(targetServer, gsd, gvp, () => gvi() ?? null);
   registerBriefTools(targetServer, gsd);
   registerVaultResources(targetServer, () => gvi() ?? null);
 }
@@ -25441,7 +25039,10 @@ function updateFlywheelConfig(config) {
   flywheelConfig = config;
   setWikilinkConfig(config);
   const ctx = getActiveVaultContext();
-  if (ctx) ctx.flywheelConfig = config;
+  if (ctx) {
+    ctx.flywheelConfig = config;
+    setActiveScope(buildVaultScope(ctx));
+  }
 }
 async function bootVault(ctx, startTime) {
   const vp = ctx.vaultPath;
@@ -25594,6 +25195,7 @@ async function main() {
   loadVaultCooccurrence(primaryCtx);
   activateVault(primaryCtx);
   await bootVault(primaryCtx, startTime);
+  activateVault(primaryCtx);
   if (vaultConfigs && vaultConfigs.length > 1) {
     const secondaryConfigs = vaultConfigs.slice(1);
     (async () => {
@@ -25622,7 +25224,8 @@ async function updateEntitiesInStateDb(vp, sd) {
     const config = loadConfig(db4);
     const excludeFolders = config.exclude_entity_folders?.length ? config.exclude_entity_folders : DEFAULT_ENTITY_EXCLUDE_FOLDERS;
     const entityIndex2 = await scanVaultEntities4(vault, {
-      excludeFolders
+      excludeFolders,
+      customCategories: config.custom_categories
     });
     db4.replaceAllEntities(entityIndex2);
     serverLog("index", `Updated ${entityIndex2._metadata.total_entities} entities in StateDb`);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@velvetmonkey/flywheel-memory",
-  "version": "2.0.148",
+  "version": "2.0.150",
   "description": "MCP server that gives Claude full read/write access to your Obsidian vault. Select from 74 tools for search, backlinks, graph queries, mutations, agent memory, and hybrid semantic search.",
   "type": "module",
   "main": "dist/index.js",
@@ -54,7 +54,7 @@
   "dependencies": {
     "@huggingface/transformers": "^3.8.1",
     "@modelcontextprotocol/sdk": "^1.25.1",
-    "@velvetmonkey/vault-core": "^2.0.148",
+    "@velvetmonkey/vault-core": "^2.0.150",
     "better-sqlite3": "^12.0.0",
     "chokidar": "^4.0.0",
     "gray-matter": "^4.0.3",
@@ -85,4 +85,4 @@
   "publishConfig": {
     "access": "public"
   }
-}
+}