stellavault 0.8.1 → 0.8.2

package/dist/stellavault.js

@@ -49,7 +49,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,8 +94,8 @@ var init_config = __esm({
       search: {
         defaultLimit: 10,
         rrfK: 60,
-        weights: { semantic: 1, bm25: 1, entity: 0.5 },
-        // B3 §1.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)
       },
@@ -3910,30 +3910,34 @@ function createSqliteVecStore(dbPath, dimensions = 384) {
         return [];
       const exactPH = entities.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 = [...entities];
+      } 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 = [...entities, ...fuzzy.map((t2) => `%${esc(t2)}%`), ...entities];
+      }
       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) {
@@ -4229,8 +4233,10 @@ 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
 };
@@ -5751,7 +5757,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.2" }, { capabilities: { tools: {} } });
   server.setRequestHandler(ListToolsRequestSchema, async () => ({
     tools: [
       searchToolDef,
@@ -10928,7 +10934,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.2" : "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.2",
   "description": "Drop anything. It compiles itself into knowledge. Claude remembers everything you know. Local-first MCP server, vault files never modified.",
   "repository": {
     "type": "git",