@cortexkit/pi-magic-context 0.44.0 → 0.44.2

package/dist/{index-2kj55q4e.js → index-gsff513e.js}

@@ -1,7 +1,8 @@
 import {
   getHarness,
+  harnessOwnsOpenCodeStore,
   log
-} from "./index-tfzybajk.js";
+} from "./index-59dmj4zs.js";
 
 // ../plugin/src/features/magic-context/reclaim-protection.ts
 var CTX_REDUCE_KEEP = 3;
@@ -2233,6 +2234,7 @@ function claimOpenCodeDbDiagnosticOnce(surface, resolution) {
 }
 
 // ../plugin/src/shared/sqlite.ts
+import { AsyncLocalStorage } from "node:async_hooks";
 var reportSlowPrivilegedWrite;
 function registerSlowWriteReporter(reporter) {
   reportSlowPrivilegedWrite = reporter;
@@ -2287,6 +2289,62 @@ var sqliteModule = await loadSqliteModule(detectedRuntime);
 var DatabaseImpl = isBun ? sqliteModule.Database : buildNodeSqliteDatabaseClass(sqliteModule.DatabaseSync);
 var trackedSqliteConnections = new Map;
 var nextSqliteConnectionSequence = 1;
+function installTransactionRouting(db, readonly) {
+  const nativeExec = db.exec.bind(db);
+  Object.defineProperty(db, "exec", {
+    configurable: true,
+    writable: true,
+    value: (sql) => {
+      if (/^\s*BEGIN\s+(?:IMMEDIATE|EXCLUSIVE)(?:\s+TRANSACTION)?\s*;?\s*$/i.test(sql)) {
+        retryAcquisition(() => nativeExec(sql), sql.trim());
+        return db;
+      }
+      return nativeExec(sql);
+    }
+  });
+  Object.defineProperty(db, "transaction", {
+    configurable: true,
+    writable: true,
+    value: (fn) => {
+      const make = (mode) => function(...args) {
+        const nested = isInTransaction(db);
+        const savepoint = "mc_tx_sp";
+        db.exec(nested ? `SAVEPOINT ${savepoint}` : `BEGIN ${mode}`);
+        try {
+          const result = fn.apply(this, args);
+          db.exec(nested ? `RELEASE ${savepoint}` : "COMMIT");
+          return result;
+        } catch (error) {
+          if (isInTransaction(db)) {
+            if (nested) {
+              db.exec(`ROLLBACK TO ${savepoint}`);
+              db.exec(`RELEASE ${savepoint}`);
+            } else
+              db.exec("ROLLBACK");
+          }
+          throw error;
+        }
+      };
+      const defaultTransaction = make(readonly ? "DEFERRED" : "IMMEDIATE");
+      const variants = {
+        default: defaultTransaction,
+        deferred: make("DEFERRED"),
+        immediate: make("IMMEDIATE"),
+        exclusive: make("EXCLUSIVE"),
+        database: db
+      };
+      for (const transaction of [
+        variants.default,
+        variants.deferred,
+        variants.immediate,
+        variants.exclusive
+      ]) {
+        Object.assign(transaction, variants);
+      }
+      return defaultTransaction;
+    }
+  });
+}
 function trackSqliteConnection(db, filename, options) {
   const originalClose = db.close.bind(db);
   const sequence = nextSqliteConnectionSequence++;
@@ -2295,6 +2353,7 @@ function trackSqliteConnection(db, filename, options) {
     filename: typeof filename === "string" ? filename : Buffer.isBuffer(filename) ? "<buffer>" : ":memory:",
     readonly: Boolean(options) && typeof options === "object" && (options.readonly === true || options.readOnly === true)
   };
+  installTransactionRouting(db, metadata.readonly);
   Object.defineProperty(db, "close", {
     configurable: true,
     value: (...args) => {
@@ -2318,7 +2377,6 @@ var TrackedDatabase = new Proxy(DatabaseImpl, {
   }
 });
 function buildNodeSqliteDatabaseClass(DatabaseSync) {
-  const SAVEPOINT = "mc_tx_sp";
 
   class NodeSqliteDatabase extends DatabaseSync {
     constructor(filename, options) {
@@ -2337,47 +2395,47 @@ function buildNodeSqliteDatabaseClass(DatabaseSync) {
       }
       return stmt;
     }
-    transaction(fn) {
-      const self = this;
-      const execute = (mode, receiver, args) => {
-        const nested = self.isTransaction === true;
-        self.exec(nested ? `SAVEPOINT ${SAVEPOINT}` : `BEGIN${mode ? ` ${mode}` : ""}`);
-        try {
-          const result = fn.apply(receiver, args);
-          self.exec(nested ? `RELEASE ${SAVEPOINT}` : "COMMIT");
-          return result;
-        } catch (error) {
-          if (nested) {
-            self.exec(`ROLLBACK TO ${SAVEPOINT}`);
-            self.exec(`RELEASE ${SAVEPOINT}`);
-          } else {
-            self.exec("ROLLBACK");
-          }
-          throw error;
-        }
-      };
-      const wrapped = function(...args) {
-        return execute("", this, args);
-      };
-      wrapped.default = function(...args) {
-        return execute("", this, args);
-      };
-      wrapped.deferred = function(...args) {
-        return execute("DEFERRED", this, args);
-      };
-      wrapped.immediate = function(...args) {
-        return execute("IMMEDIATE", this, args);
-      };
-      wrapped.exclusive = function(...args) {
-        return execute("EXCLUSIVE", this, args);
-      };
-      return wrapped;
-    }
   }
   return NodeSqliteDatabase;
 }
 var Database = TrackedDatabase;
 var privilegeDepth = new WeakMap;
+var transformPassScope = new AsyncLocalStorage;
+function isTransientSqliteError(error) {
+  if (!error || typeof error !== "object")
+    return false;
+  const value = error;
+  return typeof value.code === "string" && /^(SQLITE_BUSY|SQLITE_LOCKED)(_|$)/.test(value.code) || typeof value.errcode === "number" && [5, 6].includes(value.errcode & 255);
+}
+
+class SqliteAcquisitionBusyError extends Error {
+  code = "SQLITE_BUSY";
+  stage;
+  constructor(cause, stage = "BEGIN IMMEDIATE") {
+    super("SQLite writer acquisition remained busy after 3 attempts", { cause });
+    this.name = "SqliteAcquisitionBusyError";
+    this.stage = stage;
+  }
+}
+function retryAcquisition(acquire, stage) {
+  if (!transformPassScope.getStore()?.active) {
+    acquire();
+    return;
+  }
+  const delays = [500, 1000];
+  for (let attempt = 0;; attempt++) {
+    try {
+      acquire();
+      return;
+    } catch (error) {
+      if (error instanceof SqliteAcquisitionBusyError || !isTransientSqliteError(error))
+        throw error;
+      if (attempt === delays.length)
+        throw new SqliteAcquisitionBusyError(error, stage);
+      Atomics.wait(new Int32Array(new SharedArrayBuffer(4)), 0, 0, delays[attempt]);
+    }
+  }
+}
 function isInTransaction(db) {
   const candidate = db;
   return candidate.inTransaction === true || candidate.isTransaction === true;
@@ -2390,7 +2448,7 @@ function withPrivilegedWriter(db, operation) {
   if (nested) {
     db.exec(`SAVEPOINT ${savepoint}`);
   } else {
-    db.exec("BEGIN IMMEDIATE");
+    retryAcquisition(() => db.exec("BEGIN IMMEDIATE"), "BEGIN IMMEDIATE");
   }
   privilegeDepth.set(db, previousDepth + 1);
   try {
@@ -2441,7 +2499,7 @@ function closeQuietly(db) {
 
 // ../plugin/src/hooks/magic-context/read-session-db.ts
 function openCodeDbExists() {
-  return openCodeDbPathExists(resolveOpenCodeDbPath());
+  return harnessOwnsOpenCodeStore() && openCodeDbPathExists(resolveOpenCodeDbPath());
 }
 var cachedReadOnlyDb = null;
 function closeCachedReadOnlyDb() {
@@ -2457,6 +2515,9 @@ function closeCachedReadOnlyDb() {
   }
 }
 function getReadOnlySessionDb() {
+  if (!harnessOwnsOpenCodeStore()) {
+    throw new Error("OpenCode session database is not readable from a Pi-compatible process; its history lives in Pi sessions");
+  }
   const resolution = resolveOpenCodeDbPath();
   const dbPath = resolution.path;
   if (!openCodeDbPathExists(resolution)) {
@@ -2499,7 +2560,7 @@ function logProbeFailureOnce(resolution, error) {
 }
 function getMessageTimesFromOpenCodeDb(sessionId, messageIds) {
   const result = new Map;
-  if (messageIds.length === 0)
+  if (messageIds.length === 0 || !harnessOwnsOpenCodeStore())
     return result;
   try {
     withReadOnlySessionDb((db) => {
@@ -2606,11 +2667,23 @@ function readRawSessionMessagesFromDb(db, sessionId) {
   });
 }
 function readRawSessionMessagePageFromDb(db, sessionId, afterOrdinal, limit, finalWatermark = Number.MAX_SAFE_INTEGER) {
+  const messageRows = readRawMessagePageRows(db, sessionId, afterOrdinal, limit, finalWatermark);
+  if (messageRows.length === 0)
+    return [];
+  const placeholders = messageRows.map(() => "?").join(", ");
+  const partRows = db.prepare(`SELECT message_id, data, time_updated
+             FROM part
+             WHERE +session_id = ?
+               AND likelihood(message_id IN (${placeholders}), 0.000001)
+             ORDER BY message_id ASC, time_created ASC, id ASC`).all(sessionId, ...messageRows.map((row) => row.id)).filter(isRawPartRow);
+  return assembleRawMessagePage(messageRows, partRows);
+}
+function readRawMessagePageRows(db, sessionId, afterOrdinal, limit, finalWatermark) {
   const remaining = Math.max(0, Math.floor(finalWatermark) - Math.floor(afterOrdinal));
   const pageSize = Math.min(Math.max(1, Math.floor(limit)), remaining);
   if (pageSize === 0)
     return [];
-  const messageRows = db.prepare(`SELECT id, data, time_created, time_updated
+  return db.prepare(`SELECT id, data, time_created, time_updated
              FROM message
              WHERE session_id = ?
                AND NOT (
@@ -2626,14 +2699,8 @@ function readRawSessionMessagePageFromDb(db, sessionId, afterOrdinal, limit, fin
     ...row,
     ordinal: Math.floor(afterOrdinal) + index + 1
   }));
-  if (messageRows.length === 0)
-    return [];
-  const placeholders = messageRows.map(() => "?").join(", ");
-  const partRows = db.prepare(`SELECT message_id, data, time_updated
-             FROM part
-             WHERE +session_id = ?
-               AND likelihood(message_id IN (${placeholders}), 0.000001)
-             ORDER BY message_id ASC, time_created ASC, id ASC`).all(sessionId, ...messageRows.map((row) => row.id)).filter(isRawPartRow);
+}
+function assembleRawMessagePage(messageRows, partRows) {
   const partsByMessageId = new Map;
   for (const part of partRows) {
     const list = partsByMessageId.get(part.message_id) ?? [];
@@ -2652,6 +2719,49 @@ function readRawSessionMessagePageFromDb(db, sessionId, afterOrdinal, limit, fin
     };
   });
 }
+var RAW_SUMMARY_TEXT_MAX_CHARS = 8192;
+var RAW_SUMMARY_ARG_MAX_CHARS = 512;
+var RAW_SUMMARY_TOOL_INPUT_KEYS = [
+  "description",
+  "filePath",
+  "path",
+  "pattern",
+  "query",
+  "symbol",
+  "module",
+  "action"
+];
+function summaryStringField(jsonPath) {
+  return `CASE WHEN json_type(data, '${jsonPath}') = 'text' THEN substr(json_extract(data, '${jsonPath}'), 1, ${RAW_SUMMARY_ARG_MAX_CHARS}) END`;
+}
+var RAW_SUMMARY_PART_DATA_SQL = `CASE
+    WHEN json_extract(data, '$.type') = 'text'
+        THEN json_set(data, '$.text', substr(json_extract(data, '$.text'), 1, ${RAW_SUMMARY_TEXT_MAX_CHARS}))
+    ELSE json_object(
+        'type', 'tool',
+        'tool', ${summaryStringField("$.tool")},
+        'callID', ${summaryStringField("$.callID")},
+        'state', json_object(
+            'status', ${summaryStringField("$.state.status")},
+            'input', json_object(${RAW_SUMMARY_TOOL_INPUT_KEYS.map((key) => `'${key}', ${summaryStringField(`$.state.input.${key}`)}`).join(", ")}),
+            'metadata', json_object('description', ${summaryStringField("$.state.metadata.description")})
+        )
+    )
+END`;
+function readRawSessionMessageSummaryPageFromDb(db, sessionId, afterOrdinal, limit, finalWatermark = Number.MAX_SAFE_INTEGER) {
+  const messageRows = readRawMessagePageRows(db, sessionId, afterOrdinal, limit, finalWatermark);
+  if (messageRows.length === 0)
+    return [];
+  const placeholders = messageRows.map(() => "?").join(", ");
+  const partRows = db.prepare(`SELECT message_id, ${RAW_SUMMARY_PART_DATA_SQL} AS data, time_updated
+             FROM part
+             WHERE +session_id = ?
+               AND likelihood(message_id IN (${placeholders}), 0.000001)
+               AND json_valid(data) = 1
+               AND json_extract(data, '$.type') IN ('text', 'tool')
+             ORDER BY message_id ASC, time_created ASC, id ASC`).all(sessionId, ...messageRows.map((row) => row.id)).filter(isRawPartRow);
+  return assembleRawMessagePage(messageRows, partRows);
+}
 function countRawSessionMessageOrdinalsFromDb(db, sessionId) {
   const row = db.prepare(`SELECT COUNT(*) AS count
              FROM message
@@ -3340,6 +3450,69 @@ function isFilePart(part) {
   return p.type === "file" && typeof p.url === "string";
 }
 
+// ../plugin/src/features/magic-context/dreamer/token-budget.ts
+class DreamTokenBudgetExceeded extends Error {
+  sessionId;
+  spent;
+  constructor(sessionId, spent) {
+    super(`Dreamer child ${sessionId} spent ${spent} prompt tokens (token_budget)`);
+    this.sessionId = sessionId;
+    this.spent = spent;
+    this.name = "DreamTokenBudgetExceeded";
+  }
+}
+var TOKEN_BUDGET_FINALIZE_MESSAGE = "You're out of token budget. Stop investigating: make no more tool calls and output your result now, covering only what you've already checked; leave the rest out.";
+var TOKEN_BUDGET_TOOL_REFUSAL = "Out of token budget: no more tool calls. Output your result now.";
+var finalizingChildren = new Map;
+function registerBudgetFinalizeChild(sessionId, guard) {
+  finalizingChildren.set(sessionId, guard);
+}
+function releaseBudgetFinalizeChild(sessionId) {
+  finalizingChildren.delete(sessionId);
+}
+function createDreamTokenBudget(budget) {
+  if (!Number.isSafeInteger(budget) || budget <= 0)
+    throw new Error("Invalid token budget");
+  let spent = 0;
+  let finalizeFired = false;
+  let refusedCalls = 0;
+  let forcedStop = false;
+  const snapshot = () => ({
+    budget,
+    spent,
+    finalizeFired,
+    refusedCalls,
+    hardStopped: forcedStop || spent >= budget || refusedCalls >= 2
+  });
+  return {
+    snapshot,
+    charge(input, cacheRead, cacheWrite, completed = false, canFinalize = true) {
+      for (const value of [input, cacheRead, cacheWrite]) {
+        if (!Number.isSafeInteger(value) || value < 0)
+          throw new Error("Invalid prompt usage");
+      }
+      spent += input + cacheRead + cacheWrite;
+      if (spent >= budget)
+        return "stop";
+      if (!completed && !finalizeFired && spent >= budget * 0.8) {
+        if (!canFinalize) {
+          forcedStop = true;
+          return "stop";
+        }
+        finalizeFired = true;
+        return "finalize";
+      }
+      return "continue";
+    },
+    refuseTool() {
+      if (!finalizeFired)
+        return null;
+      refusedCalls += 1;
+      return { message: TOKEN_BUDGET_TOOL_REFUSAL, hardStopped: snapshot().hardStopped };
+    }
+  };
+}
+
 // ../plugin/src/hooks/magic-context/dropped-input-guard.ts
 var LEGACY_TRUNCATED_VALUE = /^[\s\S]{0,5}\.\.\.\[truncated\]$/;
 var MAX_QUOTED_ARGUMENTS = 160;
@@ -4239,6 +4412,45 @@ function readRawSessionMessageRangeFromSource(sessionId, fromOrdinal, toOrdinal)
   }
   return messages;
 }
+var RAW_MESSAGE_VISIT_PAGE_SIZE = 50;
+function visitRawSessionMessages2(sessionId, fromOrdinal, toOrdinal, visit, options = {}) {
+  const from = Math.max(1, Math.floor(fromOrdinal));
+  const to = Math.floor(toOrdinal);
+  if (to < from)
+    return;
+  const provider = sessionProviders.get(sessionId);
+  if (provider && !provider.readMessagePage) {
+    for (const message of provider.readMessages()) {
+      if (message.ordinal < from || message.ordinal > to)
+        continue;
+      if (!visit(message))
+        return;
+    }
+    return;
+  }
+  if (!provider && !openCodeDbExists())
+    return;
+  const pageSize = Math.max(1, Math.floor(options.pageSize ?? RAW_MESSAGE_VISIT_PAGE_SIZE));
+  let afterOrdinal = from - 1;
+  while (afterOrdinal < to) {
+    const limit = Math.min(pageSize, to - afterOrdinal);
+    const cursor = afterOrdinal;
+    const page = provider?.readMessagePage ? provider.readMessagePage(cursor, limit, to) : withReadOnlySessionDb((db) => options.summary ? readRawSessionMessageSummaryPageFromDb(db, sessionId, cursor, limit, to) : readRawSessionMessagePageFromDb(db, sessionId, cursor, limit, to));
+    if (page.length === 0)
+      return;
+    let nextOrdinal = afterOrdinal;
+    for (const message of page) {
+      if (message.ordinal < from || message.ordinal > to)
+        continue;
+      if (!visit(message))
+        return;
+      nextOrdinal = Math.max(nextOrdinal, message.ordinal);
+    }
+    if (nextOrdinal <= afterOrdinal)
+      return;
+    afterOrdinal = nextOrdinal;
+  }
+}
 function readRawSessionMessageRange2(sessionId, fromOrdinal, toOrdinal) {
   const from = Math.max(1, Math.floor(fromOrdinal));
   const to = Math.floor(toOrdinal);
@@ -4777,4 +4989,4 @@ function readSessionChunk2(sessionId, tokenBudget, offset = 1, eligibleEndOrdina
     toolResultBoundaries
   };
 }
-export { CTX_REDUCE_KEEP, newestCtxReduceTagNumbers, textMentionsRecentCommit, removeSystemReminders, hasMeaningfulUserText2, extractTexts2, extractToolCallSummaries, preloadTokenizer, estimateTokens, normalizeText, stripWellFormedLeadingTagPrefix, stripPersistedAssistantText, byteSize, stripTagPrefix, peelLeadingMcTagNotation, prependTag, isRecord, stableStringify, estimateImageTokensFromDataUrl, TODO_STATUSES, TODO_PRIORITIES, TITLE_DONE_STATUSES, normalizeTodoStateJson, buildSyntheticTodoPart, stripChannel1ReminderSpans, compareMeasuredTailPrefix, formatTailHygienePrefixMismatch, freezeTailHygieneMeasurement, effectiveTailHygiene, CHANNEL1_SENTINEL, CHANNEL1_FLOOR_TOKENS, decideChannel1, evaluateChannel2, reclaimableToolOutputCount, buildChannel2Reminder, buildChannel1Reminder, planEmergencyDrop, measureEmergencyTag, updateTagByteSize, getRecentTagOwnerMessageIds, AGE_RECLAIM_MIN_TOKENS, getOldestActiveUnprotectedToolTags, getActiveToolTagsForAgeReclaim, getTriggerTagTokenUpperBound, updateTagInputByteSize, updateTagTokenCount, getPersistedToolTagAccounting, getAllStatusTagTokenTotalsFlat, updateTagInputTokenCount, tagTokenCountIsNull, backfillTagTokenCounts, insertTag, updateTagStatus, getInertWhitespaceAssistantTags, updateTagDropMode, updateCavemanDepth, hasPiFallbackMessageTags, findAdoptableFallbackTags, hasPiFallbackToolOwnerTags, findPiFallbackToolOwnerTags, adoptPiFallbackToolOwnerTag, adoptPiFallbackMessageTag, getMaxTagNumberBySession, getTagNumberByMessageId, getAssignableTagNumberByMessageId, deriveTagLoadFloor, TAG_SELECT_COLUMNS, getTagsBySession, getActiveTagsBySession, getTagsForPendingOperations, getTagsByNumbers, getDroppedTagsByNumbers, getMaxDroppedTagNumber, getToolTagNumberByOwner, getNullOwnerToolTag, adoptNullOwnerToolTag, retreatPastHostUnservedRows, resolveOpenCodeDbPath, hasV1MessageTables, assertOpenCodeStoreGeneration, openCodeDbPathExists, getOpenCodeDbProbeDescriptions, recordOpenCodeDbReadFailure, clearOpenCodeDbReadFailure, claimOpenCodeDbDiagnosticOnce, registerSlowWriteReporter, detectSqliteRuntime, Database, withPrivilegedWriter, closeQuietly, withReadOnlySessionDb, completedToolArcCrossesBoundary, estimateTrueRawMessageTokens, buildToolArcs, fenceBoundaryForCompletedToolArcs, fenceBoundaryForToolArcs, buildTrueRawTokenIndex, computeRawRangeFingerprint, invalidateTrueRawTokenCache, droppedInputMarker, createDroppedInputGuard, isEditTool, applyEditMarkerToInput, setDatabase, loadToolDefinitionMeasurements, resolveDecisionCalibration, providerMass, localBudget, calibrationForModelKey, historyLocalBudget, SKELETON_REAL_INPUT_MAX_BYTES, toolInputStringBytes, resolveHostServedBoundaryId2, hasRawMessageProvider2, setRawMessageProvider2, withRawMessageProvider2, cleanUserText2, withRawSessionMessageCache2, readRawSessionMessages2, readRawSessionMessagePage2, getRawSessionMessageOrdinalCount2, readRawSessionMessageRange2, primeTailRawMessageCache2, getCachedAbsoluteMessageCount2, primeInMemoryTailRawMessageCache2, readRawSessionMessageIdOrdinalsForRange2, hasRawSessionMessageById2, readRawSessionMessageOrdinalById2, compareRawSessionMessageOrder2, getRawSessionMessageCount2, getRawSessionTagKeysThrough2, getLegacyProtectedTailStartOrdinal2, readSessionChunk2 };
+export { CTX_REDUCE_KEEP, newestCtxReduceTagNumbers, textMentionsRecentCommit, removeSystemReminders, hasMeaningfulUserText2, extractTexts2, extractToolCallSummaries, preloadTokenizer, estimateTokens, normalizeText, stripWellFormedLeadingTagPrefix, stripPersistedAssistantText, byteSize, stripTagPrefix, peelLeadingMcTagNotation, prependTag, isRecord, stableStringify, estimateImageTokensFromDataUrl, TODO_STATUSES, TODO_PRIORITIES, TITLE_DONE_STATUSES, normalizeTodoStateJson, buildSyntheticTodoPart, stripChannel1ReminderSpans, compareMeasuredTailPrefix, formatTailHygienePrefixMismatch, freezeTailHygieneMeasurement, effectiveTailHygiene, CHANNEL1_SENTINEL, CHANNEL1_FLOOR_TOKENS, decideChannel1, evaluateChannel2, reclaimableToolOutputCount, buildChannel2Reminder, buildChannel1Reminder, planEmergencyDrop, measureEmergencyTag, updateTagByteSize, getRecentTagOwnerMessageIds, AGE_RECLAIM_MIN_TOKENS, getOldestActiveUnprotectedToolTags, getActiveToolTagsForAgeReclaim, getTriggerTagTokenUpperBound, updateTagInputByteSize, updateTagTokenCount, getPersistedToolTagAccounting, getAllStatusTagTokenTotalsFlat, updateTagInputTokenCount, tagTokenCountIsNull, backfillTagTokenCounts, insertTag, updateTagStatus, getInertWhitespaceAssistantTags, updateTagDropMode, updateCavemanDepth, hasPiFallbackMessageTags, findAdoptableFallbackTags, hasPiFallbackToolOwnerTags, findPiFallbackToolOwnerTags, adoptPiFallbackToolOwnerTag, adoptPiFallbackMessageTag, getMaxTagNumberBySession, getTagNumberByMessageId, getAssignableTagNumberByMessageId, deriveTagLoadFloor, TAG_SELECT_COLUMNS, getTagsBySession, getActiveTagsBySession, getTagsForPendingOperations, getTagsByNumbers, getDroppedTagsByNumbers, getMaxDroppedTagNumber, getToolTagNumberByOwner, getNullOwnerToolTag, adoptNullOwnerToolTag, retreatPastHostUnservedRows, resolveOpenCodeDbPath, hasV1MessageTables, assertOpenCodeStoreGeneration, openCodeDbPathExists, getOpenCodeDbProbeDescriptions, recordOpenCodeDbReadFailure, clearOpenCodeDbReadFailure, claimOpenCodeDbDiagnosticOnce, registerSlowWriteReporter, detectSqliteRuntime, Database, withPrivilegedWriter, closeQuietly, withReadOnlySessionDb, RAW_SUMMARY_TEXT_MAX_CHARS, completedToolArcCrossesBoundary, estimateTrueRawMessageTokens, buildToolArcs, fenceBoundaryForCompletedToolArcs, fenceBoundaryForToolArcs, buildTrueRawTokenIndex, computeRawRangeFingerprint, invalidateTrueRawTokenCache, DreamTokenBudgetExceeded, TOKEN_BUDGET_FINALIZE_MESSAGE, TOKEN_BUDGET_TOOL_REFUSAL, registerBudgetFinalizeChild, releaseBudgetFinalizeChild, createDreamTokenBudget, droppedInputMarker, createDroppedInputGuard, isEditTool, applyEditMarkerToInput, setDatabase, loadToolDefinitionMeasurements, resolveDecisionCalibration, providerMass, localBudget, calibrationForModelKey, historyLocalBudget, SKELETON_REAL_INPUT_MAX_BYTES, toolInputStringBytes, resolveHostServedBoundaryId2, hasRawMessageProvider2, setRawMessageProvider2, withRawMessageProvider2, cleanUserText2, withRawSessionMessageCache2, readRawSessionMessages2, readRawSessionMessagePage2, getRawSessionMessageOrdinalCount2, visitRawSessionMessages2, readRawSessionMessageRange2, primeTailRawMessageCache2, getCachedAbsoluteMessageCount2, primeInMemoryTailRawMessageCache2, readRawSessionMessageIdOrdinalsForRange2, hasRawSessionMessageById2, readRawSessionMessageOrdinalById2, compareRawSessionMessageOrder2, getRawSessionMessageCount2, getRawSessionTagKeysThrough2, getLegacyProtectedTailStartOrdinal2, readSessionChunk2 };

package/dist/index-n8kg008j.js

@@ -0,0 +1,245 @@
+// src/subagent-telemetry.ts
+import { createHash } from "node:crypto";
+
+// ../plugin/src/agents/permissions.ts
+function applyDisallowedTools(defaults, disallowed) {
+  if (disallowed.includes("*"))
+    return [];
+  return defaults.filter((t) => !disallowed.includes(t));
+}
+
+// ../plugin/src/agents/hidden-agent-registrations.ts
+var HIDDEN_AGENT_DESCRIPTION = "Internal Magic Context maintenance agent. Not for general tasks — do not select for user work.";
+function buildHiddenAgentRegistrations(args) {
+  const historianAllowedTools = applyDisallowedTools(["read", "aft_outline", "aft_zoom", "aft_search"], args.historianDisallowed);
+  return [
+    {
+      id: "dreamer",
+      mode: "primary",
+      hidden: true,
+      description: HIDDEN_AGENT_DESCRIPTION,
+      prompt: args.dreamerPrompt,
+      allowedTools: ["ctx_memory"],
+      maxSteps: 150,
+      overrides: args.dreamerOverrides,
+      lockPermissions: true
+    },
+    {
+      id: "dreamer-docs",
+      mode: "primary",
+      hidden: true,
+      description: HIDDEN_AGENT_DESCRIPTION,
+      prompt: args.dreamerPrompt,
+      allowedTools: ["read", "grep", "glob", "aft_outline", "aft_zoom", "aft_search"],
+      maxSteps: 60,
+      overrides: args.dreamerOverrides,
+      lockPermissions: true
+    },
+    {
+      id: "dreamer-reviewer",
+      mode: "primary",
+      hidden: true,
+      description: HIDDEN_AGENT_DESCRIPTION,
+      prompt: args.dreamerPrompt,
+      allowedTools: [],
+      maxSteps: 4,
+      overrides: args.dreamerOverrides,
+      lockPermissions: true
+    },
+    {
+      id: "dreamer-retrospective",
+      mode: "primary",
+      hidden: true,
+      description: HIDDEN_AGENT_DESCRIPTION,
+      prompt: args.dreamerPrompt,
+      allowedTools: ["ctx_search"],
+      maxSteps: 40,
+      overrides: args.dreamerOverrides,
+      lockPermissions: true
+    },
+    {
+      id: "dreamer-primer-investigator",
+      mode: "primary",
+      hidden: true,
+      description: HIDDEN_AGENT_DESCRIPTION,
+      prompt: args.dreamerPrompt,
+      allowedTools: [
+        "read",
+        "grep",
+        "glob",
+        "aft_outline",
+        "aft_zoom",
+        "aft_search",
+        "ctx_search"
+      ],
+      maxSteps: 40,
+      overrides: args.dreamerOverrides,
+      lockPermissions: true
+    },
+    {
+      id: "dreamer-memory-mapper",
+      mode: "primary",
+      hidden: true,
+      description: HIDDEN_AGENT_DESCRIPTION,
+      prompt: args.dreamerPrompt,
+      allowedTools: ["read", "grep", "glob", "aft_outline", "aft_zoom", "aft_search"],
+      maxSteps: 60,
+      overrides: args.dreamerOverrides,
+      lockPermissions: true
+    },
+    {
+      id: "dreamer-classifier",
+      mode: "primary",
+      hidden: true,
+      description: HIDDEN_AGENT_DESCRIPTION,
+      prompt: args.dreamerPrompt,
+      allowedTools: [],
+      maxSteps: 4,
+      overrides: args.dreamerOverrides,
+      lockPermissions: true
+    },
+    {
+      id: "smart-note-compiler",
+      mode: "primary",
+      hidden: true,
+      description: HIDDEN_AGENT_DESCRIPTION,
+      prompt: args.smartNoteCompilerPrompt,
+      allowedTools: [],
+      maxSteps: 8,
+      overrides: args.dreamerOverrides,
+      lockPermissions: true
+    },
+    {
+      id: "historian",
+      mode: "primary",
+      hidden: true,
+      description: HIDDEN_AGENT_DESCRIPTION,
+      prompt: args.historianPrompt,
+      allowedTools: historianAllowedTools,
+      maxSteps: 40,
+      overrides: args.historianOverrides
+    },
+    {
+      id: "historian-recomp",
+      mode: "primary",
+      hidden: true,
+      description: HIDDEN_AGENT_DESCRIPTION,
+      prompt: args.historianRecompPrompt ?? args.historianPrompt,
+      allowedTools: historianAllowedTools,
+      maxSteps: 40,
+      overrides: args.historianOverrides
+    },
+    {
+      id: "historian-editor",
+      mode: "primary",
+      hidden: true,
+      description: HIDDEN_AGENT_DESCRIPTION,
+      prompt: args.historianEditorPrompt,
+      allowedTools: historianAllowedTools,
+      maxSteps: 40,
+      overrides: args.historianOverrides
+    }
+  ];
+}
+
+// src/subagent-telemetry.ts
+var stepCaps = new Map(buildHiddenAgentRegistrations({
+  dreamerPrompt: undefined,
+  historianPrompt: undefined,
+  historianEditorPrompt: undefined,
+  historianDisallowed: []
+}).map(({ id, maxSteps }) => [id, maxSteps]));
+function subagentStepCap(agent) {
+  return stepCaps.get(agent.replace(/^magic-context-/, ""));
+}
+function promptFingerprint(prompt) {
+  return {
+    bytes: Buffer.byteLength(prompt, "utf8"),
+    sha256: createHash("sha256").update(prompt).digest("hex")
+  };
+}
+function record(value) {
+  return value !== null && typeof value === "object" ? value : {};
+}
+
+class SubagentTelemetry {
+  log;
+  steps = 0;
+  intended;
+  effective = null;
+  tools = Object.create(null);
+  promptTokens = {
+    first: null,
+    last: null,
+    max: null
+  };
+  warned = false;
+  streamedSteps = 0;
+  terminalSeen = false;
+  results = new Set;
+  constructor(prompt, log) {
+    this.log = log;
+    this.intended = promptFingerprint(prompt);
+  }
+  observe(value) {
+    const event = record(value);
+    if (event.type === "mc_system_prompt" && typeof event.bytes === "number" && typeof event.sha256 === "string" && /^[a-f0-9]{64}$/.test(event.sha256)) {
+      this.effective = { bytes: event.bytes, sha256: event.sha256 };
+      if (!this.warned && event.sha256 !== this.intended.sha256) {
+        this.warned = true;
+        this.log(`subagent_system_prompt_${event.containsIntended === true ? "extended" : "replaced"} ${JSON.stringify({ intended: this.intended, effective: this.effective, extension: null })}`);
+      }
+    }
+    if (event.type === "message_end") {
+      if (record(event.message).role === "assistant") {
+        this.streamedSteps++;
+        this.assistant(event.message);
+      }
+    }
+    if (event.type === "agent_end" && Array.isArray(event.messages) && !this.terminalSeen) {
+      this.terminalSeen = true;
+      const assistants = event.messages.filter((message) => record(message).role === "assistant");
+      for (const message of assistants.slice(this.streamedSteps))
+        this.assistant(message);
+    }
+    if (event.type === "tool_execution_end" && typeof event.toolCallId === "string" && typeof event.toolName === "string" && !this.results.has(event.toolCallId)) {
+      this.results.add(event.toolCallId);
+      const tool = this.tool(event.toolName);
+      const content = record(event.result).content;
+      if (Array.isArray(content))
+        for (const part of content) {
+          const block = record(part);
+          const text = typeof block.text === "string" ? block.text : typeof block.data === "string" ? block.data : "";
+          tool.outputBytes += Buffer.byteLength(text, "utf8");
+        }
+    }
+  }
+  tool(name) {
+    this.tools[name] ??= { calls: 0, outputBytes: 0 };
+    return this.tools[name];
+  }
+  assistant(value) {
+    const message = record(value);
+    this.steps++;
+    const usage = record(message.usage);
+    const fields = [usage.input, usage.cacheRead, usage.cacheWrite];
+    const known = fields.some((value) => typeof value === "number" && Number.isFinite(value));
+    const tokens = known ? fields.reduce((sum, value) => sum + (typeof value === "number" && Number.isFinite(value) ? value : 0), 0) : null;
+    if (this.steps === 1)
+      this.promptTokens.first = tokens;
+    this.promptTokens.last = tokens;
+    if (tokens !== null)
+      this.promptTokens.max = Math.max(this.promptTokens.max ?? 0, tokens);
+    if (Array.isArray(message.content))
+      for (const part of message.content) {
+        const call = record(part);
+        if (call.type === "toolCall" && typeof call.name === "string")
+          this.tool(call.name).calls++;
+      }
+  }
+  finish(agent, outcome, cap) {
+    this.log(`subagent_telemetry ${JSON.stringify({ agent, outcome, cap, steps: this.steps, tools: this.tools, promptTokens: this.promptTokens, intended: this.intended, effective: this.effective, provenanceSource: this.effective ? "pi_context_system_prompt" : "unavailable" })}`);
+  }
+}
+
+export { subagentStepCap, promptFingerprint, SubagentTelemetry };