@context-vault/core 2.8.13 → 2.8.15

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@context-vault/core",
-  "version": "2.8.13",
+  "version": "2.8.15",
   "type": "module",
   "description": "Shared core: capture, index, retrieve, tools, and utilities for context-vault",
   "main": "src/index.js",

package/src/index/embed.js

@@ -14,38 +14,51 @@ let extractor = null;
 /** @type {null | true | false} null = unknown, true = working, false = failed */
 let embedAvailable = null;
 
+/** Shared promise for in-flight initialization — prevents concurrent loads */
+let loadingPromise = null;
+
 async function ensurePipeline() {
   if (embedAvailable === false) return null;
   if (extractor) return extractor;
+  if (loadingPromise) return loadingPromise;
 
-  try {
-    // Dynamic import — @huggingface/transformers is optional (its transitive
-    // dep `sharp` can fail to install on some platforms).  When missing, the
-    // server still works with full-text search only.
-    const { pipeline, env } = await import("@huggingface/transformers");
+  loadingPromise = (async () => {
+    try {
+      // Dynamic import — @huggingface/transformers is optional (its transitive
+      // dep `sharp` can fail to install on some platforms).  When missing, the
+      // server still works with full-text search only.
+      const { pipeline, env } = await import("@huggingface/transformers");
 
-    // Redirect model cache to ~/.context-mcp/models/ so it works when the
-    // package is installed globally in a root-owned directory (e.g. /usr/lib/node_modules/).
-    const modelCacheDir = join(homedir(), ".context-mcp", "models");
-    mkdirSync(modelCacheDir, { recursive: true });
-    env.cacheDir = modelCacheDir;
+      // Redirect model cache to ~/.context-mcp/models/ so it works when the
+      // package is installed globally in a root-owned directory (e.g. /usr/lib/node_modules/).
+      const modelCacheDir = join(homedir(), ".context-mcp", "models");
+      mkdirSync(modelCacheDir, { recursive: true });
+      env.cacheDir = modelCacheDir;
 
-    console.error(
-      "[context-vault] Loading embedding model (first run may download ~22MB)...",
-    );
-    extractor = await pipeline("feature-extraction", "Xenova/all-MiniLM-L6-v2");
-    embedAvailable = true;
-    return extractor;
-  } catch (e) {
-    embedAvailable = false;
-    console.error(
-      `[context-vault] Failed to load embedding model: ${e.message}`,
-    );
-    console.error(
-      `[context-vault] Semantic search disabled. Full-text search still works.`,
-    );
-    return null;
-  }
+      console.error(
+        "[context-vault] Loading embedding model (first run may download ~22MB)...",
+      );
+      extractor = await pipeline(
+        "feature-extraction",
+        "Xenova/all-MiniLM-L6-v2",
+      );
+      embedAvailable = true;
+      return extractor;
+    } catch (e) {
+      embedAvailable = false;
+      console.error(
+        `[context-vault] Failed to load embedding model: ${e.message}`,
+      );
+      console.error(
+        `[context-vault] Semantic search disabled. Full-text search still works.`,
+      );
+      return null;
+    } finally {
+      loadingPromise = null;
+    }
+  })();
+
+  return loadingPromise;
 }
 
 export async function embed(text) {
@@ -57,6 +70,7 @@ export async function embed(text) {
   if (!result?.data?.length) {
     extractor = null;
     embedAvailable = null;
+    loadingPromise = null;
     throw new Error("Embedding pipeline returned empty result");
   }
   return new Float32Array(result.data);
@@ -76,6 +90,7 @@ export async function embedBatch(texts) {
   if (!result?.data?.length) {
     extractor = null;
     embedAvailable = null;
+    loadingPromise = null;
     throw new Error("Embedding pipeline returned empty result");
   }
   const dim = result.data.length / texts.length;
@@ -84,15 +99,16 @@ export async function embedBatch(texts) {
       `Unexpected embedding dimension: ${result.data.length} / ${texts.length} = ${dim}`,
     );
   }
-  return texts.map(
-    (_, i) => new Float32Array(result.data.buffer, i * dim * 4, dim),
-  );
+  // subarray() creates a view into result.data's index-space, correctly
+  // accounting for any non-zero byteOffset on the source typed array.
+  return texts.map((_, i) => result.data.subarray(i * dim, (i + 1) * dim));
 }
 
 /** Force re-initialization on next embed call. */
 export function resetEmbedPipeline() {
   extractor = null;
   embedAvailable = null;
+  loadingPromise = null;
 }
 
 /** Check if embedding is currently available. */

package/src/index/index.js

@@ -307,11 +307,16 @@ export async function reindex(ctx, opts = {}) {
             created,
           );
           if (result.changes > 0) {
-            const rowid = ctx.stmts.getRowid.get(id).rowid;
-            const embeddingText = [parsed.title, parsed.body]
-              .filter(Boolean)
-              .join(" ");
-            pendingEmbeds.push({ rowid, text: embeddingText });
+            const rowidResult = ctx.stmts.getRowid.get(id);
+            if (rowidResult?.rowid) {
+              const embeddingText = [parsed.title, parsed.body]
+                .filter(Boolean)
+                .join(" ");
+              pendingEmbeds.push({
+                rowid: rowidResult.rowid,
+                text: embeddingText,
+              });
+            }
             stats.added++;
           } else {
             stats.unchanged++;

package/src/retrieve/index.js

@@ -157,9 +157,9 @@ export async function hybridSearch(
             : 15;
         const vecRows = ctx.db
           .prepare(
-            `SELECT v.rowid, v.distance FROM vault_vec v WHERE embedding MATCH ? ORDER BY distance LIMIT ${vecLimit}`,
+            `SELECT v.rowid, v.distance FROM vault_vec v WHERE embedding MATCH ? ORDER BY distance LIMIT ?`,
           )
-          .all(queryVec);
+          .all(queryVec, vecLimit);
 
         if (vecRows.length) {
           // Batch hydration: single query instead of N+1

package/src/server/tools/delete-context.js

@@ -32,10 +32,16 @@ export async function handler({ id }, ctx, { ensureIndexed }) {
   }
 
   // Delete file from disk first (source of truth)
+  let fileWarning = null;
   if (entry.file_path) {
     try {
       unlinkSync(entry.file_path);
-    } catch {}
+    } catch (e) {
+      // ENOENT = already gone — not an error worth surfacing
+      if (e.code !== "ENOENT") {
+        fileWarning = `file could not be removed from disk (${e.code}): ${entry.file_path}`;
+      }
+    }
   }
 
   // Delete vector embedding
@@ -49,5 +55,6 @@ export async function handler({ id }, ctx, { ensureIndexed }) {
   // Delete DB row (FTS trigger handles FTS cleanup)
   ctx.stmts.deleteEntry.run(id);
 
-  return ok(`Deleted ${entry.kind}: ${entry.title || "(untitled)"} [${id}]`);
+  const msg = `Deleted ${entry.kind}: ${entry.title || "(untitled)"} [${id}]`;
+  return ok(fileWarning ? `${msg}\nWarning: ${fileWarning}` : msg);
 }

package/src/server/tools.js

@@ -28,9 +28,11 @@ export function registerTools(server, ctx) {
     return async (...args) => {
       if (ctx.activeOps) ctx.activeOps.count++;
       let timer;
+      let handlerPromise;
       try {
+        handlerPromise = Promise.resolve(handler(...args));
         return await Promise.race([
-          Promise.resolve(handler(...args)),
+          handlerPromise,
           new Promise((_, reject) => {
             timer = setTimeout(
               () => reject(new Error("TOOL_TIMEOUT")),
@@ -40,6 +42,9 @@ export function registerTools(server, ctx) {
         ]);
       } catch (e) {
         if (e.message === "TOOL_TIMEOUT") {
+          // Suppress any late rejection from the still-running handler to
+          // prevent unhandled promise rejection warnings in the host process.
+          handlerPromise?.catch(() => {});
           return err(
             "Tool timed out after 60s. Try a simpler query or run `context-vault reindex` first.",
             "TIMEOUT",