@zhixuan92/multi-model-agent-core 4.0.1 → 4.0.3

package/dist/stores/file-backed-context-block-store.js

@@ -0,0 +1,293 @@
+/**
+ * FileBackedContextBlockStore — disk-backed implementation of
+ * `ContextBlockStore` so context blocks survive daemon restarts.
+ *
+ * Per the wire-telemetry-gaps plan, Gap 4: round-over-round audit is the
+ * documented use case for context blocks; a memory-only store loses
+ * everything on `npm run serve` cycles, breaking the recipe. This store
+ * persists to disk and reloads lazily on first access.
+ *
+ * Layout: `<projectCwd>/.mma/context-blocks/<id>.txt` (content) +
+ * `<id>.meta.json` ({ createdAt, ttlMs, lengthChars, sha256 }).
+ *
+ * Operational properties (per round-2 F5/F6/F7/F8/F9):
+ *   - **Authoritative cwd**: store root is constructed from `projectCwd`
+ *     supplied at construction time. NOT daemon cwd, NOT task cwd.
+ *   - **Permissions**: directory `0700`, files `0600`. Audit findings can
+ *     carry sensitive code excerpts; user-only access by default.
+ *   - **TTL**: 7 days default. Enforced on `get()` (lazy delete on
+ *     stale read), at the start of `register()` (opportunistic GC if
+ *     last sweep was over `gcCheckIntervalMs` ago), and via
+ *     `runIdleSweep()` for periodic timer-driven sweeps.
+ *   - **Size caps**: per-block 1 MiB, per-store 100 MiB total — measured
+ *     via `Buffer.byteLength(content, 'utf8')` so multi-byte content
+ *     can't sneak past the cap. Oldest-first eviction when total cap
+ *     is exceeded.
+ *   - **Atomic writes**: temp-file → fsync → rename. GC treats orphaned
+ *     pairs (content without metadata or vice versa) as corrupt and
+ *     deletes both — recovers cleanly from daemon SIGKILL mid-write.
+ *   - **Gitignore awareness**: on first directory creation, log a
+ *     stderr breadcrumb suggesting `.mma/` be added to the project's
+ *     `.gitignore` (we do NOT auto-edit the ignore file — surprising
+ *     and may not be a git repo).
+ *
+ * Pinning is a no-op on this implementation — disk persistence is
+ * stronger than the in-memory pin contract. The `pin`/`unpin` methods
+ * exist on the interface for API symmetry and to satisfy the LRU-skip
+ * contract that batch-registry expects, but disk eviction always
+ * respects TTL only.
+ */
+import { createHash, randomUUID } from 'node:crypto';
+import * as fs from 'node:fs';
+import * as path from 'node:path';
+const DEFAULT_TTL_MS = 7 * 24 * 60 * 60 * 1000;
+const DEFAULT_GC_INTERVAL_MS = 60 * 60 * 1000;
+const DEFAULT_MAX_BLOCK_BYTES = 1 * 1024 * 1024;
+const DEFAULT_MAX_TOTAL_BYTES = 100 * 1024 * 1024;
+export class FileBackedContextBlockStore {
+    rootDir;
+    _ttlMs;
+    gcCheckIntervalMs;
+    maxBlockBytes;
+    maxTotalBytes;
+    lastSweepMs = 0;
+    /** Pin counts are kept in memory only — they're an active-batch
+     *  protection signal, not a persistence concern. Lost on restart;
+     *  TTL-based GC handles the recovery side. */
+    pinCounts = new Map();
+    constructor(projectCwd, opts = {}) {
+        this.rootDir = path.resolve(projectCwd, '.mma', 'context-blocks');
+        this._ttlMs = opts.ttlMs ?? DEFAULT_TTL_MS;
+        this.gcCheckIntervalMs = opts.gcCheckIntervalMs ?? DEFAULT_GC_INTERVAL_MS;
+        this.maxBlockBytes = opts.maxBlockBytes ?? DEFAULT_MAX_BLOCK_BYTES;
+        this.maxTotalBytes = opts.maxTotalBytes ?? DEFAULT_MAX_TOTAL_BYTES;
+        this.ensureRoot();
+    }
+    register(content, opts = {}) {
+        const byteSize = Buffer.byteLength(content, 'utf8');
+        if (byteSize > this.maxBlockBytes) {
+            throw new Error(`Context block exceeds per-block cap of ${this.maxBlockBytes} bytes (got ${byteSize}).`);
+        }
+        const id = opts.id ?? randomUUID();
+        const now = Date.now();
+        const sha256 = createHash('sha256').update(content).digest('hex');
+        const meta = {
+            createdAt: now,
+            ttlMs: this._ttlMs,
+            lengthChars: content.length,
+            sha256,
+        };
+        // Opportunistic GC + size-cap enforcement before we write.
+        if (now - this.lastSweepMs > this.gcCheckIntervalMs) {
+            this.runIdleSweep(now, this._ttlMs);
+            this.lastSweepMs = now;
+        }
+        this.evictUntilFits(byteSize);
+        this.atomicWrite(this.contentPath(id), content);
+        this.atomicWrite(this.metaPath(id), JSON.stringify(meta) + '\n');
+        return { id, lengthChars: content.length, sha256 };
+    }
+    get(id) {
+        const meta = this.readMeta(id);
+        if (!meta)
+            return undefined;
+        const now = Date.now();
+        if (now - meta.createdAt > meta.ttlMs) {
+            // Expired: lazy-delete (matches in-memory store semantics)
+            this.deleteFiles(id);
+            return undefined;
+        }
+        try {
+            return fs.readFileSync(this.contentPath(id), 'utf8');
+        }
+        catch {
+            // Content missing but meta present — orphan; clean up.
+            this.deleteFiles(id);
+            return undefined;
+        }
+    }
+    delete(id) {
+        const existed = fs.existsSync(this.contentPath(id)) || fs.existsSync(this.metaPath(id));
+        this.deleteFiles(id);
+        this.pinCounts.delete(id);
+        return existed;
+    }
+    pin(id) {
+        this.pinCounts.set(id, (this.pinCounts.get(id) ?? 0) + 1);
+    }
+    unpin(id) {
+        const cur = this.pinCounts.get(id) ?? 0;
+        if (cur <= 1)
+            this.pinCounts.delete(id);
+        else
+            this.pinCounts.set(id, cur - 1);
+    }
+    refcount(id) {
+        return this.pinCounts.get(id) ?? 0;
+    }
+    runIdleSweep(now, idleTtlMs) {
+        let evicted = 0;
+        for (const id of this.listIds()) {
+            if ((this.pinCounts.get(id) ?? 0) > 0)
+                continue;
+            const meta = this.readMeta(id);
+            if (!meta) {
+                // Orphan content (no meta) — corrupt half-write recovery
+                this.deleteFiles(id);
+                evicted++;
+                continue;
+            }
+            // Use the entry's stored ttlMs (may differ from caller's idleTtlMs)
+            // OR caller's idleTtlMs, whichever is shorter — caller wins for
+            // explicit forced sweeps.
+            const effectiveTtl = Math.min(meta.ttlMs, idleTtlMs);
+            if (now - meta.createdAt > effectiveTtl) {
+                this.deleteFiles(id);
+                evicted++;
+            }
+        }
+        return evicted;
+    }
+    get ttlMs() {
+        return this._ttlMs;
+    }
+    /** Number of entries currently on disk (counted by id, not by file —
+     *  orphan halves count once). */
+    get size() {
+        return this.listIds().length;
+    }
+    /** Delete every block. Used by project-registry on idle eviction. */
+    clear() {
+        for (const id of this.listIds())
+            this.deleteFiles(id);
+        this.pinCounts.clear();
+    }
+    /** Sum of all on-disk content bytes (no caching — fresh read each call).
+     *  Used by tests + `evictUntilFits`. */
+    totalBytesOnDisk() {
+        let total = 0;
+        for (const id of this.listIds()) {
+            try {
+                total += fs.statSync(this.contentPath(id)).size;
+            }
+            catch {
+                // missing file — skip
+            }
+        }
+        return total;
+    }
+    // ── private helpers ────────────────────────────────────────────────
+    ensureRoot() {
+        const created = !fs.existsSync(this.rootDir);
+        fs.mkdirSync(this.rootDir, { recursive: true, mode: 0o700 });
+        if (created) {
+            // First-creation breadcrumb: nudge the user about gitignore. We
+            // intentionally do NOT modify their .gitignore — that would
+            // surprise users + may not be a git repo at all.
+            process.stderr.write(`[mmagent] created ${this.rootDir} — add ".mma/" to .gitignore (contains worker-output context blocks)\n`);
+        }
+    }
+    contentPath(id) {
+        return path.join(this.rootDir, `${id}.txt`);
+    }
+    metaPath(id) {
+        return path.join(this.rootDir, `${id}.meta.json`);
+    }
+    listIds() {
+        let entries;
+        try {
+            entries = fs.readdirSync(this.rootDir);
+        }
+        catch {
+            return [];
+        }
+        const ids = new Set();
+        for (const name of entries) {
+            if (name.endsWith('.meta.json'))
+                ids.add(name.slice(0, -'.meta.json'.length));
+            else if (name.endsWith('.txt'))
+                ids.add(name.slice(0, -'.txt'.length));
+        }
+        return [...ids];
+    }
+    readMeta(id) {
+        try {
+            const raw = fs.readFileSync(this.metaPath(id), 'utf8');
+            const parsed = JSON.parse(raw);
+            if (typeof parsed.createdAt === 'number'
+                && typeof parsed.ttlMs === 'number'
+                && typeof parsed.lengthChars === 'number'
+                && typeof parsed.sha256 === 'string') {
+                return parsed;
+            }
+        }
+        catch {
+            // unreadable / corrupt
+        }
+        return undefined;
+    }
+    deleteFiles(id) {
+        try {
+            fs.unlinkSync(this.contentPath(id));
+        }
+        catch { /* ignore */ }
+        try {
+            fs.unlinkSync(this.metaPath(id));
+        }
+        catch { /* ignore */ }
+    }
+    /**
+     * Atomic write: temp-file → fsync → rename. Recovers cleanly from
+     * daemon SIGKILL mid-write — readers either see the old content or
+     * the new content, never a partial write. The .tmp suffix uses a
+     * random nonce so concurrent writers don't collide.
+     */
+    atomicWrite(targetPath, content) {
+        const tmpPath = `${targetPath}.tmp.${process.pid}.${randomUUID()}`;
+        const fd = fs.openSync(tmpPath, 'w', 0o600);
+        try {
+            fs.writeSync(fd, content);
+            fs.fsyncSync(fd);
+        }
+        finally {
+            fs.closeSync(fd);
+        }
+        fs.renameSync(tmpPath, targetPath);
+    }
+    /**
+     * Oldest-first eviction loop until total disk usage + incoming bytes
+     * fits under maxTotalBytes. Pinned entries are skipped (matches
+     * in-memory store semantics — pinned blocks are held by an active
+     * batch and shouldn't be evicted mid-run).
+     */
+    evictUntilFits(incomingBytes) {
+        let total = this.totalBytesOnDisk();
+        if (total + incomingBytes <= this.maxTotalBytes)
+            return;
+        const candidates = [];
+        for (const id of this.listIds()) {
+            if ((this.pinCounts.get(id) ?? 0) > 0)
+                continue;
+            const meta = this.readMeta(id);
+            if (!meta)
+                continue;
+            let bytes = 0;
+            try {
+                bytes = fs.statSync(this.contentPath(id)).size;
+            }
+            catch {
+                continue;
+            }
+            candidates.push({ id, createdAt: meta.createdAt, bytes });
+        }
+        // Oldest-first
+        candidates.sort((a, b) => a.createdAt - b.createdAt);
+        for (const c of candidates) {
+            if (total + incomingBytes <= this.maxTotalBytes)
+                break;
+            this.deleteFiles(c.id);
+            total -= c.bytes;
+        }
+    }
+}
+//# sourceMappingURL=file-backed-context-block-store.js.map

package/dist/stores/file-backed-context-block-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"file-backed-context-block-store.js","sourceRoot":"","sources":["../../src/stores/file-backed-context-block-store.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAsCG;AACH,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACrD,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAMlC,MAAM,cAAc,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;AAC/C,MAAM,sBAAsB,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;AAC9C,MAAM,uBAAuB,GAAG,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC;AAChD,MAAM,uBAAuB,GAAG,GAAG,GAAG,IAAI,GAAG,IAAI,CAAC;AAoBlD,MAAM,OAAO,2BAA2B;IAC9B,OAAO,CAAS;IAChB,MAAM,CAAS;IACf,iBAAiB,CAAS;IAC1B,aAAa,CAAS;IACtB,aAAa,CAAS;IACtB,WAAW,GAAG,CAAC,CAAC;IACxB;;kDAE8C;IACtC,SAAS,GAAG,IAAI,GAAG,EAAkB,CAAC;IAE9C,YAAY,UAAkB,EAAE,OAA2C,EAAE;QAC3E,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,MAAM,EAAE,gBAAgB,CAAC,CAAC;QAClE,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,KAAK,IAAI,cAAc,CAAC;QAC3C,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC,iBAAiB,IAAI,sBAAsB,CAAC;QAC1E,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,IAAI,uBAAuB,CAAC;QACnE,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,IAAI,uBAAuB,CAAC;QACnE,IAAI,CAAC,UAAU,EAAE,CAAC;IACpB,CAAC;IAED,QAAQ,CAAC,OAAe,EAAE,OAAwB,EAAE;QAClD,MAAM,QAAQ,GAAG,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACpD,IAAI,QAAQ,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CACb,0CAA0C,IAAI,CAAC,aAAa,eAAe,QAAQ,IAAI,CACxF,CAAC;QACJ,CAAC;QAED,MAAM,EAAE,GAAG,IAAI,CAAC,EAAE,IAAI,UAAU,EAAE,CAAC;QACnC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,MAAM,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAClE,MAAM,IAAI,GAAa;YACrB,SAAS,EAAE,GAAG;YACd,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,WAAW,EAAE,OAAO,CAAC,MAAM;YAC3B,MAAM;SACP,CAAC;QAEF,2DAA2D;QAC3D,IAAI,GAAG,GAAG,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACpD,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YACpC,IAAI,CAAC,WAAW,GAAG,GAAG,CAAC;QACzB,CAAC;QACD,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QAE9B,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,OAAO,CAAC,CAAC;QAChD,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC;QAEjE,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC;IACrD,CAAC;IAED,GAAG,CAAC,EAAU;QACZ,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QAC/B,IAAI,CAAC,IAAI;YAAE,OAAO,SAAS,CAAC;QAE5B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,GAAG,GAAG,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;YACtC,2DAA2D;YAC3D,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;YACrB,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,IAAI,CAAC;YACH,OAAO,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QACvD,CAAC;QAAC,MAAM,CAAC;YACP,uDAAuD;YACvD,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;YACrB,OAAO,SAAS,CAAC;QACnB,CAAC;IACH,CAAC;IAED,MAAM,CAAC,EAAU;QACf,MAAM,OAAO,GAAG,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;QACxF,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QACrB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC1B,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,GAAG,CAAC,EAAU;QACZ,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,EAAU;QACd,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;QACxC,IAAI,GAAG,IAAI,CAAC;YAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;;YACnC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC;IACvC,CAAC;IAED,QAAQ,CAAC,EAAU;QACjB,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;IACrC,CAAC;IAED,YAAY,CAAC,GAAW,EAAE,SAAiB;QACzC,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC;YAChC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC;gBAAE,SAAS;YAChD,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAC/B,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,yDAAyD;gBACzD,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;gBACrB,OAAO,EAAE,CAAC;gBACV,SAAS;YACX,CAAC;YACD,oEAAoE;YACpE,gEAAgE;YAChE,0BAA0B;YAC1B,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;YACrD,IAAI,GAAG,GAAG,IAAI,CAAC,SAAS,GAAG,YAAY,EAAE,CAAC;gBACxC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;gBACrB,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED;qCACiC;IACjC,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC,MAAM,CAAC;IAC/B,CAAC;IAED,qEAAqE;IACrE,KAAK;QACH,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,OAAO,EAAE;YAAE,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QACtD,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;IAED;4CACwC;IACxC,gBAAgB;QACd,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC;YAChC,IAAI,CAAC;gBACH,KAAK,IAAI,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;YAClD,CAAC;YAAC,MAAM,CAAC;gBACP,sBAAsB;YACxB,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,sEAAsE;IAE9D,UAAU;QAChB,MAAM,OAAO,GAAG,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7C,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7D,IAAI,OAAO,EAAE,CAAC;YACZ,gEAAgE;YAChE,4DAA4D;YAC5D,iDAAiD;YACjD,OAAO,CAAC,MAAM,CAAC,KAAK,CAClB,qBAAqB,IAAI,CAAC,OAAO,wEAAwE,CAC1G,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,WAAW,CAAC,EAAU;QAC5B,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;IAC9C,CAAC;IAEO,QAAQ,CAAC,EAAU;QACzB,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE,YAAY,CAAC,CAAC;IACpD,CAAC;IAEO,OAAO;QACb,IAAI,OAAiB,CAAC;QACtB,IAAI,CAAC;YACH,OAAO,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACzC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,MAAM,GAAG,GAAG,IAAI,GAAG,EAAU,CAAC;QAC9B,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;YAC3B,IAAI,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC;gBAAE,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC;iBACzE,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;gBAAE,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;QACzE,CAAC;QACD,OAAO,CAAC,GAAG,GAAG,CAAC,CAAC;IAClB,CAAC;IAEO,QAAQ,CAAC,EAAU;QACzB,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;YACvD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAa,CAAC;YAC3C,IACE,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ;mBACjC,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ;mBAChC,OAAO,MAAM,CAAC,WAAW,KAAK,QAAQ;mBACtC,OAAO,MAAM,CAAC,MAAM,KAAK,QAAQ,EACpC,CAAC;gBACD,OAAO,MAAM,CAAC;YAChB,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,uBAAuB;QACzB,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,WAAW,CAAC,EAAU;QAC5B,IAAI,CAAC;YAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC;QAAC,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;QACnE,IAAI,CAAC;YAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;QAAC,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAClE,CAAC;IAED;;;;;OAKG;IACK,WAAW,CAAC,UAAkB,EAAE,OAAe;QACrD,MAAM,OAAO,GAAG,GAAG,UAAU,QAAQ,OAAO,CAAC,GAAG,IAAI,UAAU,EAAE,EAAE,CAAC;QACnE,MAAM,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE,KAAK,CAAC,CAAC;QAC5C,IAAI,CAAC;YACH,EAAE,CAAC,SAAS,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YAC1B,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QACnB,CAAC;gBAAS,CAAC;YACT,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QACnB,CAAC;QACD,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IACrC,CAAC;IAED;;;;;OAKG;IACK,cAAc,CAAC,aAAqB;QAC1C,IAAI,KAAK,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACpC,IAAI,KAAK,GAAG,aAAa,IAAI,IAAI,CAAC,aAAa;YAAE,OAAO;QAGxD,MAAM,UAAU,GAAgB,EAAE,CAAC;QACnC,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC;YAChC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC;gBAAE,SAAS;YAChD,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAC/B,IAAI,CAAC,IAAI;gBAAE,SAAS;YACpB,IAAI,KAAK,GAAG,CAAC,CAAC;YACd,IAAI,CAAC;gBAAC,KAAK,GAAG,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC;gBAAC,SAAS;YAAC,CAAC;YAC3E,UAAU,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;QAC5D,CAAC;QACD,eAAe;QACf,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;QAErD,KAAK,MAAM,CAAC,IAAI,UAAU,EAAE,CAAC;YAC3B,IAAI,KAAK,GAAG,aAAa,IAAI,IAAI,CAAC,aAAa;gBAAE,MAAM;YACvD,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACvB,KAAK,IAAI,CAAC,CAAC,KAAK,CAAC;QACnB,CAAC;IACH,CAAC;CACF"}

package/dist/stores/project-context-registry.d.ts

@@ -1,8 +1,10 @@
-import { InMemoryContextBlockStore } from './context-block-tool.js';
+import { type ContextBlockStore } from './context-block-tool.js';
 import { BatchCache } from './batch-cache.js';
 export interface ProjectContext {
     readonly cwd: string;
-    readonly contextBlocks: InMemoryContextBlockStore;
+    /** 4.0.3+: store is the interface — concrete class chosen at
+     *  createProjectContext (file-backed by default, in-memory in tests). */
+    readonly contextBlocks: ContextBlockStore;
     /** Per-project terminal-only retention index; authoritative live-batch lookup is via BatchRegistry.countActiveForProject(cwd). */
     readonly batchCache: BatchCache;
     readonly createdAt: number;
@@ -13,5 +15,16 @@ export interface ProjectContext {
     readonly activeSessions: Set<string>;
     pendingReservations: number;
 }
-export declare function createProjectContext(cwd: string): ProjectContext;
+export interface CreateProjectContextOptions {
+    /** Override the context-block store. When omitted, defaults to a
+     *  FileBackedContextBlockStore rooted at `<cwd>/.mma/context-blocks/`
+     *  so blocks survive daemon restarts (Gap 4 fix). Tests pass an
+     *  InMemoryContextBlockStore to avoid filesystem side effects. */
+    contextBlockStore?: ContextBlockStore;
+}
+export declare function createProjectContext(cwd: string, opts?: CreateProjectContextOptions): ProjectContext;
+/** Test-only convenience constructor that uses the in-memory store. Call
+ *  sites that need filesystem isolation (and don't care about persistence)
+ *  should use this instead of passing the option each time. */
+export declare function createInMemoryProjectContext(cwd: string): ProjectContext;
 //# sourceMappingURL=project-context-registry.d.ts.map

package/dist/stores/project-context-registry.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"project-context-registry.d.ts","sourceRoot":"","sources":["../../src/stores/project-context-registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,yBAAyB,EAAE,MAAM,yBAAyB,CAAC;AACpE,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAE9C,MAAM,WAAW,cAAc;IAC7B,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,aAAa,EAAE,yBAAyB,CAAC;IAClD,kIAAkI;IAClI,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAC;IAChC,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;IAC3B,oGAAoG;IACpG,cAAc,EAAE,MAAM,CAAC;IACvB,sDAAsD;IACtD,cAAc,EAAE,MAAM,CAAC;IAEvB,QAAQ,CAAC,cAAc,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IACrC,mBAAmB,EAAE,MAAM,CAAC;CAC7B;AAED,wBAAgB,oBAAoB,CAAC,GAAG,EAAE,MAAM,GAAG,cAAc,CAYhE"}
+{"version":3,"file":"project-context-registry.d.ts","sourceRoot":"","sources":["../../src/stores/project-context-registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAA6B,KAAK,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAE5F,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAE9C,MAAM,WAAW,cAAc;IAC7B,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC;IACrB;6EACyE;IACzE,QAAQ,CAAC,aAAa,EAAE,iBAAiB,CAAC;IAC1C,kIAAkI;IAClI,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAC;IAChC,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;IAC3B,oGAAoG;IACpG,cAAc,EAAE,MAAM,CAAC;IACvB,sDAAsD;IACtD,cAAc,EAAE,MAAM,CAAC;IAEvB,QAAQ,CAAC,cAAc,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IACrC,mBAAmB,EAAE,MAAM,CAAC;CAC7B;AAED,MAAM,WAAW,2BAA2B;IAC1C;;;sEAGkE;IAClE,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;CACvC;AAED,wBAAgB,oBAAoB,CAClC,GAAG,EAAE,MAAM,EACX,IAAI,GAAE,2BAAgC,GACrC,cAAc,CAYhB;AAED;;+DAE+D;AAC/D,wBAAgB,4BAA4B,CAAC,GAAG,EAAE,MAAM,GAAG,cAAc,CAExE"}

package/dist/stores/project-context-registry.js

@@ -1,10 +1,11 @@
 import { InMemoryContextBlockStore } from './context-block-tool.js';
+import { FileBackedContextBlockStore } from './file-backed-context-block-store.js';
 import { BatchCache } from './batch-cache.js';
-export function createProjectContext(cwd) {
+export function createProjectContext(cwd, opts = {}) {
     const now = Date.now();
     return {
         cwd,
-        contextBlocks: new InMemoryContextBlockStore(),
+        contextBlocks: opts.contextBlockStore ?? new FileBackedContextBlockStore(cwd),
         batchCache: new BatchCache(),
         createdAt: now,
         lastActivityAt: now,
@@ -13,4 +14,10 @@ export function createProjectContext(cwd) {
         pendingReservations: 0,
     };
 }
+/** Test-only convenience constructor that uses the in-memory store. Call
+ *  sites that need filesystem isolation (and don't care about persistence)
+ *  should use this instead of passing the option each time. */
+export function createInMemoryProjectContext(cwd) {
+    return createProjectContext(cwd, { contextBlockStore: new InMemoryContextBlockStore() });
+}
 //# sourceMappingURL=project-context-registry.js.map

package/dist/stores/project-context-registry.js.map

@@ -1 +1 @@
-{"version":3,"file":"project-context-registry.js","sourceRoot":"","sources":["../../src/stores/project-context-registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,yBAAyB,EAAE,MAAM,yBAAyB,CAAC;AACpE,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAiB9C,MAAM,UAAU,oBAAoB,CAAC,GAAW;IAC9C,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACvB,OAAO;QACL,GAAG;QACH,aAAa,EAAE,IAAI,yBAAyB,EAAE;QAC9C,UAAU,EAAE,IAAI,UAAU,EAAE;QAC5B,SAAS,EAAE,GAAG;QACd,cAAc,EAAE,GAAG;QACnB,cAAc,EAAE,IAAI,GAAG,EAAU;QACjC,cAAc,EAAE,CAAC;QACjB,mBAAmB,EAAE,CAAC;KACvB,CAAC;AACJ,CAAC"}
+{"version":3,"file":"project-context-registry.js","sourceRoot":"","sources":["../../src/stores/project-context-registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,yBAAyB,EAA0B,MAAM,yBAAyB,CAAC;AAC5F,OAAO,EAAE,2BAA2B,EAAE,MAAM,sCAAsC,CAAC;AACnF,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AA2B9C,MAAM,UAAU,oBAAoB,CAClC,GAAW,EACX,OAAoC,EAAE;IAEtC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACvB,OAAO;QACL,GAAG;QACH,aAAa,EAAE,IAAI,CAAC,iBAAiB,IAAI,IAAI,2BAA2B,CAAC,GAAG,CAAC;QAC7E,UAAU,EAAE,IAAI,UAAU,EAAE;QAC5B,SAAS,EAAE,GAAG;QACd,cAAc,EAAE,GAAG;QACnB,cAAc,EAAE,IAAI,GAAG,EAAU;QACjC,cAAc,EAAE,CAAC;QACjB,mBAAmB,EAAE,CAAC;KACvB,CAAC;AACJ,CAAC;AAED;;+DAE+D;AAC/D,MAAM,UAAU,4BAA4B,CAAC,GAAW;IACtD,OAAO,oBAAoB,CAAC,GAAG,EAAE,EAAE,iBAAiB,EAAE,IAAI,yBAAyB,EAAE,EAAE,CAAC,CAAC;AAC3F,CAAC"}

package/dist/tools/audit/schema.d.ts

@@ -2,15 +2,15 @@ import { z } from 'zod';
 export declare const inputSchema: z.ZodObject<{
     document: z.ZodOptional<z.ZodString>;
     auditType: z.ZodUnion<readonly [z.ZodEnum<{
-        style: "style";
         security: "security";
         performance: "performance";
+        style: "style";
         correctness: "correctness";
         general: "general";
     }>, z.ZodArray<z.ZodEnum<{
-        style: "style";
         security: "security";
         performance: "performance";
+        style: "style";
         correctness: "correctness";
     }>>]>;
     filePaths: z.ZodOptional<z.ZodArray<z.ZodString>>;

package/dist/tools/retry/tool-config.js

@@ -36,7 +36,7 @@ export const toolConfig = {
             sandboxPolicy: origTask?.sandboxPolicy ?? defaults?.sandboxPolicy ?? 'cwd-only',
             cwd: origTask?.cwd ?? ctx.projectContext?.cwd ?? ctx.cwd,
             contextBlockIds: origTask?.contextBlockIds ?? [],
-            mainModel: origTask?.mainModel ?? defaults?.mainModel ?? undefined,
+            mainModel: origTask?.mainModel,
             done: origTask?.done,
             autoCommit: origTask?.autoCommit ?? false,
         };

package/dist/tools/retry/tool-config.js.map

@@ -1 +1 @@
-{"version":3,"file":"tool-config.js","sourceRoot":"","sources":["../../../src/tools/retry/tool-config.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAI1C,OAAO,EAAE,iBAAiB,EAAE,MAAM,qDAAqD,CAAC;AACxF,OAAO,EAAE,qBAAqB,EAAE,MAAM,6CAA6C,CAAC;AACpF,OAAO,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAC;AAElE,MAAM,UAAU,aAAa,CAAC,QAA6B;IACzD,QAAQ,CAAC,QAAQ,CAAC;QAChB,SAAS,EAAE,aAAa;QACxB,UAAU,EAAE,MAAM;QAClB,QAAQ,EAAE,QAAQ;QAClB,OAAO,EAAE,MAAM;QACf,MAAM,EAAE,WAAW;QACnB,YAAY,EAAE,QAAQ;QACtB,gBAAgB,EAAE,UAAU;QAC5B,oBAAoB,EAAE,IAAI;QAC1B,iBAAiB,EAAE,eAAe;KACnC,CAAC,CAAC;AACL,CAAC;AAOD,MAAM,CAAC,MAAM,UAAU,GAA2C;IAChE,IAAI,EAAE,OAAO;IACb,QAAQ,EAAE,QAAQ;IAClB,SAAS,EAAE,UAAU;IACrB,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE,CACnB,KAAK,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;IAC9E,aAAa,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;QAC5B,MAAM,UAAU,GAAG,GAAG,CAAC,cAAc,EAAE,UAAU,CAAC;QAClD,MAAM,KAAK,GAAG,UAAU,EAAE,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,QAAQ,GAAG,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,SAAS,CAAyB,CAAC;QACvE,MAAM,QAAQ,GAAG,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC;QACrC,OAAO;YACL,MAAM,EAAE,QAAQ,EAAE,MAAM,IAAI,cAAc,KAAK,CAAC,SAAS,eAAe,KAAK,CAAC,OAAO,EAAE;YACvF,SAAS,EAAE,QAAQ,EAAE,SAAS,IAAI,UAAU;YAC5C,YAAY,EAAE,QAAQ,EAAE,YAAY,IAAI,MAAM;YAC9C,kBAAkB,EAAE,QAAQ,EAAE,kBAAkB,IAAI,KAAK;YACzD,KAAK,EAAE,QAAQ,EAAE,KAAK,IAAI,QAAQ,EAAE,KAAK,IAAI,MAAM;YACnD,SAAS,EAAE,QAAQ,EAAE,SAAS,IAAI,QAAQ,EAAE,SAAS,IAAI,SAAS;YAClE,UAAU,EAAE,QAAQ,EAAE,UAAU,IAAI,QAAQ,EAAE,UAAU,IAAI,EAAE;YAC9D,aAAa,EAAE,QAAQ,EAAE,aAAa,IAAI,QAAQ,EAAE,aAAa,IAAI,UAAU;YAC/E,GAAG,EAAE,QAAQ,EAAE,GAAG,IAAI,GAAG,CAAC,cAAc,EAAE,GAAG,IAAI,GAAG,CAAC,GAAG;YACxD,eAAe,EAAE,QAAQ,EAAE,eAAe,IAAI,EAAE;YAChD,SAAS,EAAE,QAAQ,EAAE,SAAS,IAAI,QAAQ,EAAE,SAAS,IAAI,SAAS;YAClE,IAAI,EAAE,QAAQ,EAAE,IAAI;YACpB,UAAU,EAAE,QAAQ,EAAE,UAAU,IAAI,KAAK;SAC1C,CAAC;IACJ,CAAC;IACD,YAAY,EAAE,iBAAiB;IAC/B,gBAAgB,EAAE,qBAAqB;IACvC,mBAAmB,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,EAAE;QACrC,MAAM,OAAO,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAgB,CAAC;QACzF,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,QAAQ,CAAC,QAAQ,GAAG,UAAU,KAAK,IAAI,KAAK,iBAAiB,CAAC;QAC9D,QAAQ,CAAC,gBAAgB,GAAG,aAAa,CAAC,+CAA+C,CAAC,CAAC;QAC3F,OAAO,QAAQ,CAAC,iBAAiB,CAAC;QAClC,OAAO,QAAQ,CAAC,oBAAoB,CAAC;QACrC,OAAO,QAAQ,CAAC,UAAU,CAAC;QAC3B,IAAI,GAAG,EAAE,OAAO,EAAE,CAAC;YACjB,QAAQ,CAAC,YAAY,GAAG,GAAG,CAAC,OAAO,CAAC;QACtC,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF,CAAC"}
+{"version":3,"file":"tool-config.js","sourceRoot":"","sources":["../../../src/tools/retry/tool-config.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAI1C,OAAO,EAAE,iBAAiB,EAAE,MAAM,qDAAqD,CAAC;AACxF,OAAO,EAAE,qBAAqB,EAAE,MAAM,6CAA6C,CAAC;AACpF,OAAO,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAC;AAElE,MAAM,UAAU,aAAa,CAAC,QAA6B;IACzD,QAAQ,CAAC,QAAQ,CAAC;QAChB,SAAS,EAAE,aAAa;QACxB,UAAU,EAAE,MAAM;QAClB,QAAQ,EAAE,QAAQ;QAClB,OAAO,EAAE,MAAM;QACf,MAAM,EAAE,WAAW;QACnB,YAAY,EAAE,QAAQ;QACtB,gBAAgB,EAAE,UAAU;QAC5B,oBAAoB,EAAE,IAAI;QAC1B,iBAAiB,EAAE,eAAe;KACnC,CAAC,CAAC;AACL,CAAC;AAOD,MAAM,CAAC,MAAM,UAAU,GAA2C;IAChE,IAAI,EAAE,OAAO;IACb,QAAQ,EAAE,QAAQ;IAClB,SAAS,EAAE,UAAU;IACrB,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE,CACnB,KAAK,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;IAC9E,aAAa,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;QAC5B,MAAM,UAAU,GAAG,GAAG,CAAC,cAAc,EAAE,UAAU,CAAC;QAClD,MAAM,KAAK,GAAG,UAAU,EAAE,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,QAAQ,GAAG,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,SAAS,CAAyB,CAAC;QACvE,MAAM,QAAQ,GAAG,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC;QACrC,OAAO;YACL,MAAM,EAAE,QAAQ,EAAE,MAAM,IAAI,cAAc,KAAK,CAAC,SAAS,eAAe,KAAK,CAAC,OAAO,EAAE;YACvF,SAAS,EAAE,QAAQ,EAAE,SAAS,IAAI,UAAU;YAC5C,YAAY,EAAE,QAAQ,EAAE,YAAY,IAAI,MAAM;YAC9C,kBAAkB,EAAE,QAAQ,EAAE,kBAAkB,IAAI,KAAK;YACzD,KAAK,EAAE,QAAQ,EAAE,KAAK,IAAI,QAAQ,EAAE,KAAK,IAAI,MAAM;YACnD,SAAS,EAAE,QAAQ,EAAE,SAAS,IAAI,QAAQ,EAAE,SAAS,IAAI,SAAS;YAClE,UAAU,EAAE,QAAQ,EAAE,UAAU,IAAI,QAAQ,EAAE,UAAU,IAAI,EAAE;YAC9D,aAAa,EAAE,QAAQ,EAAE,aAAa,IAAI,QAAQ,EAAE,aAAa,IAAI,UAAU;YAC/E,GAAG,EAAE,QAAQ,EAAE,GAAG,IAAI,GAAG,CAAC,cAAc,EAAE,GAAG,IAAI,GAAG,CAAC,GAAG;YACxD,eAAe,EAAE,QAAQ,EAAE,eAAe,IAAI,EAAE;YAChD,SAAS,EAAE,QAAQ,EAAE,SAAS;YAC9B,IAAI,EAAE,QAAQ,EAAE,IAAI;YACpB,UAAU,EAAE,QAAQ,EAAE,UAAU,IAAI,KAAK;SAC1C,CAAC;IACJ,CAAC;IACD,YAAY,EAAE,iBAAiB;IAC/B,gBAAgB,EAAE,qBAAqB;IACvC,mBAAmB,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,EAAE;QACrC,MAAM,OAAO,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAgB,CAAC;QACzF,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,QAAQ,CAAC,QAAQ,GAAG,UAAU,KAAK,IAAI,KAAK,iBAAiB,CAAC;QAC9D,QAAQ,CAAC,gBAAgB,GAAG,aAAa,CAAC,+CAA+C,CAAC,CAAC;QAC3F,OAAO,QAAQ,CAAC,iBAAiB,CAAC;QAClC,OAAO,QAAQ,CAAC,oBAAoB,CAAC;QACrC,OAAO,QAAQ,CAAC,UAAU,CAAC;QAC3B,IAAI,GAAG,EAAE,OAAO,EAAE,CAAC;YACjB,QAAQ,CAAC,YAAY,GAAG,GAAG,CAAC,OAAO,CAAC;QACtC,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF,CAAC"}

package/dist/tools/review/schema.d.ts

@@ -2,9 +2,9 @@ import { z } from 'zod';
 export declare const inputSchema: z.ZodObject<{
     code: z.ZodOptional<z.ZodString>;
     focus: z.ZodOptional<z.ZodArray<z.ZodEnum<{
-        style: "style";
         security: "security";
         performance: "performance";
+        style: "style";
         correctness: "correctness";
     }>>>;
     filePaths: z.ZodOptional<z.ZodArray<z.ZodString>>;

package/dist/types/enums.d.ts

@@ -4,13 +4,13 @@ export declare const TierEnum: z.ZodEnum<{
     complex: "complex";
 }>;
 export declare const RouteEnum: z.ZodEnum<{
-    review: "review";
-    retry: "retry";
     delegate: "delegate";
     audit: "audit";
+    review: "review";
     verify: "verify";
     debug: "debug";
     "execute-plan": "execute-plan";
+    retry: "retry";
     investigate: "investigate";
     explore: "explore";
     explore_internal: "explore_internal";
@@ -90,13 +90,6 @@ export declare const InvestigationConfidenceEnum: z.ZodEnum<{
 }>;
 export declare const FindingConfidenceSchema: z.ZodNumber;
 export declare const EventTypeEnum: z.ZodEnum<{
-    fallback: "fallback";
-    worker_start: "worker_start";
-    turn_start: "turn_start";
-    tool_call: "tool_call";
-    text_emission: "text_emission";
-    turn_complete: "turn_complete";
-    heartbeat: "heartbeat";
     batch_completed: "batch_completed";
     batch_failed: "batch_failed";
     cost_check: "cost_check";
@@ -110,7 +103,9 @@ export declare const EventTypeEnum: z.ZodEnum<{
     explore_synthesize_start: "explore_synthesize_start";
     explore_thread_completed: "explore_thread_completed";
     explore_thread_started: "explore_thread_started";
+    fallback: "fallback";
     fallback_unavailable: "fallback_unavailable";
+    heartbeat: "heartbeat";
     "read_only_review.quality": "read_only_review.quality";
     "read_only_review.terminal": "read_only_review.terminal";
     review_decision: "review_decision";
@@ -118,9 +113,14 @@ export declare const EventTypeEnum: z.ZodEnum<{
     stall_abort: "stall_abort";
     task_completed: "task_completed";
     task_started: "task_started";
+    text_emission: "text_emission";
     time_check: "time_check";
+    tool_call: "tool_call";
+    turn_complete: "turn_complete";
+    turn_start: "turn_start";
     verify_skipped: "verify_skipped";
     verify_step: "verify_step";
+    worker_start: "worker_start";
 }>;
 export declare const ReviewEngineTypeEnum: z.ZodEnum<{
     reviewer: "reviewer";
@@ -146,9 +146,9 @@ export declare const IncompleteReasonEnum: z.ZodEnum<{
  * Defined here so the closed-enum ratchet covers it before adoption.
  */
 export declare const EvidenceKindEnum: z.ZodEnum<{
-    fix: "fix";
     reproducer: "reproducer";
     code_path: "code_path";
+    fix: "fix";
 }>;
 export declare const DiagLoopEnum: z.ZodEnum<{
     spec: "spec";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zhixuan92/multi-model-agent-core",
-  "version": "4.0.1",
+  "version": "4.0.3",
   "type": "module",
   "license": "MIT",
   "description": "Core library for multi-model-agent: provider runners (Claude, Codex, OpenAI-compatible), routing logic, config schema, and tool/sandbox primitives.",
@@ -56,6 +56,10 @@
       "types": "./dist/lifecycle/task-runner.d.ts",
       "import": "./dist/lifecycle/task-runner.js"
     },
+    "./lifecycle/stage-progression": {
+      "types": "./dist/lifecycle/stage-progression.d.ts",
+      "import": "./dist/lifecycle/stage-progression.js"
+    },
     "./events/verbose-line": {
       "types": "./dist/events/verbose-line.d.ts",
       "import": "./dist/events/verbose-line.js"