@cortexkit/pi-magic-context 0.21.6 → 0.21.8

package/dist/subagent-entry.js

@@ -8909,6 +8909,65 @@ var MIGRATIONS = [
         db.exec("ALTER TABLE session_meta ADD COLUMN pending_pi_compaction_marker_state TEXT");
       }
     }
+  },
+  {
+    version: 19,
+    description: "Add compartment state lease table",
+    up: (db) => {
+      db.exec(`
+                CREATE TABLE IF NOT EXISTS compartment_state_lease (
+                    session_id TEXT PRIMARY KEY NOT NULL,
+                    holder_id TEXT NOT NULL,
+                    acquired_at INTEGER NOT NULL,
+                    expires_at INTEGER NOT NULL
+                );
+                CREATE INDEX IF NOT EXISTS idx_compartment_state_lease_expires
+                    ON compartment_state_lease(expires_at);
+            `);
+    }
+  },
+  {
+    version: 20,
+    description: "Add subagent invocation token accounting",
+    up: (db) => {
+      db.exec(`
+                CREATE TABLE IF NOT EXISTS subagent_invocations (
+                    id INTEGER PRIMARY KEY AUTOINCREMENT,
+                    session_id TEXT NOT NULL,
+                    harness TEXT NOT NULL,
+                    subagent TEXT NOT NULL,
+                    task TEXT,
+                    provider_id TEXT,
+                    model_id TEXT,
+                    started_at INTEGER NOT NULL,
+                    ended_at INTEGER,
+                    status TEXT NOT NULL,
+                    input_tokens INTEGER NOT NULL DEFAULT 0,
+                    output_tokens INTEGER NOT NULL DEFAULT 0,
+                    cache_read_tokens INTEGER NOT NULL DEFAULT 0,
+                    cache_write_tokens INTEGER NOT NULL DEFAULT 0,
+                    error TEXT,
+                    parent_invocation_id INTEGER
+                );
+                CREATE INDEX IF NOT EXISTS idx_sai_session_started
+                    ON subagent_invocations(session_id, started_at DESC);
+                CREATE INDEX IF NOT EXISTS idx_sai_subagent
+                    ON subagent_invocations(subagent, started_at DESC);
+            `);
+    }
+  },
+  {
+    version: 21,
+    description: "Add session lifetime work metrics",
+    up: (db) => {
+      const cols = db.prepare("PRAGMA table_info(session_meta)").all();
+      if (!cols.some((c) => c.name === "new_work_tokens")) {
+        db.exec("ALTER TABLE session_meta ADD COLUMN new_work_tokens INTEGER NOT NULL DEFAULT 0");
+      }
+      if (!cols.some((c) => c.name === "total_input_tokens")) {
+        db.exec("ALTER TABLE session_meta ADD COLUMN total_input_tokens INTEGER NOT NULL DEFAULT 0");
+      }
+    }
   }
 ];
 function ensureMigrationsTable(db) {
@@ -141956,6 +142015,15 @@ function initializeDatabase(db) {
     );
     CREATE INDEX IF NOT EXISTS idx_compartments_session ON compartments(session_id);
 
+    CREATE TABLE IF NOT EXISTS compartment_state_lease (
+      session_id TEXT PRIMARY KEY NOT NULL,
+      holder_id TEXT NOT NULL,
+      acquired_at INTEGER NOT NULL,
+      expires_at INTEGER NOT NULL
+    );
+    CREATE INDEX IF NOT EXISTS idx_compartment_state_lease_expires
+      ON compartment_state_lease(expires_at);
+
     CREATE TABLE IF NOT EXISTS compression_depth (
       session_id TEXT NOT NULL,
       message_ordinal INTEGER NOT NULL,
@@ -142152,12 +142220,37 @@ CREATE INDEX IF NOT EXISTS idx_dream_queue_pending ON dream_queue(started_at, en
       -- is a valid JSON blob written via setPendingPiCompactionMarkerState.
       -- Excluded from healNullTextColumns.
       pending_pi_compaction_marker_state TEXT,
+      new_work_tokens INTEGER NOT NULL DEFAULT 0,
+      total_input_tokens INTEGER NOT NULL DEFAULT 0,
       -- deferred_execute_state: intentionally NULLABLE without a default.
       -- Absence is SQL NULL; presence is a JSON blob written via
       -- setDeferredExecutePendingIfAbsent. Excluded from healNullTextColumns.
       deferred_execute_state TEXT
     );
 
+    CREATE TABLE IF NOT EXISTS subagent_invocations (
+      id INTEGER PRIMARY KEY AUTOINCREMENT,
+      session_id TEXT NOT NULL,
+      harness TEXT NOT NULL,
+      subagent TEXT NOT NULL,
+      task TEXT,
+      provider_id TEXT,
+      model_id TEXT,
+      started_at INTEGER NOT NULL,
+      ended_at INTEGER,
+      status TEXT NOT NULL,
+      input_tokens INTEGER NOT NULL DEFAULT 0,
+      output_tokens INTEGER NOT NULL DEFAULT 0,
+      cache_read_tokens INTEGER NOT NULL DEFAULT 0,
+      cache_write_tokens INTEGER NOT NULL DEFAULT 0,
+      error TEXT,
+      parent_invocation_id INTEGER
+    );
+    CREATE INDEX IF NOT EXISTS idx_sai_session_started
+      ON subagent_invocations(session_id, started_at DESC);
+    CREATE INDEX IF NOT EXISTS idx_sai_subagent
+      ON subagent_invocations(subagent, started_at DESC);
+
     CREATE INDEX IF NOT EXISTS idx_tags_session_tag_number ON tags(session_id, tag_number);
     CREATE INDEX IF NOT EXISTS idx_tags_session_message_id ON tags(session_id, message_id);
     CREATE INDEX IF NOT EXISTS idx_pending_ops_session ON pending_ops(session_id);
@@ -142249,6 +142342,8 @@ CREATE INDEX IF NOT EXISTS idx_dream_queue_pending ON dream_queue(started_at, en
   ensureColumn(db, "session_meta", "needs_emergency_recovery", "INTEGER DEFAULT 0");
   ensureColumn(db, "session_meta", "pending_compaction_marker_state", "TEXT");
   ensureColumn(db, "session_meta", "pending_pi_compaction_marker_state", "TEXT");
+  ensureColumn(db, "session_meta", "new_work_tokens", "INTEGER NOT NULL DEFAULT 0");
+  ensureColumn(db, "session_meta", "total_input_tokens", "INTEGER NOT NULL DEFAULT 0");
   ensureColumn(db, "session_meta", "deferred_execute_state", "TEXT");
   ensureColumn(db, "tags", "harness", "TEXT NOT NULL DEFAULT 'opencode'");
   ensureColumn(db, "pending_ops", "harness", "TEXT NOT NULL DEFAULT 'opencode'");
@@ -142312,7 +142407,9 @@ function healNullIntegerColumns(db) {
     ["tool_call_tokens", 0],
     ["note_nudge_trigger_pending", 0],
     ["observed_safe_input_tokens", 0],
-    ["cache_alert_sent", 0]
+    ["cache_alert_sent", 0],
+    ["new_work_tokens", 0],
+    ["total_input_tokens", 0]
   ];
   for (const [column, fallback] of columns) {
     try {
@@ -142384,6 +142481,56 @@ import { existsSync as existsSync6, readFileSync as readFileSync3 } from "node:f
 import { homedir as homedir3 } from "node:os";
 import { join as join6 } from "node:path";
 
+// ../plugin/src/config/agent-disable.ts
+function isDreamerRunnable(config) {
+  return !!config.dreamer && config.dreamer.disable !== true;
+}
+function clonePlainObject(value) {
+  if (typeof value !== "object" || value === null || Array.isArray(value)) {
+    return;
+  }
+  return { ...value };
+}
+function migrateLegacyEnabledForAgent(args) {
+  const agent = clonePlainObject(args.patched[args.agentName]);
+  if (!agent || !("enabled" in agent))
+    return;
+  const enabled = agent.enabled;
+  const disable = agent.disable;
+  delete agent.enabled;
+  if (args.agentName === "historian") {
+    args.warnings.push('Removed invalid "historian.enabled" in-memory (run doctor to persist).');
+    args.patched.historian = agent;
+    return;
+  }
+  if (args.agentName === "dreamer") {
+    if (disable !== true && enabled === false) {
+      agent.disable = true;
+      args.warnings.push('Migrated "dreamer.enabled=false" → "dreamer.disable=true" in-memory (run doctor to persist). This now also disables manual /ctx-dream; for manual-only remove disable and set schedule="".');
+    }
+    args.patched.dreamer = agent;
+    return;
+  }
+  if (disable !== true && enabled === false) {
+    agent.disable = true;
+    args.warnings.push('Migrated "sidekick.enabled=false" → "sidekick.disable=true" in-memory (run doctor to persist).');
+  }
+  args.patched.sidekick = agent;
+}
+function migrateLegacyAgentEnabledInMemory(rawConfig, warnings) {
+  const shouldPatch = ["dreamer", "sidekick", "historian"].some((key) => {
+    const agent = rawConfig[key];
+    return typeof agent === "object" && agent !== null && !Array.isArray(agent) && "enabled" in agent;
+  });
+  if (!shouldPatch)
+    return rawConfig;
+  const patched = { ...rawConfig };
+  migrateLegacyEnabledForAgent({ patched, agentName: "dreamer", warnings });
+  migrateLegacyEnabledForAgent({ patched, agentName: "sidekick", warnings });
+  migrateLegacyEnabledForAgent({ patched, agentName: "historian", warnings });
+  return patched;
+}
+
 // ../../node_modules/.bun/zod@4.4.3/node_modules/zod/v4/classic/external.js
 var exports_external = {};
 __export(exports_external, {
@@ -156723,7 +156870,6 @@ var DEFAULT_DREAMER_TASKS = [
 ];
 var PiThinkingLevelSchema = exports_external.enum(["off", "minimal", "low", "medium", "high", "xhigh"]).optional();
 var DreamerConfigSchema = AgentOverrideConfigSchema.merge(exports_external.object({
-  enabled: exports_external.boolean().default(false),
   schedule: exports_external.string().default("02:00-06:00"),
   max_runtime_minutes: exports_external.number().min(10).default(120),
   tasks: exports_external.array(DreamingTaskSchema).default(DEFAULT_DREAMER_TASKS),
@@ -156741,7 +156887,6 @@ var DreamerConfigSchema = AgentOverrideConfigSchema.merge(exports_external.objec
   thinking_level: PiThinkingLevelSchema
 }));
 var SidekickConfigSchema = AgentOverrideConfigSchema.extend({
-  enabled: exports_external.boolean().default(false),
   timeout_ms: exports_external.number().default(30000),
   system_prompt: exports_external.string().optional(),
   thinking_level: PiThinkingLevelSchema
@@ -157131,9 +157276,11 @@ function mergeRawConfigs(base, override) {
   return merged;
 }
 function parsePiConfig(rawConfig, recoveredTopLevelKeys = []) {
-  const parsed = MagicContextConfigSchema.safeParse(rawConfig);
+  const preMigrationWarnings = [];
+  const migrated = migrateLegacyAgentEnabledInMemory(rawConfig, preMigrationWarnings);
+  const parsed = MagicContextConfigSchema.safeParse(migrated);
   if (parsed.success) {
-    return { config: parsed.data, warnings: [] };
+    return { config: parsed.data, warnings: preMigrationWarnings };
   }
   const defaults = MagicContextConfigSchema.parse({});
   const errorPaths = new Set;
@@ -157143,8 +157290,8 @@ function parsePiConfig(rawConfig, recoveredTopLevelKeys = []) {
       errorPaths.add(String(topKey));
     }
   }
-  const patched = { ...rawConfig };
-  const warnings = [];
+  const patched = { ...migrated };
+  const warnings = [...preMigrationWarnings];
   for (const key of errorPaths) {
     recoveredTopLevelKeys.push(key);
     const isAgentConfig = key === "historian" || key === "dreamer" || key === "sidekick";
@@ -163148,7 +163295,7 @@ function convertEntriesToRawMessages(entries) {
       if (pendingToolParts.length > 0) {
         result.push({
           ordinal: nextOrdinal++,
-          id: pendingFirstRealId,
+          id: `synth-user-${pendingFirstRealId}`,
           role: "user",
           parts: pendingToolParts
         });
@@ -163173,7 +163320,7 @@ function convertEntriesToRawMessages(entries) {
   if (pendingToolParts.length > 0) {
     result.push({
       ordinal: nextOrdinal,
-      id: pendingFirstRealId,
+      id: `synth-user-${pendingFirstRealId}`,
       role: "user",
       parts: pendingToolParts
     });
@@ -163325,6 +163472,99 @@ function createCtxExpandTool(_deps) {
     }
   };
 }
+// ../plugin/src/features/magic-context/compartment-lease.ts
+var COMPARTMENT_LEASE_TTL_MS = 5 * 60 * 1000;
+var COMPARTMENT_LEASE_RENEWAL_MS = 60 * 1000;
+function acquireCompartmentLease(db, sessionId, holderId) {
+  const acquiredAt = Date.now();
+  const expiresAt = acquiredAt + COMPARTMENT_LEASE_TTL_MS;
+  const result = db.prepare(`INSERT INTO compartment_state_lease (session_id, holder_id, acquired_at, expires_at)
+             VALUES (?, ?, ?, ?)
+             ON CONFLICT(session_id) DO UPDATE SET
+                holder_id = excluded.holder_id,
+                acquired_at = excluded.acquired_at,
+                expires_at = excluded.expires_at
+             WHERE compartment_state_lease.holder_id = excluded.holder_id
+                OR compartment_state_lease.expires_at <= ?`).run(sessionId, holderId, acquiredAt, expiresAt, acquiredAt);
+  if (result.changes !== 1) {
+    return null;
+  }
+  return { sessionId, holderId, acquiredAt, expiresAt };
+}
+function renewCompartmentLease(db, sessionId, holderId) {
+  const now = Date.now();
+  const expiresAt = now + COMPARTMENT_LEASE_TTL_MS;
+  const result = db.prepare(`UPDATE compartment_state_lease
+             SET expires_at = ?, acquired_at = ?
+             WHERE session_id = ? AND holder_id = ? AND expires_at > ?`).run(expiresAt, now, sessionId, holderId, now);
+  return result.changes === 1;
+}
+function releaseCompartmentLease(db, sessionId, holderId) {
+  db.prepare("DELETE FROM compartment_state_lease WHERE session_id = ? AND holder_id = ?").run(sessionId, holderId);
+}
+function isCompartmentLeaseHeld(db, sessionId, holderId) {
+  const row = db.prepare("SELECT 1 FROM compartment_state_lease WHERE session_id = ? AND holder_id = ? AND expires_at > ?").get(sessionId, holderId, Date.now());
+  return row != null;
+}
+
+// ../plugin/src/features/magic-context/compression-depth-storage.ts
+var incrementDepthStatements = new WeakMap;
+var totalDepthStatements = new WeakMap;
+var maxDepthStatements = new WeakMap;
+var clearDepthStatements = new WeakMap;
+function getIncrementDepthStatement(db) {
+  let stmt = incrementDepthStatements.get(db);
+  if (!stmt) {
+    stmt = db.prepare("INSERT INTO compression_depth (session_id, message_ordinal, depth, harness) VALUES (?, ?, 1, ?) ON CONFLICT(session_id, message_ordinal) DO UPDATE SET depth = depth + 1");
+    incrementDepthStatements.set(db, stmt);
+  }
+  return stmt;
+}
+function getTotalDepthStatement(db) {
+  let stmt = totalDepthStatements.get(db);
+  if (!stmt) {
+    stmt = db.prepare("SELECT COALESCE(SUM(depth), 0) AS total_depth FROM compression_depth WHERE session_id = ? AND message_ordinal BETWEEN ? AND ?");
+    totalDepthStatements.set(db, stmt);
+  }
+  return stmt;
+}
+function getClearDepthStatement(db) {
+  let stmt = clearDepthStatements.get(db);
+  if (!stmt) {
+    stmt = db.prepare("DELETE FROM compression_depth WHERE session_id = ?");
+    clearDepthStatements.set(db, stmt);
+  }
+  return stmt;
+}
+function incrementCompressionDepth(db, sessionId, startOrdinal, endOrdinal) {
+  if (endOrdinal < startOrdinal) {
+    return;
+  }
+  db.transaction(() => {
+    const stmt = getIncrementDepthStatement(db);
+    for (let ordinal = startOrdinal;ordinal <= endOrdinal; ordinal += 1) {
+      stmt.run(sessionId, ordinal, getHarness());
+    }
+  })();
+}
+function getAverageCompressionDepth(db, sessionId, startOrdinal, endOrdinal) {
+  if (endOrdinal < startOrdinal) {
+    return 0;
+  }
+  const row = getTotalDepthStatement(db).get(sessionId, startOrdinal, endOrdinal);
+  const totalDepth = typeof row?.total_depth === "number" ? row.total_depth : 0;
+  const messageCount = endOrdinal - startOrdinal + 1;
+  return totalDepth / messageCount;
+}
+function clearCompressionDepth(db, sessionId) {
+  getClearDepthStatement(db).run(sessionId);
+}
+function clearCompressionDepthRange(db, sessionId, startOrdinal, endOrdinal) {
+  if (endOrdinal < startOrdinal) {
+    return;
+  }
+  db.prepare("DELETE FROM compression_depth WHERE session_id = ? AND message_ordinal BETWEEN ? AND ?").run(sessionId, startOrdinal, endOrdinal);
+}
 
 // ../plugin/src/features/magic-context/compartment-storage.ts
 var insertCompartmentStatements = new WeakMap;
@@ -163435,6 +163675,38 @@ function replaceAllCompartmentState(db, sessionId, compartments, facts) {
     db.prepare("UPDATE session_meta SET memory_block_cache = '', memory_block_ids = '' WHERE session_id = ?").run(sessionId);
   })();
 }
+function replaceAllCompartmentStateAndBumpDepth(db, holderId, sessionId, compartments, facts, depthStartOrdinal, depthEndOrdinal) {
+  const now = Date.now();
+  db.exec("BEGIN IMMEDIATE");
+  let finished = false;
+  try {
+    if (!isCompartmentLeaseHeld(db, sessionId, holderId)) {
+      db.exec("ROLLBACK");
+      finished = true;
+      return false;
+    }
+    db.prepare("DELETE FROM compartments WHERE session_id = ?").run(sessionId);
+    db.prepare("DELETE FROM session_facts WHERE session_id = ?").run(sessionId);
+    insertCompartmentRows(db, sessionId, compartments, now);
+    insertFactRows(db, sessionId, facts, now);
+    db.prepare("UPDATE session_meta SET memory_block_cache = '', memory_block_ids = '' WHERE session_id = ?").run(sessionId);
+    if (depthEndOrdinal >= depthStartOrdinal) {
+      const stmt = getIncrementDepthStatement(db);
+      for (let ordinal = depthStartOrdinal;ordinal <= depthEndOrdinal; ordinal += 1) {
+        stmt.run(sessionId, ordinal, getHarness());
+      }
+    }
+    db.exec("COMMIT");
+    finished = true;
+    return true;
+  } finally {
+    if (!finished) {
+      try {
+        db.exec("ROLLBACK");
+      } catch {}
+    }
+  }
+}
 function buildCompartmentBlock(compartments, facts, memoryBlock, dateRanges) {
   const lines = [];
   if (memoryBlock) {
@@ -163502,12 +163774,37 @@ function getRecompStaging(db, sessionId) {
     lastEndMessage: lastEnd
   };
 }
-function promoteRecompStaging(db, sessionId) {
+function promoteRecompStaging(db, sessionId, holderId) {
   const now = Date.now();
-  return db.transaction(() => {
+  if (!holderId) {
+    return db.transaction(() => {
+      const staging = getRecompStaging(db, sessionId);
+      if (!staging || staging.compartments.length === 0)
+        return null;
+      db.prepare("DELETE FROM compartments WHERE session_id = ?").run(sessionId);
+      db.prepare("DELETE FROM session_facts WHERE session_id = ?").run(sessionId);
+      insertCompartmentRows(db, sessionId, staging.compartments, now);
+      insertFactRows(db, sessionId, staging.facts, now);
+      db.prepare("DELETE FROM recomp_compartments WHERE session_id = ?").run(sessionId);
+      db.prepare("DELETE FROM recomp_facts WHERE session_id = ?").run(sessionId);
+      db.prepare("UPDATE session_meta SET memory_block_cache = '', memory_block_ids = '' WHERE session_id = ?").run(sessionId);
+      return { compartments: staging.compartments, facts: staging.facts };
+    })();
+  }
+  db.exec("BEGIN IMMEDIATE");
+  let finished = false;
+  try {
+    if (!isCompartmentLeaseHeld(db, sessionId, holderId)) {
+      db.exec("ROLLBACK");
+      finished = true;
+      return null;
+    }
     const staging = getRecompStaging(db, sessionId);
-    if (!staging || staging.compartments.length === 0)
+    if (!staging || staging.compartments.length === 0) {
+      db.exec("ROLLBACK");
+      finished = true;
       return null;
+    }
     db.prepare("DELETE FROM compartments WHERE session_id = ?").run(sessionId);
     db.prepare("DELETE FROM session_facts WHERE session_id = ?").run(sessionId);
     insertCompartmentRows(db, sessionId, staging.compartments, now);
@@ -163515,8 +163812,16 @@ function promoteRecompStaging(db, sessionId) {
     db.prepare("DELETE FROM recomp_compartments WHERE session_id = ?").run(sessionId);
     db.prepare("DELETE FROM recomp_facts WHERE session_id = ?").run(sessionId);
     db.prepare("UPDATE session_meta SET memory_block_cache = '', memory_block_ids = '' WHERE session_id = ?").run(sessionId);
+    db.exec("COMMIT");
+    finished = true;
     return { compartments: staging.compartments, facts: staging.facts };
-  })();
+  } finally {
+    if (!finished) {
+      try {
+        db.exec("ROLLBACK");
+      } catch {}
+    }
+  }
 }
 function invalidateAllMemoryBlockCaches(db) {
   try {
@@ -164376,64 +164681,6 @@ function createCtxMemoryTool(deps) {
     }
   };
 }
-// ../plugin/src/features/magic-context/compression-depth-storage.ts
-var incrementDepthStatements = new WeakMap;
-var totalDepthStatements = new WeakMap;
-var maxDepthStatements = new WeakMap;
-var clearDepthStatements = new WeakMap;
-function getIncrementDepthStatement(db) {
-  let stmt = incrementDepthStatements.get(db);
-  if (!stmt) {
-    stmt = db.prepare("INSERT INTO compression_depth (session_id, message_ordinal, depth, harness) VALUES (?, ?, 1, ?) ON CONFLICT(session_id, message_ordinal) DO UPDATE SET depth = depth + 1");
-    incrementDepthStatements.set(db, stmt);
-  }
-  return stmt;
-}
-function getTotalDepthStatement(db) {
-  let stmt = totalDepthStatements.get(db);
-  if (!stmt) {
-    stmt = db.prepare("SELECT COALESCE(SUM(depth), 0) AS total_depth FROM compression_depth WHERE session_id = ? AND message_ordinal BETWEEN ? AND ?");
-    totalDepthStatements.set(db, stmt);
-  }
-  return stmt;
-}
-function getClearDepthStatement(db) {
-  let stmt = clearDepthStatements.get(db);
-  if (!stmt) {
-    stmt = db.prepare("DELETE FROM compression_depth WHERE session_id = ?");
-    clearDepthStatements.set(db, stmt);
-  }
-  return stmt;
-}
-function incrementCompressionDepth(db, sessionId, startOrdinal, endOrdinal) {
-  if (endOrdinal < startOrdinal) {
-    return;
-  }
-  db.transaction(() => {
-    const stmt = getIncrementDepthStatement(db);
-    for (let ordinal = startOrdinal;ordinal <= endOrdinal; ordinal += 1) {
-      stmt.run(sessionId, ordinal, getHarness());
-    }
-  })();
-}
-function getAverageCompressionDepth(db, sessionId, startOrdinal, endOrdinal) {
-  if (endOrdinal < startOrdinal) {
-    return 0;
-  }
-  const row = getTotalDepthStatement(db).get(sessionId, startOrdinal, endOrdinal);
-  const totalDepth = typeof row?.total_depth === "number" ? row.total_depth : 0;
-  const messageCount = endOrdinal - startOrdinal + 1;
-  return totalDepth / messageCount;
-}
-function clearCompressionDepth(db, sessionId) {
-  getClearDepthStatement(db).run(sessionId);
-}
-function clearCompressionDepthRange(db, sessionId, startOrdinal, endOrdinal) {
-  if (endOrdinal < startOrdinal) {
-    return;
-  }
-  db.prepare("DELETE FROM compression_depth WHERE session_id = ? AND message_ordinal BETWEEN ? AND ?").run(sessionId, startOrdinal, endOrdinal);
-}
 // ../plugin/src/features/magic-context/message-index.ts
 var lastIndexedStatements = new WeakMap;
 var insertMessageStatements = new WeakMap;
@@ -165077,6 +165324,7 @@ function getPendingPiCompactionMarkerState(db, sessionId) {
       return parsed;
     }
   } catch {}
+  db.prepare("UPDATE session_meta SET pending_pi_compaction_marker_state = NULL WHERE session_id = ? AND pending_pi_compaction_marker_state = ?").run(sessionId, raw);
   return null;
 }
 function setPendingPiCompactionMarkerState(db, sessionId, state) {
@@ -165120,6 +165368,12 @@ function clearDeferredExecutePendingIfMatches(db, sessionId, expected) {
              WHERE session_id = ? AND deferred_execute_state = ?`).run(sessionId, expectedBlob);
   return result.changes > 0;
 }
+function setSessionWorkMetrics(db, sessionId, newWorkTokens, totalInputTokens) {
+  ensureSessionMetaRow(db, sessionId);
+  db.prepare(`UPDATE session_meta
+         SET new_work_tokens = ?, total_input_tokens = ?
+         WHERE session_id = ?`).run(Math.max(0, Math.floor(newWorkTokens)), Math.max(0, Math.floor(totalInputTokens)), sessionId);
+}
 // ../plugin/src/features/magic-context/resolve-subagent-fallback.ts
 init_logger();
 await init_read_session_db();
@@ -165435,6 +165689,19 @@ function getSourceContents(db, sessionId, tagIds) {
   }
   return sources;
 }
+// ../plugin/src/features/magic-context/storage-subagent-invocations.ts
+function clampToken(value) {
+  return Number.isFinite(value) && value > 0 ? Math.floor(value) : 0;
+}
+function recordSubagentInvocation(db, input) {
+  const result = db.prepare(`INSERT INTO subagent_invocations (
+                session_id, harness, subagent, task, provider_id, model_id,
+                started_at, ended_at, status,
+                input_tokens, output_tokens, cache_read_tokens, cache_write_tokens,
+                error, parent_invocation_id
+            ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`).run(input.sessionId, input.harness, input.subagent, input.task ?? null, input.providerId ?? null, input.modelId ?? null, input.startedAt, input.endedAt, input.status, clampToken(input.inputTokens), clampToken(input.outputTokens), clampToken(input.cacheReadTokens), clampToken(input.cacheWriteTokens), input.error ?? null, input.parentInvocationId ?? null);
+  return Number(result.lastInsertRowid);
+}
 // ../plugin/src/features/magic-context/storage-tags.ts
 var insertTagStatements = new WeakMap;
 var updateTagStatusStatements = new WeakMap;
@@ -167367,8 +167634,7 @@ function magicContextSubagentExtension(pi) {
     try {
       const db = openDatabase();
       if (!db) {
-        log("[pi-subagent] storage open failed; tools will not register");
-        return;
+        throw new Error("storage open failed; refusing to start without Magic Context tools");
       }
       openedDb = db;
       const directory = process.cwd();
@@ -167382,7 +167648,10 @@ function magicContextSubagentExtension(pi) {
       });
       log(`[pi-subagent] registered tools: ctx_search, ctx_memory, ctx_note, ctx_expand` + ` (memory=${cfg.memory.enabled}, embedding=${cfg.embedding.provider !== "off"},` + ` git_commits=${cfg.experimental.git_commit_indexing.enabled}, dreamer_actions=${dreamerActionsEnabled})`);
     } catch (err5) {
-      log(`[pi-subagent] startup failed: ${err5 instanceof Error ? err5.message : String(err5)}`);
+      const message = err5 instanceof Error ? err5.message : String(err5);
+      log(`[pi-subagent] startup failed: ${message}`);
+      process.exitCode = 1;
+      throw err5;
     }
   });
   pi.on("session_shutdown", () => {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@cortexkit/pi-magic-context",
-	"version": "0.21.6",
+	"version": "0.21.8",
 	"type": "module",
 	"description": "Pi coding agent extension for Magic Context — cross-session memory and context management",
 	"main": "dist/index.js",