context-vault 2.8.12 → 2.8.14

package/bin/cli.js

@@ -14,6 +14,7 @@ import { createInterface } from "node:readline";
 import {
   existsSync,
   statSync,
+  readdirSync,
   readFileSync,
   writeFileSync,
   mkdirSync,
@@ -455,6 +456,9 @@ async function runSetup() {
     if (create.toLowerCase() !== "n") {
       mkdirSync(resolvedVaultDir, { recursive: true });
       console.log(`  ${green("+")} Created ${resolvedVaultDir}`);
+    } else {
+      console.log(red("\n  Setup cancelled — vault directory is required."));
+      process.exit(1);
     }
   }
 
@@ -506,16 +510,45 @@ async function runSetup() {
       const spinnerFrames = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"];
       let frame = 0;
       const start = Date.now();
+      const modelDir = join(homedir(), ".context-mcp", "models");
       const spinner = setInterval(() => {
         const elapsed = ((Date.now() - start) / 1000).toFixed(0);
+        let downloadedMB = "?";
+        try {
+          const files = readdirSync(modelDir, {
+            recursive: true,
+            withFileTypes: true,
+          });
+          const totalBytes = files
+            .filter((f) => f.isFile())
+            .reduce(
+              (sum, f) =>
+                sum + statSync(join(f.parentPath ?? f.path, f.name)).size,
+              0,
+            );
+          downloadedMB = (totalBytes / 1024 / 1024).toFixed(1);
+        } catch {}
         process.stdout.write(
-          `\r  ${spinnerFrames[frame++ % spinnerFrames.length]} Downloading... ${dim(`${elapsed}s`)}`,
+          `\r  ${spinnerFrames[frame++ % spinnerFrames.length]} Downloading... ${downloadedMB} MB / ~22 MB  ${dim(`${elapsed}s`)}`,
         );
       }, 100);
 
       try {
         const { embed } = await import("@context-vault/core/index/embed");
-        await embed("warmup");
+        let timeoutHandle;
+        const timeout = new Promise((_, reject) => {
+          timeoutHandle = setTimeout(
+            () =>
+              reject(
+                Object.assign(new Error("Download timed out after 90s"), {
+                  code: "ETIMEDOUT",
+                }),
+              ),
+            90_000,
+          );
+        });
+        await Promise.race([embed("warmup"), timeout]);
+        clearTimeout(timeoutHandle);
 
         clearInterval(spinner);
         process.stdout.write(

package/node_modules/@context-vault/core/package.json

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

package/node_modules/@context-vault/core/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;
@@ -93,6 +108,7 @@ export async function embedBatch(texts) {
 export function resetEmbedPipeline() {
   extractor = null;
   embedAvailable = null;
+  loadingPromise = null;
 }
 
 /** Check if embedding is currently available. */

package/node_modules/@context-vault/core/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/node_modules/@context-vault/core/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/package.json

@@ -1,6 +1,6 @@
 {
   "name": "context-vault",
-  "version": "2.8.12",
+  "version": "2.8.14",
   "type": "module",
   "description": "Persistent memory for AI agents — saves and searches knowledge across sessions",
   "bin": {
@@ -55,7 +55,7 @@
     "@context-vault/core"
   ],
   "dependencies": {
-    "@context-vault/core": "^2.8.12",
+    "@context-vault/core": "^2.8.14",
     "@modelcontextprotocol/sdk": "^1.26.0",
     "better-sqlite3": "^12.6.2",
     "sqlite-vec": "^0.1.0"