pi-mega-compact 0.4.24 → 0.4.25

package/dist/src/store/vectorIndex.test.js

@@ -0,0 +1,99 @@
+/**
+ * vectorIndex.test.ts — Slice 2 async PGlite/HNSW vector index.
+ *
+ * Proves: cross-repo nearest-neighbor recall, repoId scoping, the dimension
+ * guard (non-512 vectors skipped, never corrupt the index), and graceful
+ * degradation when the index is disabled (kill-switch) — all without touching
+ * the synchronous node:sqlite store.
+ *
+ * The index is a WASM Postgres (PGlite) — fully local, zero network
+ * (PREVENT-PI-004). Each test isolates state via MEGACOMPACT_VECTOR_INDEX_DIR.
+ */
+import { test } from "node:test";
+import assert from "node:assert/strict";
+import { mkdtempSync, rmSync } from "node:fs";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+import { EMBEDDING_DIM, initVectorIndex, upsertEmbedding, searchAsync, closeVectorIndex, isVectorIndexDisabled, } from "./vectorIndex.js";
+/** A 512-dim unit-ish vector with a single spike at `idx` (deterministic NN). */
+function spikeVec(idx, magnitude = 1) {
+    const v = new Array(EMBEDDING_DIM).fill(0);
+    v[idx % EMBEDDING_DIM] = magnitude;
+    return v;
+}
+function isolateIndexDir() {
+    const dir = mkdtempSync(join(tmpdir(), "mc-vecidx-"));
+    process.env.MEGACOMPACT_VECTOR_INDEX_DIR = dir;
+    return dir;
+}
+test("cross-repo HNSW nearest-neighbor recall across repos + repoId scoping", async () => {
+    delete process.env.MEGACOMPACT_PGLITE_DISABLED;
+    const dir = isolateIndexDir();
+    try {
+        await closeVectorIndex(); // ensure a fresh singleton for this dir
+        const pg = await initVectorIndex();
+        assert.ok(pg, "index should initialize (PGlite WASM available)");
+        // repoA: two checkpoints; repoB: one checkpoint. Distinct spike directions.
+        await upsertEmbedding("/repoA/.pi/mega-compact", "sessA", "chkpt_001", spikeVec(0));
+        await upsertEmbedding("/repoA/.pi/mega-compact", "sessA", "chkpt_002", spikeVec(5));
+        await upsertEmbedding("/repoB/.pi/mega-compact", "sessB", "chkpt_001", spikeVec(0));
+        // Cross-repo query near spike(0): nearest are the two spike(0) rows, one per repo.
+        const cross = await searchAsync(spikeVec(0), { k: 2 });
+        assert.equal(cross.length, 2, "cross-repo returns two nearest");
+        const repos = new Set(cross.map((h) => h.repoId));
+        assert.ok(repos.has("/repoA/.pi/mega-compact"), "hit from repoA");
+        assert.ok(repos.has("/repoB/.pi/mega-compact"), "hit from repoB");
+        assert.ok(cross[0].score > 0.99, "top hit is near-identical (cosine ~1)");
+        // Scoped to repoA only: excludes repoB even though repoB has an identical vec.
+        const scoped = await searchAsync(spikeVec(0), { k: 5, repoId: "/repoA/.pi/mega-compact" });
+        assert.ok(scoped.length >= 1, "scoped returns repoA hits");
+        assert.ok(scoped.every((h) => h.repoId === "/repoA/.pi/mega-compact"), "repoId filter excludes other repos");
+    }
+    finally {
+        await closeVectorIndex();
+        rmSync(dir, { recursive: true, force: true });
+        delete process.env.MEGACOMPACT_VECTOR_INDEX_DIR;
+    }
+});
+test("dimension guard: non-512 vectors are skipped, never corrupt the index", async () => {
+    delete process.env.MEGACOMPACT_PGLITE_DISABLED;
+    const dir = isolateIndexDir();
+    try {
+        await closeVectorIndex();
+        await initVectorIndex();
+        // Wrong-dimension vector (BYO embedder mismatch) must be silently skipped.
+        await upsertEmbedding("/repoC/.pi/mega-compact", "sessC", "chkpt_001", [1, 2, 3]);
+        const hits = await searchAsync(spikeVec(0), { k: 5 });
+        assert.equal(hits.length, 0, "no rows stored for a mismatched-dim vector");
+        // A correct-dim vector still stores fine afterward (index not corrupted).
+        await upsertEmbedding("/repoC/.pi/mega-compact", "sessC", "chkpt_002", spikeVec(3));
+        const ok = await searchAsync(spikeVec(3), { k: 1 });
+        assert.equal(ok.length, 1, "valid vector stored after a skipped one");
+        assert.equal(ok[0].checkpointId, "chkpt_002");
+    }
+    finally {
+        await closeVectorIndex();
+        rmSync(dir, { recursive: true, force: true });
+        delete process.env.MEGACOMPACT_VECTOR_INDEX_DIR;
+    }
+});
+test("kill-switch: MEGACOMPACT_PGLITE_DISABLED disables the index gracefully", async () => {
+    const dir = isolateIndexDir();
+    process.env.MEGACOMPACT_PGLITE_DISABLED = "true";
+    try {
+        await closeVectorIndex();
+        assert.equal(isVectorIndexDisabled(), true, "kill-switch reported disabled");
+        const pg = await initVectorIndex();
+        assert.equal(pg, undefined, "init returns undefined when disabled");
+        // Upsert + search are no-ops that never throw and return empty.
+        await upsertEmbedding("/repoD/.pi/mega-compact", "sessD", "chkpt_001", spikeVec(0));
+        const hits = await searchAsync(spikeVec(0), { k: 3 });
+        assert.deepEqual(hits, [], "search returns [] when disabled");
+    }
+    finally {
+        delete process.env.MEGACOMPACT_PGLITE_DISABLED;
+        await closeVectorIndex();
+        rmSync(dir, { recursive: true, force: true });
+        delete process.env.MEGACOMPACT_VECTOR_INDEX_DIR;
+    }
+});

package/dist/src/vectorStore.js

@@ -19,7 +19,8 @@ import { isNearDuplicate } from "./dedup/l1-verify.js";
 import { mmrRerank } from "./dedup/mmr.js";
 import { topK } from "./dedup/topk.js";
 import { openBloom, saveBloom } from "./store/bloom.js";
-import { listCheckpoints, nextCheckpointId, upsertCheckpoint, loadSessionState, saveSessionState, upsertMinhashSignature, insertLshBuckets, lshCandidateChunks, setDedupStatus, addTokensSaved, getDedupStats, bumpDedupStats, repoStats as repoStatsFromStore, dataInvariantStats, } from "./store/sqlite.js";
+import { listCheckpoints, nextCheckpointId, upsertCheckpoint, getCheckpoint, loadSessionState, saveSessionState, upsertMinhashSignature, insertLshBuckets, lshCandidateChunks, setDedupStatus, addTokensSaved, getDedupStats, bumpDedupStats, repoStats as repoStatsFromStore, dataInvariantStats, } from "./store/sqlite.js";
+import { upsertEmbedding as indexUpsertEmbedding, initVectorIndex, searchAsync as vectorIndexSearch, } from "./store/vectorIndex.js";
 import { rehydrateRaptorTree } from "./dedup/raptor/index.js";
 import { stagedExpansion } from "./dedup/raptor/retrieval.js";
 import { migrateJsonToSqlite } from "./store/migrate.js";
@@ -39,9 +40,17 @@ export class VectorStore {
     cfg;
     /** Optional monitoring target (Sprint 14). Undefined → no monitoring. */
     eventsPath;
+    /**
+     * Repo key for the async PGlite vector index (Slice 2). We use the stateDir
+     * itself as the repo id — it is already unique per repo and available here
+     * without crossing into the pi-runtime layer (src/ stays pi-agnostic). The
+     * global index keys on repoId so recall can span repos.
+     */
+    repoId;
     constructor(opts = {}) {
         this.embedder = opts.embedder ?? defaultEmbedder();
         this.stateDir = opts.stateDir ?? getStateDir();
+        this.repoId = opts.repoId ?? this.stateDir;
         // Sprint 14: all tier flags/thresholds flow from the single config source
         // (DedupConfig). The legacy opts.dedupSim / opts.l2Enabled remain accepted
         // for backward-compat callers but flags are authoritative via `cfg`.
@@ -283,6 +292,12 @@ export class VectorStore {
         // Cumulative store-wide dedup accounting (attempt, not collapsed).
         bumpDedupStats(false, this.stateDir);
         onTier?.({ tier: "new", status: "stored" });
+        // Slice 2: best-effort, fire-and-forget mirror of this new checkpoint into
+        // the async global PGlite/HNSW index. NEVER awaited — must not block or
+        // throw into the synchronous add() path. On failure the index degrades to
+        // the sync scan (handled inside vectorIndex). The node:sqlite store remains
+        // authoritative; the index is rebuildable from it at any time.
+        void indexUpsertEmbedding(this.repoId, sessionId, checkpointId, checkpoint.embedding);
         return { checkpoint, deduped: false };
     }
     /**
@@ -370,6 +385,56 @@ export class VectorStore {
         const ranked = mmrRerank(mmrItems, k, this.cfg.MMR_LAMBDA);
         return ranked;
     }
+    /**
+     * Slice 2: async cross-repo (or single-repo) recall via the PGlite/HNSW index.
+     *
+     * This is the ONLY async recall surface and is a BONUS path — the synchronous
+     * `search()` above remains the default. `opts.repoId` scopes to one repo; omit
+     * it for cross-repo nearest-neighbor recall (the headline capability the sync
+     * per-session scan cannot provide).
+     *
+     * Best-effort: if the index is disabled/empty/failing, we fall back to the
+     * synchronous per-session `search()` for THIS repo so callers always get a
+     * sensible result. Hydrates each hit's StoredCheckpoint from the authoritative
+     * node:sqlite store (the hit's repoId doubles as that repo's stateDir), then
+     * MMR-dedupes the merged set.
+     */
+    async searchAsync(sessionId, query, k = 3, opts = {}) {
+        const sid = normalizeSessionId(sessionId);
+        const qv = this.embedder.embed(query);
+        // repoId filter: explicit opts.repoId wins; else this repo unless crossRepo.
+        const repoId = opts.repoId ?? (opts.crossRepo ? undefined : this.repoId);
+        let indexHits = [];
+        try {
+            await initVectorIndex();
+            indexHits = await vectorIndexSearch(qv, { k: Math.max(k * 2, k), repoId });
+        }
+        catch {
+            indexHits = [];
+        }
+        if (indexHits.length === 0) {
+            // Index empty/unavailable → synchronous per-session fallback (this repo).
+            return this.search(sid, query, k);
+        }
+        // Hydrate each index hit from the authoritative node:sqlite store. repoId is
+        // that repo's stateDir, so cross-repo hits resolve against their own store.
+        const hydrated = [];
+        for (const h of indexHits) {
+            const cp = getCheckpoint(h.sessionId, h.checkpointId, h.repoId);
+            if (cp && cp.dedupStatus !== "removed") {
+                hydrated.push({ checkpoint: cp, score: h.score });
+            }
+        }
+        if (hydrated.length === 0)
+            return this.search(sid, query, k);
+        // MMR-dedupe the merged candidate set for diversity (mirrors sync search).
+        const mmrItems = hydrated.map((h) => ({
+            item: h,
+            vector: h.checkpoint.embedding,
+            relevance: h.score,
+        }));
+        return mmrRerank(mmrItems, k, this.cfg.MMR_LAMBDA);
+    }
     /**
      * Serve the RAPTOR tree for a query (Fix D): rehydrate the persisted tree and
      * return its staged-expansion leaf hits as SearchHits. Returns [] when no tree

package/dist/src/wordpiece.js

@@ -0,0 +1,129 @@
+/**
+ * wordpiece.ts — a self-contained WordPiece tokenizer for BERT/MiniLM.
+ *
+ * Loads the canonical `vocab.txt` (bert-base-uncased, ~30K tokens) from disk and
+ * implements the standard uncased BERT preprocessing + greedy longest-match
+ * WordPiece segmentation. No native dependency, no network — the vocab file is a
+ * local artifact fetched once by scripts/setup-minilm.mjs (PREVENT-PI-004).
+ *
+ * This mirrors HuggingFace `BertTokenizer` closely enough for sentence-embedding
+ * use: lowercase, strip accents, split on whitespace + punctuation, then
+ * WordPiece each token with the `##` continuation convention. Special tokens
+ * [CLS]/[SEP] are added by the caller's encode().
+ */
+import { readFileSync, existsSync } from "node:fs";
+const UNK = "[UNK]";
+const CLS = "[CLS]";
+const SEP = "[SEP]";
+const PAD = "[PAD]";
+const MAX_INPUT_CHARS_PER_WORD = 200;
+export class WordPieceTokenizer {
+    vocab;
+    clsId;
+    sepId;
+    padId;
+    unkId;
+    constructor(vocab) {
+        this.vocab = vocab;
+        this.clsId = vocab.get(CLS) ?? 101;
+        this.sepId = vocab.get(SEP) ?? 102;
+        this.padId = vocab.get(PAD) ?? 0;
+        this.unkId = vocab.get(UNK) ?? 100;
+    }
+    /** Build a tokenizer from a vocab.txt file (one token per line, index = line). */
+    static fromVocabFile(path) {
+        if (!existsSync(path)) {
+            throw new Error(`WordPiece vocab not found at ${path}. Run: node scripts/setup-minilm.mjs`);
+        }
+        const lines = readFileSync(path, "utf-8").split("\n");
+        const vocab = new Map();
+        for (let i = 0; i < lines.length; i++) {
+            const tok = lines[i].replace(/\r$/, "");
+            if (tok.length > 0 || i < lines.length - 1)
+                vocab.set(tok, i);
+        }
+        return new WordPieceTokenizer(vocab);
+    }
+    /** Uncased BERT basic tokenization: lowercase, strip accents, split on ws+punct. */
+    basicTokenize(text) {
+        // NFD + strip combining marks (accent removal), then lowercase.
+        const cleaned = text
+            .normalize("NFD")
+            .replace(/[̀-ͯ]/g, "")
+            .toLowerCase();
+        const tokens = [];
+        let buf = "";
+        const flush = () => {
+            if (buf.length > 0) {
+                tokens.push(buf);
+                buf = "";
+            }
+        };
+        for (const ch of cleaned) {
+            if (/\s/.test(ch)) {
+                flush();
+            }
+            else if (/[!-/:-@[-`{-~¡-¿]/.test(ch)) {
+                // Punctuation becomes its own token.
+                flush();
+                tokens.push(ch);
+            }
+            else {
+                buf += ch;
+            }
+        }
+        flush();
+        return tokens;
+    }
+    /** Greedy longest-match WordPiece for a single word. */
+    wordpiece(word) {
+        if (word.length > MAX_INPUT_CHARS_PER_WORD)
+            return [UNK];
+        const pieces = [];
+        let start = 0;
+        while (start < word.length) {
+            let end = word.length;
+            let cur = null;
+            while (start < end) {
+                let sub = word.slice(start, end);
+                if (start > 0)
+                    sub = "##" + sub;
+                if (this.vocab.has(sub)) {
+                    cur = sub;
+                    break;
+                }
+                end--;
+            }
+            if (cur === null)
+                return [UNK]; // any unmatchable piece → whole word is UNK
+            pieces.push(cur);
+            start = end;
+        }
+        return pieces;
+    }
+    /** Tokenize text into WordPiece token strings (no special tokens). */
+    tokenize(text) {
+        const out = [];
+        for (const word of this.basicTokenize(text)) {
+            for (const piece of this.wordpiece(word))
+                out.push(piece);
+        }
+        return out;
+    }
+    /**
+     * Encode text into model inputs with [CLS]…[SEP], truncated to `maxLen`.
+     * attention_mask is all 1s (no padding for single-sequence inference).
+     */
+    encode(text, maxLen = 256) {
+        const pieces = this.tokenize(text).slice(0, Math.max(0, maxLen - 2));
+        const inputIds = [this.clsId];
+        for (const p of pieces)
+            inputIds.push(this.vocab.get(p) ?? this.unkId);
+        inputIds.push(this.sepId);
+        return {
+            inputIds,
+            attentionMask: inputIds.map(() => 1),
+            tokenTypeIds: inputIds.map(() => 0),
+        };
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-mega-compact",
-  "version": "0.4.24",
+  "version": "0.4.25",
   "description": "Layered, local, vector-backed context compressor for pi — supersede/collapse/cluster compaction with deduped inline recall.",
   "type": "module",
   "license": "BSD-2-Clause",
@@ -57,6 +57,8 @@
     "typescript": "^5.4.0"
   },
   "dependencies": {
+    "@electric-sql/pglite": "^0.5.4",
+    "@electric-sql/pglite-pgvector": "^0.0.5",
     "@mongodb-js/zstd": "^7.0.0"
   }
 }

package/src/recall.ts

@@ -137,3 +137,66 @@ export function recallAndInline(
     empty: toInject.length === 0,
   };
 }
+
+/**
+ * Slice 2 async cross-repo recall. Same dedup/bound/inline contract as
+ * `recallAndInline`, but backed by `VectorStore.searchAsync` so it can recall
+ * across repos (HNSW NN over the global PGlite index) when `opts.crossRepo` is
+ * set. The synchronous `recallAndInline` is unchanged and remains the default
+ * per-session path. Inline-window dedupe + token cap (Fix C) apply here too.
+ *
+ * `store` must provide `searchAsync` (the live VectorStore does). Errors fall
+ * back to an empty result — recall is a bonus, never a hard dependency.
+ */
+export async function recallAndInlineAsync(
+  opts: RecallInjectOptions & { crossRepo?: boolean; repoId?: string },
+  store: Pick<VectorStore, "searchAsync" | "wasInjected" | "markInjected">,
+): Promise<RecallInjectResult> {
+  const limit = opts.limit ?? 3;
+  const skip = opts.skipInjected ?? true;
+  const maxTokens = opts.recallMaxTokens ?? 0;
+  const doWindowDedupe = opts.windowDedupe ?? false;
+  const dedupSim = opts.dedupSim ?? 0.9;
+
+  let hits: SearchHit[] = [];
+  try {
+    hits = await store.searchAsync(opts.sessionId, opts.query, limit, {
+      crossRepo: opts.crossRepo,
+      repoId: opts.repoId,
+    });
+  } catch {
+    hits = [];
+  }
+
+  let liveEmbeddings: number[][] = [];
+  if (doWindowDedupe && opts.liveWindow && opts.liveWindow.length > 0) {
+    const embedder = defaultEmbedder();
+    liveEmbeddings = opts.liveWindow.map((m) => embedder.embed(m));
+  }
+
+  const toInject: SearchHit[] = [];
+  const parts: string[] = [];
+  let blockTokens = 0;
+
+  for (const h of hits) {
+    if (skip && store.wasInjected(opts.sessionId, h.checkpoint.checkpointId)) continue;
+    if (doWindowDedupe && liveEmbeddings.length > 0) {
+      const hitVec = defaultEmbedder().embed(h.checkpoint.summary);
+      if (liveEmbeddings.some((v) => cosineSimilarity(v, hitVec) >= dedupSim)) continue;
+    }
+    const part = formatRecallBlock([h]);
+    const partTokens = estimateBlockTokens(part);
+    if (maxTokens > 0 && blockTokens + partTokens > maxTokens) break;
+    parts.push(part);
+    toInject.push(h);
+    blockTokens += partTokens;
+    store.markInjected(opts.sessionId, h.checkpoint.checkpointId);
+  }
+
+  const block = parts.join("\n");
+  const report = toInject.map(
+    (h) => `  • ${h.checkpoint.checkpointId} (${h.checkpoint.summary.slice(0, 60).replace(/\n/g, " ")}…)`,
+  );
+
+  return { toInject, report, block, empty: toInject.length === 0 };
+}

package/src/store/sqlite.ts

@@ -872,6 +872,19 @@ export function hasCheckpoint(sessionId: string, checkpointId: string, stateDir:
   return row !== undefined;
 }
 
+/** Fetch a single checkpoint by (session, id), or undefined if absent. */
+export function getCheckpoint(
+  sessionId: string,
+  checkpointId: string,
+  stateDir: string = getStateDir(),
+): StoredCheckpoint | undefined {
+  const db = openStore(stateDir);
+  const row = db
+    .prepare("SELECT * FROM context_chunks WHERE session_id = ? AND id = ? LIMIT 1")
+    .get(normalizeSessionId(sessionId), checkpointId) as any;
+  return row ? rowToCheckpoint(row) : undefined;
+}
+
 /** Mark a checkpoint's dedup_status (e.g. 'removed' by SemDeDup). */
 export function setDedupStatus(
   checkpointId: string,

package/src/store/vectorIndex.test.ts

@@ -0,0 +1,116 @@
+/**
+ * vectorIndex.test.ts — Slice 2 async PGlite/HNSW vector index.
+ *
+ * Proves: cross-repo nearest-neighbor recall, repoId scoping, the dimension
+ * guard (non-512 vectors skipped, never corrupt the index), and graceful
+ * degradation when the index is disabled (kill-switch) — all without touching
+ * the synchronous node:sqlite store.
+ *
+ * The index is a WASM Postgres (PGlite) — fully local, zero network
+ * (PREVENT-PI-004). Each test isolates state via MEGACOMPACT_VECTOR_INDEX_DIR.
+ */
+
+import { test } from "node:test";
+import assert from "node:assert/strict";
+import { mkdtempSync, rmSync } from "node:fs";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+import {
+  EMBEDDING_DIM,
+  initVectorIndex,
+  upsertEmbedding,
+  searchAsync,
+  closeVectorIndex,
+  isVectorIndexDisabled,
+} from "./vectorIndex.js";
+
+/** A 512-dim unit-ish vector with a single spike at `idx` (deterministic NN). */
+function spikeVec(idx: number, magnitude = 1): number[] {
+  const v = new Array<number>(EMBEDDING_DIM).fill(0);
+  v[idx % EMBEDDING_DIM] = magnitude;
+  return v;
+}
+
+function isolateIndexDir(): string {
+  const dir = mkdtempSync(join(tmpdir(), "mc-vecidx-"));
+  process.env.MEGACOMPACT_VECTOR_INDEX_DIR = dir;
+  return dir;
+}
+
+test("cross-repo HNSW nearest-neighbor recall across repos + repoId scoping", async () => {
+  delete process.env.MEGACOMPACT_PGLITE_DISABLED;
+  const dir = isolateIndexDir();
+  try {
+    await closeVectorIndex(); // ensure a fresh singleton for this dir
+    const pg = await initVectorIndex();
+    assert.ok(pg, "index should initialize (PGlite WASM available)");
+
+    // repoA: two checkpoints; repoB: one checkpoint. Distinct spike directions.
+    await upsertEmbedding("/repoA/.pi/mega-compact", "sessA", "chkpt_001", spikeVec(0));
+    await upsertEmbedding("/repoA/.pi/mega-compact", "sessA", "chkpt_002", spikeVec(5));
+    await upsertEmbedding("/repoB/.pi/mega-compact", "sessB", "chkpt_001", spikeVec(0));
+
+    // Cross-repo query near spike(0): nearest are the two spike(0) rows, one per repo.
+    const cross = await searchAsync(spikeVec(0), { k: 2 });
+    assert.equal(cross.length, 2, "cross-repo returns two nearest");
+    const repos = new Set(cross.map((h) => h.repoId));
+    assert.ok(repos.has("/repoA/.pi/mega-compact"), "hit from repoA");
+    assert.ok(repos.has("/repoB/.pi/mega-compact"), "hit from repoB");
+    assert.ok(cross[0].score > 0.99, "top hit is near-identical (cosine ~1)");
+
+    // Scoped to repoA only: excludes repoB even though repoB has an identical vec.
+    const scoped = await searchAsync(spikeVec(0), { k: 5, repoId: "/repoA/.pi/mega-compact" });
+    assert.ok(scoped.length >= 1, "scoped returns repoA hits");
+    assert.ok(
+      scoped.every((h) => h.repoId === "/repoA/.pi/mega-compact"),
+      "repoId filter excludes other repos",
+    );
+  } finally {
+    await closeVectorIndex();
+    rmSync(dir, { recursive: true, force: true });
+    delete process.env.MEGACOMPACT_VECTOR_INDEX_DIR;
+  }
+});
+
+test("dimension guard: non-512 vectors are skipped, never corrupt the index", async () => {
+  delete process.env.MEGACOMPACT_PGLITE_DISABLED;
+  const dir = isolateIndexDir();
+  try {
+    await closeVectorIndex();
+    await initVectorIndex();
+    // Wrong-dimension vector (BYO embedder mismatch) must be silently skipped.
+    await upsertEmbedding("/repoC/.pi/mega-compact", "sessC", "chkpt_001", [1, 2, 3]);
+    const hits = await searchAsync(spikeVec(0), { k: 5 });
+    assert.equal(hits.length, 0, "no rows stored for a mismatched-dim vector");
+
+    // A correct-dim vector still stores fine afterward (index not corrupted).
+    await upsertEmbedding("/repoC/.pi/mega-compact", "sessC", "chkpt_002", spikeVec(3));
+    const ok = await searchAsync(spikeVec(3), { k: 1 });
+    assert.equal(ok.length, 1, "valid vector stored after a skipped one");
+    assert.equal(ok[0].checkpointId, "chkpt_002");
+  } finally {
+    await closeVectorIndex();
+    rmSync(dir, { recursive: true, force: true });
+    delete process.env.MEGACOMPACT_VECTOR_INDEX_DIR;
+  }
+});
+
+test("kill-switch: MEGACOMPACT_PGLITE_DISABLED disables the index gracefully", async () => {
+  const dir = isolateIndexDir();
+  process.env.MEGACOMPACT_PGLITE_DISABLED = "true";
+  try {
+    await closeVectorIndex();
+    assert.equal(isVectorIndexDisabled(), true, "kill-switch reported disabled");
+    const pg = await initVectorIndex();
+    assert.equal(pg, undefined, "init returns undefined when disabled");
+    // Upsert + search are no-ops that never throw and return empty.
+    await upsertEmbedding("/repoD/.pi/mega-compact", "sessD", "chkpt_001", spikeVec(0));
+    const hits = await searchAsync(spikeVec(0), { k: 3 });
+    assert.deepEqual(hits, [], "search returns [] when disabled");
+  } finally {
+    delete process.env.MEGACOMPACT_PGLITE_DISABLED;
+    await closeVectorIndex();
+    rmSync(dir, { recursive: true, force: true });
+    delete process.env.MEGACOMPACT_VECTOR_INDEX_DIR;
+  }
+});