context-vault 2.4.0 → 2.4.2

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

@@ -1,6 +1,6 @@
 {
   "name": "@context-vault/core",
-  "version": "2.4.0",
+  "version": "2.4.2",
   "type": "module",
   "description": "Shared core: capture, index, retrieve, tools, and utilities for context-vault",
   "main": "src/index.js",
@@ -21,16 +21,18 @@
     "./core/frontmatter": "./src/core/frontmatter.js",
     "./core/status": "./src/core/status.js"
   },
-  "files": ["src/"],
+  "files": [
+    "src/"
+  ],
   "license": "MIT",
-  "engines": { "node": ">=20" },
+  "engines": {
+    "node": ">=20"
+  },
   "author": "Felix Hellstrom",
-  "repository": { "type": "git", "url": "git+https://github.com/fellanH/context-mcp.git", "directory": "packages/core" },
-  "homepage": "https://github.com/fellanH/context-mcp",
-  "dependencies": {
-    "@huggingface/transformers": "^3.0.0",
-    "@modelcontextprotocol/sdk": "^1.26.0",
-    "better-sqlite3": "^12.6.2",
-    "sqlite-vec": "^0.1.0"
-  }
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/fellanH/context-mcp.git",
+    "directory": "packages/core"
+  },
+  "homepage": "https://github.com/fellanH/context-mcp"
 }

package/node_modules/@context-vault/core/src/core/config.js

@@ -4,7 +4,7 @@
  * Resolution chain (highest priority last):
  *   1. Convention defaults
  *   2. Config file (~/.context-mcp/config.json)
- *   3. Environment variables (CONTEXT_MCP_*)
+ *   3. Environment variables (CONTEXT_VAULT_* or CONTEXT_MCP_*)
  *   4. CLI arguments
  */
 
@@ -29,7 +29,7 @@ export function resolveConfig() {
   const cliArgs = parseArgs(process.argv);
 
   // 1. Convention defaults
-  const dataDir = resolve(cliArgs.dataDir || process.env.CONTEXT_MCP_DATA_DIR || join(HOME, ".context-mcp"));
+  const dataDir = resolve(cliArgs.dataDir || process.env.CONTEXT_VAULT_DATA_DIR || process.env.CONTEXT_MCP_DATA_DIR || join(HOME, ".context-mcp"));
   const config = {
     vaultDir: join(HOME, "vault"),
     dataDir,
@@ -51,16 +51,16 @@ export function resolveConfig() {
       if (fc.eventDecayDays) config.eventDecayDays = fc.eventDecayDays;
       config.resolvedFrom = "config file";
     } catch (e) {
-      throw new Error(`[context-mcp] Invalid config at ${configPath}: ${e.message}`);
+      throw new Error(`[context-vault] Invalid config at ${configPath}: ${e.message}`);
     }
   }
   config.configPath = configPath;
 
-  // 3. Environment variable overrides
-  if (process.env.CONTEXT_MCP_VAULT_DIR) { config.vaultDir = process.env.CONTEXT_MCP_VAULT_DIR; config.resolvedFrom = "env"; }
-  if (process.env.CONTEXT_MCP_DB_PATH) { config.dbPath = process.env.CONTEXT_MCP_DB_PATH; config.resolvedFrom = "env"; }
-  if (process.env.CONTEXT_MCP_DEV_DIR) { config.devDir = process.env.CONTEXT_MCP_DEV_DIR; config.resolvedFrom = "env"; }
-  if (process.env.CONTEXT_MCP_EVENT_DECAY_DAYS) { config.eventDecayDays = Number(process.env.CONTEXT_MCP_EVENT_DECAY_DAYS); config.resolvedFrom = "env"; }
+  // 3. Environment variable overrides (CONTEXT_VAULT_* takes priority over CONTEXT_MCP_*)
+  if (process.env.CONTEXT_VAULT_VAULT_DIR || process.env.CONTEXT_MCP_VAULT_DIR) { config.vaultDir = process.env.CONTEXT_VAULT_VAULT_DIR || process.env.CONTEXT_MCP_VAULT_DIR; config.resolvedFrom = "env"; }
+  if (process.env.CONTEXT_VAULT_DB_PATH || process.env.CONTEXT_MCP_DB_PATH) { config.dbPath = process.env.CONTEXT_VAULT_DB_PATH || process.env.CONTEXT_MCP_DB_PATH; config.resolvedFrom = "env"; }
+  if (process.env.CONTEXT_VAULT_DEV_DIR || process.env.CONTEXT_MCP_DEV_DIR) { config.devDir = process.env.CONTEXT_VAULT_DEV_DIR || process.env.CONTEXT_MCP_DEV_DIR; config.resolvedFrom = "env"; }
+  if (process.env.CONTEXT_VAULT_EVENT_DECAY_DAYS || process.env.CONTEXT_MCP_EVENT_DECAY_DAYS) { config.eventDecayDays = Number(process.env.CONTEXT_VAULT_EVENT_DECAY_DAYS || process.env.CONTEXT_MCP_EVENT_DECAY_DAYS); config.resolvedFrom = "env"; }
 
   // 4. CLI arg overrides (highest priority)
   if (cliArgs.vaultDir) { config.vaultDir = cliArgs.vaultDir; config.resolvedFrom = "CLI args"; }

package/node_modules/@context-vault/core/src/index/db.js

@@ -136,11 +136,7 @@ export async function initDatabase(dbPath) {
     try {
       sqliteVec.load(db);
     } catch (e) {
-      console.error(`[context-mcp] Failed to load sqlite-vec native module.`);
-      console.error(`[context-mcp] This usually means prebuilt binaries aren't available for your platform.`);
-      console.error(`[context-mcp] Try: npm rebuild sqlite-vec`);
-      console.error(`[context-mcp] Error: ${e.message}`);
-      throw e;
+      throw new NativeModuleError(e);
     }
     return db;
   }
@@ -150,7 +146,7 @@ export async function initDatabase(dbPath) {
 
   // Enforce fresh-DB-only — old schemas get a full rebuild (with backup)
   if (version > 0 && version < 5) {
-    console.error(`[context-mcp] Schema v${version} is outdated. Rebuilding database...`);
+    console.error(`[context-vault] Schema v${version} is outdated. Rebuilding database...`);
 
     // Backup old DB before destroying it
     const backupPath = `${dbPath}.v${version}.backup`;
@@ -158,10 +154,10 @@ export async function initDatabase(dbPath) {
       db.close();
       if (existsSync(dbPath)) {
         copyFileSync(dbPath, backupPath);
-        console.error(`[context-mcp] Backed up old database to: ${backupPath}`);
+        console.error(`[context-vault] Backed up old database to: ${backupPath}`);
       }
     } catch (backupErr) {
-      console.error(`[context-mcp] Warning: could not backup old database: ${backupErr.message}`);
+      console.error(`[context-vault] Warning: could not backup old database: ${backupErr.message}`);
     }
 
     unlinkSync(dbPath);
@@ -222,7 +218,7 @@ export function prepareStatements(db) {
   } catch (e) {
     throw new Error(
       `Failed to prepare database statements. The database may be corrupted.\n` +
-      `Try deleting and rebuilding: rm "${db.name}" && context-mcp reindex\n` +
+      `Try deleting and rebuilding: rm "${db.name}" && context-vault reindex\n` +
       `Original error: ${e.message}`
     );
   }

package/node_modules/@context-vault/core/src/index/embed.js

@@ -5,17 +5,10 @@
  * disk issues), semantic search is disabled but FTS still works.
  */
 
-import { pipeline, env } from "@huggingface/transformers";
 import { join } from "node:path";
 import { homedir } from "node:os";
 import { mkdirSync } from "node:fs";
 
-// 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;
-
 let extractor = null;
 
 /** @type {null | true | false} null = unknown, true = working, false = failed */
@@ -26,14 +19,25 @@ async function ensurePipeline() {
   if (extractor) return extractor;
 
   try {
-    console.error("[context-mcp] Loading embedding model (first run may download ~22MB)...");
+    // 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;
+
+    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-mcp] Failed to load embedding model: ${e.message}`);
-    console.error(`[context-mcp] Semantic search disabled. Full-text search still works.`);
+    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;
   }
 }

package/node_modules/@context-vault/core/src/server/tools.js

@@ -16,6 +16,7 @@ import { gatherVaultStatus } from "../core/status.js";
 import { categoryFor } from "../core/categories.js";
 import { normalizeKind } from "../core/files.js";
 import { ok, err, ensureVaultExists, ensureValidKind } from "./helpers.js";
+import { isEmbedAvailable } from "../index/embed.js";
 
 /**
  * Register all MCP tools on the server.
@@ -23,10 +24,37 @@ import { ok, err, ensureVaultExists, ensureValidKind } from "./helpers.js";
  * @param {import("@modelcontextprotocol/sdk/server/mcp.js").McpServer} server
  * @param {{ db, config, stmts, embed, insertVec, deleteVec }} ctx
  */
+const TOOL_TIMEOUT_MS = 60_000;
+
 export function registerTools(server, ctx) {
   const { config } = ctx;
   const userId = ctx.userId !== undefined ? ctx.userId : undefined;
 
+  // ─── Tool wrapper: tracks in-flight ops for graceful shutdown + timeout ────
+
+  function tracked(handler) {
+    return async (...args) => {
+      if (ctx.activeOps) ctx.activeOps.count++;
+      let timer;
+      try {
+        return await Promise.race([
+          Promise.resolve(handler(...args)),
+          new Promise((_, reject) => {
+            timer = setTimeout(() => reject(new Error("TOOL_TIMEOUT")), TOOL_TIMEOUT_MS);
+          }),
+        ]);
+      } catch (e) {
+        if (e.message === "TOOL_TIMEOUT") {
+          return err("Tool timed out after 60s. Try a simpler query or run `context-vault reindex` first.", "TIMEOUT");
+        }
+        throw e;
+      } finally {
+        clearTimeout(timer);
+        if (ctx.activeOps) ctx.activeOps.count--;
+      }
+    };
+  }
+
   // ─── Auto-Reindex (runs once per session, on first tool call) ──────────────
 
   // In hosted mode, skip reindex — DB is always in sync via writeEntry→indexEntry
@@ -44,14 +72,14 @@ export function registerTools(server, ctx) {
         reindexDone = true;
         const total = stats.added + stats.updated + stats.removed;
         if (total > 0) {
-          console.error(`[context-mcp] Auto-reindex: +${stats.added} ~${stats.updated} -${stats.removed} (${stats.unchanged} unchanged)`);
+          console.error(`[context-vault] Auto-reindex: +${stats.added} ~${stats.updated} -${stats.removed} (${stats.unchanged} unchanged)`);
         }
       })
       .catch((e) => {
         reindexAttempts++;
-        console.error(`[context-mcp] Auto-reindex failed (attempt ${reindexAttempts}/${MAX_REINDEX_ATTEMPTS}): ${e.message}`);
+        console.error(`[context-vault] Auto-reindex failed (attempt ${reindexAttempts}/${MAX_REINDEX_ATTEMPTS}): ${e.message}`);
         if (reindexAttempts >= MAX_REINDEX_ATTEMPTS) {
-          console.error(`[context-mcp] Giving up on auto-reindex. Run \`context-mcp reindex\` manually to diagnose.`);
+          console.error(`[context-vault] Giving up on auto-reindex. Run \`context-vault reindex\` manually to diagnose.`);
           reindexDone = true;
           reindexFailed = true;
         } else {
@@ -76,7 +104,7 @@ export function registerTools(server, ctx) {
       until: z.string().optional().describe("ISO date, return entries created before this"),
       limit: z.number().optional().describe("Max results to return (default 10)"),
     },
-    async ({ query, kind, category, identity_key, tags, since, until, limit }) => {
+    tracked(async ({ query, kind, category, identity_key, tags, since, until, limit }) => {
       const hasQuery = query?.trim();
       const hasFilters = kind || category || tags?.length || since || until || identity_key;
       if (!hasQuery && !hasFilters) return err("Required: query or at least one filter (kind, category, tags, since, until, identity_key)", "INVALID_INPUT");
@@ -175,7 +203,8 @@ export function registerTools(server, ctx) {
       }
 
       const lines = [];
-      if (reindexFailed) lines.push(`> **Warning:** Auto-reindex failed. Results may be stale. Run \`context-mcp reindex\` to fix.\n`);
+      if (reindexFailed) lines.push(`> **Warning:** Auto-reindex failed. Results may be stale. Run \`context-vault reindex\` to fix.\n`);
+      if (hasQuery && isEmbedAvailable() === false) lines.push(`> **Note:** Semantic search unavailable — results ranked by keyword match only. Run \`context-vault setup\` to download the embedding model.\n`);
       const heading = hasQuery ? `Results for "${query}"` : "Filtered entries";
       lines.push(`## ${heading} (${filtered.length} matches)\n`);
       for (let i = 0; i < filtered.length; i++) {
@@ -192,7 +221,7 @@ export function registerTools(server, ctx) {
         lines.push(`_Showing events from last ${config.eventDecayDays || 30} days. Use since/until for custom range._`);
       }
       return ok(lines.join("\n"));
-    }
+    })
   );
 
   // ─── save_context (write / update) ────────────────────────────────────────
@@ -212,7 +241,7 @@ export function registerTools(server, ctx) {
       identity_key: z.string().optional().describe("Required for entity kinds (contact, project, tool, source). The unique identifier for this entity."),
       expires_at: z.string().optional().describe("ISO date for TTL expiry"),
     },
-    async ({ id, kind, title, body, tags, meta, folder, source, identity_key, expires_at }) => {
+    tracked(async ({ id, kind, title, body, tags, meta, folder, source, identity_key, expires_at }) => {
       const vaultErr = ensureVaultExists(config);
       if (vaultErr) return vaultErr;
 
@@ -264,6 +293,17 @@ export function registerTools(server, ctx) {
         return err(`Entity kind "${kind}" requires identity_key`, "MISSING_IDENTITY_KEY");
       }
 
+      // Hosted tier limit enforcement (skipped in local mode — no checkLimits on ctx)
+      if (ctx.checkLimits) {
+        const usage = ctx.checkLimits();
+        if (usage.entryCount >= usage.maxEntries) {
+          return err(`Entry limit reached (${usage.maxEntries}). Upgrade to Pro for unlimited entries.`, "LIMIT_EXCEEDED");
+        }
+        if (usage.storageMb >= usage.maxStorageMb) {
+          return err(`Storage limit reached (${usage.maxStorageMb} MB). Upgrade to Pro for more storage.`, "LIMIT_EXCEEDED");
+        }
+      }
+
       await ensureIndexed();
 
       const mergedMeta = { ...(meta || {}) };
@@ -277,7 +317,7 @@ export function registerTools(server, ctx) {
       if (tags?.length) parts.push(`  tags: ${tags.join(", ")}`);
       parts.push("", "_Use this id to update or delete later._");
       return ok(parts.join("\n"));
-    }
+    })
   );
 
   // ─── list_context (browse) ────────────────────────────────────────────────
@@ -294,7 +334,7 @@ export function registerTools(server, ctx) {
       limit: z.number().optional().describe("Max results to return (default 20, max 100)"),
       offset: z.number().optional().describe("Skip first N results for pagination"),
     },
-    async ({ kind, category, tags, since, until, limit, offset }) => {
+    tracked(async ({ kind, category, tags, since, until, limit, offset }) => {
       await ensureIndexed();
 
       const clauses = [];
@@ -342,7 +382,9 @@ export function registerTools(server, ctx) {
 
       if (!filtered.length) return ok("No entries found matching the given filters.");
 
-      const lines = [`## Vault Entries (${filtered.length} shown, ${total} total)\n`];
+      const lines = [];
+      if (reindexFailed) lines.push(`> **Warning:** Auto-reindex failed. Results may be stale. Run \`context-vault reindex\` to fix.\n`);
+      lines.push(`## Vault Entries (${filtered.length} shown, ${total} total)\n`);
       for (const r of filtered) {
         const entryTags = r.tags ? JSON.parse(r.tags) : [];
         const tagStr = entryTags.length ? entryTags.join(", ") : "none";
@@ -355,7 +397,7 @@ export function registerTools(server, ctx) {
       }
 
       return ok(lines.join("\n"));
-    }
+    })
   );
 
   // ─── delete_context (remove) ──────────────────────────────────────────────
@@ -366,7 +408,7 @@ export function registerTools(server, ctx) {
     {
       id: z.string().describe("The entry ULID to delete"),
     },
-    async ({ id }) => {
+    tracked(async ({ id }) => {
       if (!id?.trim()) return err("Required: id (non-empty string)", "INVALID_INPUT");
       await ensureIndexed();
 
@@ -393,7 +435,7 @@ export function registerTools(server, ctx) {
       ctx.stmts.deleteEntry.run(id);
 
       return ok(`Deleted ${entry.kind}: ${entry.title || "(untitled)"} [${id}]`);
-    }
+    })
   );
 
   // ─── submit_feedback (bug/feature reports) ────────────────────────────────
@@ -407,7 +449,7 @@ export function registerTools(server, ctx) {
       body: z.string().describe("Detailed description"),
       severity: z.enum(["low", "medium", "high"]).optional().describe("Severity level (default: medium)"),
     },
-    async ({ type, title, body, severity }) => {
+    tracked(async ({ type, title, body, severity }) => {
       const vaultErr = ensureVaultExists(config);
       if (vaultErr) return vaultErr;
 
@@ -430,7 +472,7 @@ export function registerTools(server, ctx) {
 
       const relPath = entry.filePath ? entry.filePath.replace(config.vaultDir + "/", "") : entry.filePath;
       return ok(`Feedback submitted: ${type} [${effectiveSeverity}] → ${relPath}\n  id: ${entry.id}\n  title: ${title}`);
-    }
+    })
   );
 
   // ─── context_status (diagnostics) ──────────────────────────────────────────
@@ -446,7 +488,7 @@ export function registerTools(server, ctx) {
       const healthIcon = hasIssues ? "⚠" : "✓";
 
       const lines = [
-        `## ${healthIcon} Vault Status`,
+        `## ${healthIcon} Vault Status (connected)`,
         ``,
         `Vault:     ${config.vaultDir} (${config.vaultDirExists ? status.fileCount + " files" : "missing"})`,
         `Database:  ${config.dbPath} (${status.dbSize})`,
@@ -501,9 +543,9 @@ export function registerTools(server, ctx) {
 
       // Suggested actions
       const actions = [];
-      if (status.stalePaths) actions.push("- Run `context-mcp reindex` to fix stale paths");
-      if (status.embeddingStatus?.missing > 0) actions.push("- Run `context-mcp reindex` to generate missing embeddings");
-      if (!config.vaultDirExists) actions.push("- Run `context-mcp setup` to create the vault directory");
+      if (status.stalePaths) actions.push("- Run `context-vault reindex` to fix stale paths");
+      if (status.embeddingStatus?.missing > 0) actions.push("- Run `context-vault reindex` to generate missing embeddings");
+      if (!config.vaultDirExists) actions.push("- Run `context-vault setup` to create the vault directory");
       if (status.kindCounts.length === 0 && config.vaultDirExists) actions.push("- Use `save_context` to add your first entry");
 
       if (actions.length) {

package/package.json

@@ -1,9 +1,10 @@
 {
   "name": "context-vault",
-  "version": "2.4.0",
+  "version": "2.4.2",
   "type": "module",
   "description": "Persistent memory for AI agents — saves and searches knowledge across sessions",
   "bin": {
+    "context-vault": "bin/cli.js",
     "context-mcp": "bin/cli.js"
   },
   "main": "src/server/index.js",
@@ -19,19 +20,44 @@
     "bin/",
     "src/",
     "scripts/",
-    "ui/",
     "README.md",
     "LICENSE"
   ],
   "license": "MIT",
-  "engines": { "node": ">=20" },
+  "engines": {
+    "node": ">=20"
+  },
   "author": "Felix Hellstrom",
-  "repository": { "type": "git", "url": "git+https://github.com/fellanH/context-mcp.git" },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/fellanH/context-mcp.git"
+  },
   "homepage": "https://github.com/fellanH/context-mcp",
-  "bugs": { "url": "https://github.com/fellanH/context-mcp/issues" },
-  "keywords": ["mcp", "model-context-protocol", "ai", "knowledge-base", "knowledge-management", "vault", "rag", "sqlite", "embeddings", "claude", "cursor", "cline", "windsurf"],
-  "bundledDependencies": ["@context-vault/core"],
+  "bugs": {
+    "url": "https://github.com/fellanH/context-mcp/issues"
+  },
+  "keywords": [
+    "mcp",
+    "model-context-protocol",
+    "ai",
+    "knowledge-base",
+    "knowledge-management",
+    "vault",
+    "rag",
+    "sqlite",
+    "embeddings",
+    "claude",
+    "cursor",
+    "cline",
+    "windsurf"
+  ],
+  "bundledDependencies": [
+    "@context-vault/core"
+  ],
   "dependencies": {
-    "@context-vault/core": "^2.4.0"
+    "@context-vault/core": "^2.4.2",
+    "@modelcontextprotocol/sdk": "^1.26.0",
+    "better-sqlite3": "^12.6.2",
+    "sqlite-vec": "^0.1.0"
   }
 }