stellavault 0.8.1 → 0.8.3

package/dist/stellavault.js

@@ -40,7 +40,9 @@ function mergeConfig(defaults, overrides) {
       ...defaults.search,
       ...overrides.search,
       // B3 §4 — deep-merge weights so a partial override keeps the other defaults.
-      weights: { ...defaults.search.weights, ...overrides.search?.weights }
+      weights: { ...defaults.search.weights, ...overrides.search?.weights },
+      // B2.2 — merge alias groups (override wins per-key).
+      entityAliases: { ...defaults.search.entityAliases, ...overrides.search?.entityAliases }
     },
     mcp: { ...defaults.mcp, ...overrides.mcp }
   };
@@ -49,7 +51,7 @@ function resolveSearchWeights(config, env = process.env) {
   const base = {
     semantic: config.search.weights?.semantic ?? 1,
     bm25: config.search.weights?.bm25 ?? 1,
-    entity: config.search.weights?.entity ?? 0.5,
+    entity: config.search.weights?.entity ?? 1.5,
     recency: config.search.recencyWeight ?? 0.2
   };
   const parse = (raw, min, max) => {
@@ -94,10 +96,12 @@ var init_config = __esm({
       search: {
         defaultLimit: 10,
         rrfK: 60,
-        weights: { semantic: 1, bm25: 1, entity: 0.5 },
-        // B3 §1.2
-        recencyWeight: 0.2
+        weights: { semantic: 1, bm25: 1, entity: 1.5 },
+        // B2.1: entity leads (per-doc cap prevents flooding)
+        recencyWeight: 0.2,
         // B3 §1.3 (±10% bound)
+        entityAliases: {}
+        // B2.2 — user-defined synonym groups
       },
       mcp: {
         mode: "stdio",
@@ -496,6 +500,34 @@ function extractQueryTerms(query) {
   }
   return [...set].slice(0, MAX_QUERY_TERMS);
 }
+function buildAliasIndex(aliases) {
+  const index = /* @__PURE__ */ new Map();
+  if (!aliases)
+    return index;
+  for (const [key, arr] of Object.entries(aliases)) {
+    const group = [normalize(key), ...(Array.isArray(arr) ? arr : []).map(normalize)].filter(Boolean);
+    const uniq = [...new Set(group)];
+    if (uniq.length < 2)
+      continue;
+    for (const term of uniq) {
+      const others = uniq.filter((t2) => t2 !== term);
+      index.set(term, [.../* @__PURE__ */ new Set([...index.get(term) ?? [], ...others])]);
+    }
+  }
+  return index;
+}
+function expandWithAliases(terms, aliasIndex) {
+  if (!aliasIndex || aliasIndex.size === 0)
+    return terms;
+  const out = new Set(terms);
+  for (const t2 of terms) {
+    const syn = aliasIndex.get(t2);
+    if (syn)
+      for (const s of syn)
+        out.add(s);
+  }
+  return [...out].slice(0, MAX_QUERY_TERMS);
+}
 var MAX_ENTITIES_PER_CHUNK, MAX_QUERY_TERMS, STOPWORDS;
 var init_entity_extractor = __esm({
   "packages/core/dist/indexer/entity-extractor.js"() {
@@ -3905,35 +3937,40 @@ function createSqliteVecStore(dbPath, dimensions = 384) {
         // FTS5 rank is negative (lower = better)
       }));
     },
-    async searchEntities(entities, limit) {
-      if (!entities || entities.length === 0)
+    async searchEntities(entities, limit, exactExtra = []) {
+      if ((!entities || entities.length === 0) && exactExtra.length === 0)
         return [];
-      const exactPH = entities.map(() => "?").join(",");
+      const allExact = [...entities, ...exactExtra];
+      const exactPH = allExact.map(() => "?").join(",");
       const fuzzy = entities.filter((t2) => t2.length >= 4 && (/\s/.test(t2) || /[^\x00-\x7f]/.test(t2) || t2.length >= 6)).slice(0, 16);
+      let matched;
+      let matchedParams;
       if (fuzzy.length === 0) {
-        const rows2 = db.prepare(`
-          SELECT chunk_id, COUNT(*) AS score
-          FROM chunk_entities
-          WHERE entity IN (${exactPH})
-          GROUP BY chunk_id
-          ORDER BY score DESC
-          LIMIT ?
-        `).all(...entities, limit);
-        return rows2.map((r) => ({ chunkId: r.chunk_id, score: r.score }));
-      }
-      const esc = (t2) => t2.replace(/[\\%_]/g, "\\$&");
-      const likeClause = fuzzy.map(() => `entity LIKE ? ESCAPE '\\'`).join(" OR ");
+        matched = `SELECT chunk_id, CAST(COUNT(*) AS REAL) AS score FROM chunk_entities WHERE entity IN (${exactPH}) GROUP BY chunk_id`;
+        matchedParams = [...allExact];
+      } else {
+        const esc = (t2) => t2.replace(/[\\%_]/g, "\\$&");
+        const likeClause = fuzzy.map(() => `entity LIKE ? ESCAPE '\\'`).join(" OR ");
+        matched = `
+          SELECT chunk_id, SUM(w) AS score FROM (
+            SELECT chunk_id, 1.0 AS w FROM chunk_entities WHERE entity IN (${exactPH})
+            UNION ALL
+            SELECT chunk_id, 0.4 AS w FROM chunk_entities
+              WHERE (${likeClause}) AND entity NOT IN (${exactPH})
+          ) GROUP BY chunk_id`;
+        matchedParams = [...allExact, ...fuzzy.map((t2) => `%${esc(t2)}%`), ...allExact];
+      }
       const rows = db.prepare(`
-        SELECT chunk_id, SUM(w) AS score FROM (
-          SELECT chunk_id, 1.0 AS w FROM chunk_entities WHERE entity IN (${exactPH})
-          UNION ALL
-          SELECT chunk_id, 0.4 AS w FROM chunk_entities
-            WHERE (${likeClause}) AND entity NOT IN (${exactPH})
+        SELECT chunk_id, score FROM (
+          SELECT m.chunk_id AS chunk_id, m.score AS score,
+                 ROW_NUMBER() OVER (PARTITION BY c.document_id ORDER BY m.score DESC, m.chunk_id) AS rn
+          FROM (${matched}) m
+          JOIN chunks c ON c.id = m.chunk_id
         )
-        GROUP BY chunk_id
-        ORDER BY score DESC
+        WHERE rn <= 2
+        ORDER BY score DESC, chunk_id
         LIMIT ?
-      `).all(...entities, ...fuzzy.map((t2) => `%${esc(t2)}%`), ...entities, limit);
+      `).all(...matchedParams, limit);
       return rows.map((r) => ({ chunkId: r.chunk_id, score: r.score }));
     },
     async getDocument(documentId) {
@@ -4139,13 +4176,14 @@ async function searchSemantic(store, embedder, query, limit) {
 
 // packages/core/dist/search/entity.js
 init_entity_extractor();
-async function searchEntities(store, query, limit) {
+async function searchEntities(store, query, limit, aliasIndex) {
   if (typeof store.searchEntities !== "function")
     return [];
   const terms = extractQueryTerms(query);
   if (terms.length === 0)
     return [];
-  return store.searchEntities(terms, limit);
+  const aliasExact = expandWithAliases(terms, aliasIndex).filter((t2) => !terms.includes(t2));
+  return store.searchEntities(terms, limit, aliasExact);
 }
 
 // packages/core/dist/search/rrf.js
@@ -4169,6 +4207,9 @@ function rrfFusionN(lists, k = 60, limit = 10, opts = {}) {
   return [...scores.entries()].sort((a, b) => b[1] - a[1]).slice(0, limit).map(([chunkId, score]) => ({ chunkId, score }));
 }
 
+// packages/core/dist/search/index.js
+init_entity_extractor();
+
 // packages/core/dist/search/adaptive.js
 function createAdaptiveSearch(deps) {
   const { baseSearch } = deps;
@@ -4229,14 +4270,17 @@ function createAdaptiveSearch(deps) {
 var DEFAULT_SIGNAL_WEIGHTS = {
   semantic: 1,
   bm25: 1,
-  entity: 0.5,
-  // conservative: ~20% candidate coverage (arxiv 2508.01405)
+  entity: 1.5,
+  // B2.1: leading curated-graph signal. Per-doc cap in searchEntities
+  // prevents one large note flooding top-k. Tune via STELLAVAULT_W_ENTITY
+  // (e.g. 2.0 for aggressive project-name surfacing, 0.5 for conservative).
   recency: 0.2
   // ±10% bound on relevance
 };
 function createSearchEngine(deps) {
   const { store, embedder, rrfK = 60, getDecayEngine } = deps;
   const baseWeights = { ...DEFAULT_SIGNAL_WEIGHTS, ...deps.weights };
+  const aliasIndex = buildAliasIndex(deps.entityAliases);
   const FETCH_LIMIT = 30;
   return {
     async search(options) {
@@ -4245,7 +4289,7 @@ function createSearchEngine(deps) {
       const [bm25Results, semanticResults, entityResults] = await Promise.all([
         searchBm25(store, query, FETCH_LIMIT),
         searchSemantic(store, embedder, query, FETCH_LIMIT),
-        searchEntities(store, query, FETCH_LIMIT)
+        searchEntities(store, query, FETCH_LIMIT, aliasIndex)
       ]);
       const lists = [semanticResults, bm25Results, entityResults];
       const weights = [w.semantic, w.bm25, w.entity];
@@ -5751,7 +5795,7 @@ function createMcpServer(options) {
   const askTool = createAskTool(searchEngine, vaultPath);
   const generateDraftTool = createGenerateDraftTool(searchEngine, vaultPath);
   const agenticTools = embedder ? createAgenticGraphTools(store, embedder, vaultPath) : [];
-  const server = new Server({ name: "stellavault", version: "0.8.1" }, { capabilities: { tools: {} } });
+  const server = new Server({ name: "stellavault", version: "0.8.3" }, { capabilities: { tools: {} } });
   server.setRequestHandler(ListToolsRequestSchema, async () => ({
     tools: [
       searchToolDef,
@@ -7957,7 +8001,9 @@ function createKnowledgeHub(config, options = {}) {
     embedder,
     rrfK: config.search.rrfK,
     weights: { semantic: sw.semantic, bm25: sw.bm25, entity: sw.entity, recency: sw.recency },
-    getDecayEngine
+    getDecayEngine,
+    entityAliases: config.search.entityAliases
+    // B2.2 — cross-lingual/synonym groups
   });
   const mcpServer = createMcpServer({ store, searchEngine, vaultPath: config.vaultPath, ready: options.ready });
   return { store, embedder, searchEngine, mcpServer, config };
@@ -8094,7 +8140,9 @@ async function searchCommand(query, options, cmd) {
     store,
     embedder,
     rrfK: config.search.rrfK,
-    weights: { semantic: sw.semantic, bm25: sw.bm25, entity: sw.entity, recency: sw.recency }
+    weights: { semantic: sw.semantic, bm25: sw.bm25, entity: sw.entity, recency: sw.recency },
+    entityAliases: config.search.entityAliases
+    // B2.2
   });
   const results = await engine.search({ query, limit });
   await store.close();
@@ -10928,7 +10976,7 @@ if (nodeVersion < 20) {
   process.exit(1);
 }
 var program = new Command();
-var SV_VERSION = true ? "0.8.1" : "0.0.0-dev";
+var SV_VERSION = true ? "0.8.3" : "0.0.0-dev";
 program.name("stellavault").description("Stellavault \u2014 Self-compiling knowledge base for your Obsidian vault").version(SV_VERSION).option("--json", "Output in JSON format (for scripting)").option("--quiet", "Suppress non-essential output");
 program.command("init").description("Interactive setup wizard \u2014 get started in 3 minutes").action(initCommand);
 program.command("doctor").description("Diagnose setup issues (config, vault, DB, model, Node version)").action(doctorCommand);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "stellavault",
-  "version": "0.8.1",
+  "version": "0.8.3",
   "description": "Drop anything. It compiles itself into knowledge. Claude remembers everything you know. Local-first MCP server, vault files never modified.",
   "repository": {
     "type": "git",