@velvetmonkey/flywheel-memory 2.0.147 → 2.0.149

package/dist/index.js

@@ -3396,7 +3396,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();
@@ -3689,6 +3690,12 @@ function isLikelyArticleTitle(name) {
   }
   return false;
 }
+function getTypeBoost(category, customCategories) {
+  if (customCategories?.[category]?.type_boost != null) {
+    return customCategories[category].type_boost;
+  }
+  return TYPE_BOOST[category] || 0;
+}
 function getCrossFolderBoost(entityPath, notePath) {
   if (!entityPath || !notePath) return 0;
   const entityParts = entityPath.split("/");
@@ -3869,7 +3876,7 @@ async function suggestRelatedLinks(content, options = {}) {
     if (contentScore > 0) {
       entitiesWithContentMatch.add(entityName);
     }
-    const layerTypeBoost = disabled.has("type_boost") ? 0 : TYPE_BOOST[category] || 0;
+    const layerTypeBoost = disabled.has("type_boost") ? 0 : getTypeBoost(category, getConfig()?.custom_categories);
     score += layerTypeBoost;
     const layerContextBoost = disabled.has("context_boost") ? 0 : contextBoosts[category] || 0;
     score += layerContextBoost;
@@ -3961,7 +3968,7 @@ async function suggestRelatedLinks(content, options = {}) {
           if (hasContentOverlap || strongCooccurrence) {
             entitiesWithContentMatch.add(entityName);
           }
-          const typeBoost = disabled.has("type_boost") ? 0 : TYPE_BOOST[category] || 0;
+          const typeBoost = disabled.has("type_boost") ? 0 : getTypeBoost(category, getConfig()?.custom_categories);
           const contextBoost = disabled.has("context_boost") ? 0 : contextBoosts[category] || 0;
           const recencyBoostVal = hasContentOverlap && !disabled.has("recency") ? recencyIndex ? getRecencyBoost(entityName, recencyIndex) : 0 : 0;
           const rawCrossFolderBoost = disabled.has("cross_folder") ? 0 : notePath && entity.path ? getCrossFolderBoost(entity.path, notePath) : 0;
@@ -4022,7 +4029,7 @@ async function suggestRelatedLinks(content, options = {}) {
           if (!disabled.has("length_filter") && match.entityName.length > MAX_ENTITY_LENGTH) continue;
           if (!disabled.has("article_filter") && isLikelyArticleTitle(match.entityName)) continue;
           const { entity, category } = entityWithType;
-          const layerTypeBoost = disabled.has("type_boost") ? 0 : TYPE_BOOST[category] || 0;
+          const layerTypeBoost = disabled.has("type_boost") ? 0 : getTypeBoost(category, getConfig()?.custom_categories);
           const layerContextBoost = disabled.has("context_boost") ? 0 : contextBoosts[category] || 0;
           const layerHubBoost = disabled.has("hub_boost") ? 0 : getHubBoost(entity);
           const layerCrossFolderBoost = disabled.has("cross_folder") ? 0 : notePath && entity.path ? getCrossFolderBoost(entity.path, notePath) : 0;
@@ -14759,6 +14766,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",
@@ -14780,7 +14788,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`);
@@ -22565,7 +22574,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({
@@ -25616,7 +25626,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.147",
+  "version": "2.0.149",
   "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.147",
+    "@velvetmonkey/vault-core": "^2.0.149",
     "better-sqlite3": "^12.0.0",
     "chokidar": "^4.0.0",
     "gray-matter": "^4.0.3",