@cortexkit/opencode-magic-context 0.44.0 → 0.44.2

package/dist/{index-cj5qe2nw.js → index-3x9caerx.js}

@@ -1,6 +1,6 @@
 import {
   sessionLog
-} from "./index-tjq5jztd.js";
+} from "./index-3bygzk2d.js";
 import {
   COMPARTMENT_LEASE_RENEWAL_MS,
   acquireCompartmentLease,
@@ -20,16 +20,18 @@ import {
   runCompartmentAgent,
   getActiveCompartmentRun,
   markActiveCompartmentRunPublished,
-  registerActiveCompartmentRun,
-  setRecompStarting2,
-  setRecompTerminal2
-} from "./index-rdey8znw.js";
+  registerActiveCompartmentRun
+} from "./index-ttdn47aq.js";
 import {
   pushNotification2
 } from "./index-jdxxq8cz.js";
 import {
   sendStatusNotification2
-} from "./index-c906f3ff.js";
+} from "./index-r9vkp4n9.js";
+import {
+  setRecompStarting2,
+  setRecompTerminal2
+} from "./index-40s1xz22.js";
 
 // src/hooks/magic-context/wrapup-orchestrator.ts
 import * as crypto from "node:crypto";

package/dist/index-40s1xz22.js

@@ -0,0 +1,125 @@
+import {
+  renderUserFacingFailure,
+  userFacingFailureCode,
+  clearEmergencyRecovery,
+  dropSlot,
+  executeContextRecomp
+} from "./index-ttdn47aq.js";
+import {
+  sessionLog
+} from "./index-3bygzk2d.js";
+
+// src/hooks/magic-context/recomp-orchestrator.ts
+function resolveLiveModelKey(liveSessionState, sessionId) {
+  const model = liveSessionState.liveModelBySession.get(sessionId);
+  return model ? `${model.providerID}/${model.modelID}` : undefined;
+}
+function isRecompFailure(message) {
+  return /—\s*(Failed|Skipped)/.test(message);
+}
+function isRecompSkip(message) {
+  return /—\s*Skipped|already mutating compartment state|already running/i.test(message);
+}
+function extractRecompReason(raw) {
+  const meaningful = raw.split(`
+`).map((l) => l.trim()).filter((l) => l.length > 0 && !l.startsWith("#"));
+  return meaningful.join(" ").trim() || "Recomp finished";
+}
+var RECOMP_DONE_GRACE_MS = 30000;
+function setRecompStarting2(liveSessionState, sessionId, note, kind = "recomp") {
+  dropSlot(sessionId, "recomp-start");
+  liveSessionState.recompProgressBySession.set(sessionId, {
+    sessionId,
+    kind,
+    phase: "recomp",
+    processedMessages: 0,
+    totalMessages: 0,
+    passCount: 0,
+    compartmentsCreated: 0,
+    startedAt: Date.now(),
+    updatedAt: Date.now(),
+    note
+  });
+}
+function setRecompTerminal2(liveSessionState, sessionId, phase, message) {
+  const existing = liveSessionState.recompProgressBySession.get(sessionId);
+  liveSessionState.recompProgressBySession.set(sessionId, {
+    sessionId,
+    kind: existing?.kind ?? "recomp",
+    phase,
+    processedMessages: existing?.processedMessages ?? 0,
+    totalMessages: existing?.totalMessages ?? 0,
+    passCount: existing?.passCount ?? 0,
+    compartmentsCreated: existing?.compartmentsCreated ?? 0,
+    startedAt: existing?.startedAt ?? Date.now(),
+    updatedAt: Date.now(),
+    message
+  });
+  if (phase === "done" || phase === "skipped") {
+    const t = setTimeout(() => {
+      const cur = liveSessionState.recompProgressBySession.get(sessionId);
+      if (cur?.phase === phase)
+        liveSessionState.recompProgressBySession.delete(sessionId);
+    }, RECOMP_DONE_GRACE_MS);
+    t.unref?.();
+  }
+}
+function buildRecompDeps(ctx, sessionId) {
+  return {
+    client: ctx.client,
+    hiddenCompletionExecutor: ctx.hiddenCompletionExecutor,
+    db: ctx.db,
+    sessionId,
+    historianChunkTokens: ctx.historianChunkTokens,
+    historianTimeoutMs: ctx.historianTimeoutMs,
+    directory: ctx.directory,
+    memoryEnabled: ctx.memoryEnabled,
+    autoPromote: ctx.autoPromote,
+    model: ctx.historianModel,
+    historianContextLimit: ctx.historianContextLimit,
+    historianMaxOutputTokens: ctx.historianMaxOutputTokens,
+    fallbackModels: ctx.fallbackModels,
+    language: ctx.language,
+    fallbackModelId: ctx.fallbackModelId ?? resolveLiveModelKey(ctx.liveSessionState, sessionId),
+    historianTwoPass: ctx.historianTwoPass,
+    ensureProjectRegistered: ctx.ensureProjectRegistered,
+    getNotificationParams: () => ctx.getNotificationParams(sessionId),
+    onCompartmentStatePublished: (sid) => {
+      ctx.liveSessionState.historyRefreshSessions.add(sid);
+      ctx.liveSessionState.pendingMaterializationSessions.add(sid);
+    },
+    onDeferredMarkerPending: (sid) => {
+      ctx.liveSessionState.deferredHistoryRefreshSessions.add(sid);
+    },
+    onRecompProgress: (p) => {
+      const prevKind = ctx.liveSessionState.recompProgressBySession.get(sessionId)?.kind ?? "recomp";
+      ctx.liveSessionState.recompProgressBySession.set(sessionId, {
+        ...p,
+        kind: p.kind ?? prevKind
+      });
+    }
+  };
+}
+async function runManagedRecomp2(ctx, sessionId, options) {
+  setRecompStarting2(ctx.liveSessionState, sessionId, "Starting recomp…", "recomp");
+  try {
+    const message = await executeContextRecomp(buildRecompDeps(ctx, sessionId), options);
+    const terminalPhase = isRecompSkip(message) ? "skipped" : isRecompFailure(message) ? "failed" : "done";
+    if (terminalPhase === "done") {
+      try {
+        clearEmergencyRecovery(ctx.db, sessionId);
+      } catch {}
+    }
+    setRecompTerminal2(ctx.liveSessionState, sessionId, terminalPhase, extractRecompReason(message));
+    return message;
+  } catch (error) {
+    const failure = renderUserFacingFailure("recomp_unavailable");
+    sessionLog(sessionId, `recomp failed code=${userFacingFailureCode("recomp_unavailable")}`, error);
+    setRecompTerminal2(ctx.liveSessionState, sessionId, "failed", failure);
+    return `## Magic Recomp — Failed
+
+${failure}`;
+  }
+}
+
+export { setRecompStarting2, setRecompTerminal2, runManagedRecomp2 };

package/dist/{index-pye48m79.js → index-egkfvhms.js}

@@ -1,6 +1,9 @@
+import {
+  harnessOwnsOpenCodeStore
+} from "./index-s70wtd8k.js";
 import {
   log
-} from "./index-tjq5jztd.js";
+} from "./index-3bygzk2d.js";
 
 // src/shared/opencode-db-path.ts
 import { existsSync, readdirSync, statSync } from "node:fs";
@@ -204,6 +207,7 @@ function closeQuietly(db) {
 }
 
 // src/shared/sqlite.ts
+import { AsyncLocalStorage } from "node:async_hooks";
 import { statSync as statSync2 } from "node:fs";
 var reportSlowPrivilegedWrite;
 function registerSlowWriteReporter(reporter) {
@@ -259,6 +263,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++;
@@ -267,6 +327,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) => {
@@ -290,7 +351,6 @@ var TrackedDatabase = new Proxy(DatabaseImpl, {
   }
 });
 function buildNodeSqliteDatabaseClass(DatabaseSync) {
-  const SAVEPOINT = "mc_tx_sp";
 
   class NodeSqliteDatabase extends DatabaseSync {
     constructor(filename, options) {
@@ -309,42 +369,6 @@ 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;
 }
@@ -429,6 +453,63 @@ function getSqliteMemoryStats() {
   };
 }
 var privilegeDepth = new WeakMap;
+var transformPassScope = new AsyncLocalStorage;
+function withSqliteTransformPass(operation) {
+  const lease = { active: true };
+  return transformPassScope.run(lease, () => {
+    try {
+      const result = operation();
+      if (result && typeof result.then === "function") {
+        return Promise.resolve(result).finally(() => {
+          lease.active = false;
+        });
+      }
+      lease.active = false;
+      return result;
+    } catch (error) {
+      lease.active = false;
+      throw error;
+    }
+  });
+}
+function withoutSqliteTransformPass(operation) {
+  return transformPassScope.run(undefined, operation);
+}
+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;
@@ -441,7 +522,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 {
@@ -483,7 +564,7 @@ function withPrivilegedWriter(db, operation) {
 
 // src/hooks/magic-context/read-session-db.ts
 function openCodeDbExists() {
-  return openCodeDbPathExists(resolveOpenCodeDbPath());
+  return harnessOwnsOpenCodeStore() && openCodeDbPathExists(resolveOpenCodeDbPath());
 }
 var cachedReadOnlyDb = null;
 function closeCachedReadOnlyDb() {
@@ -499,6 +580,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)) {
@@ -684,6 +768,8 @@ function logProbeFailureOnce(resolution, error) {
 }
 function resolvedDbIsAvailable() {
   const resolution = resolveOpenCodeDbPath();
+  if (!harnessOwnsOpenCodeStore())
+    return { resolution, available: false };
   const available = openCodeDbPathExists(resolution);
   if (available)
     clearOpenCodeDbReadFailure(resolution.path);
@@ -825,7 +911,7 @@ function hasUnansweredRealUser(db, sessionId) {
 }
 function getMessageTimesFromOpenCodeDb(sessionId, messageIds) {
   const result = new Map;
-  if (messageIds.length === 0)
+  if (messageIds.length === 0 || !harnessOwnsOpenCodeStore())
     return result;
   try {
     withReadOnlySessionDb((db) => {
@@ -843,6 +929,8 @@ function getMessageTimesFromOpenCodeDb(sessionId, messageIds) {
   return result;
 }
 function findLastAssistantModelFromOpenCodeDb(sessionId) {
+  if (!harnessOwnsOpenCodeStore())
+    return null;
   const resolution = resolveOpenCodeDbPath();
   try {
     if (!openCodeDbPathExists(resolution)) {
@@ -899,4 +987,4 @@ function findLastAssistantModelFromOpenCodeDb(sessionId) {
   }
 }
 
-export { resolveOpenCodeDbPath, hasV1MessageTables, assertOpenCodeStoreGeneration, openCodeDbPathExists, getOpenCodeDbProbeDescriptions, formatOpenCodeDbMissingBanner, formatOpenCodeDbMissingStatusLine, formatOpenCodeDbReadFailureStatusLine, recordOpenCodeDbReadFailure, clearOpenCodeDbReadFailure, getOpenCodeDbReadFailure, claimOpenCodeDbDiagnosticOnce, registerSlowWriteReporter, detectSqliteRuntime, Database, getSqliteMemoryStats, withPrivilegedWriter, closeQuietly, openCodeDbExists, withReadOnlySessionDb, getRawSessionMessageCountFromDb, observeOpenCodeTurnEvent, clearTrackedOpenCodeSession, assistantAwaitingTools, shouldHoldIgnoredNotification, latestPersistedMessageForRecovery, getMessageTimesFromOpenCodeDb, findLastAssistantModelFromOpenCodeDb };
+export { resolveOpenCodeDbPath, hasV1MessageTables, assertOpenCodeStoreGeneration, openCodeDbPathExists, getOpenCodeDbProbeDescriptions, formatOpenCodeDbMissingBanner, formatOpenCodeDbMissingStatusLine, formatOpenCodeDbReadFailureStatusLine, recordOpenCodeDbReadFailure, clearOpenCodeDbReadFailure, getOpenCodeDbReadFailure, claimOpenCodeDbDiagnosticOnce, registerSlowWriteReporter, detectSqliteRuntime, Database, getSqliteMemoryStats, withSqliteTransformPass, withoutSqliteTransformPass, isTransientSqliteError, SqliteAcquisitionBusyError, withPrivilegedWriter, closeQuietly, openCodeDbExists, withReadOnlySessionDb, getRawSessionMessageCountFromDb, observeOpenCodeTurnEvent, clearTrackedOpenCodeSession, assistantAwaitingTools, shouldHoldIgnoredNotification, latestPersistedMessageForRecovery, getMessageTimesFromOpenCodeDb, findLastAssistantModelFromOpenCodeDb };

package/dist/{index-2zf5fv89.js → index-esxfcz5z.js}

@@ -1,12 +1,12 @@
 import {
   getHarness
-} from "./index-6z9g64js.js";
+} from "./index-s70wtd8k.js";
 import {
   openCodeDbExists,
   withReadOnlySessionDb,
   getRawSessionMessageCountFromDb,
   getMessageTimesFromOpenCodeDb
-} from "./index-pye48m79.js";
+} from "./index-egkfvhms.js";
 import {
   UNKNOWN_FIT_RATIO,
   resolveDecisionCalibration,
@@ -3051,11 +3051,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 (
@@ -3071,14 +3083,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) ?? [];
@@ -3097,6 +3103,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
@@ -3935,6 +3984,72 @@ function buildFileSourceContent(parts) {
   return content.length > 0 ? content : null;
 }
 
+// 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 refuseBudgetedToolCall(sessionId) {
+  return finalizingChildren.get(sessionId)?.refuseTool() ?? null;
+}
+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 };
+    }
+  };
+}
+
 // src/hooks/magic-context/dropped-input-guard.ts
 var LEGACY_TRUNCATED_VALUE = /^[\s\S]{0,5}\.\.\.\[truncated\]$/;
 var MAX_QUOTED_ARGUMENTS = 160;
@@ -4101,6 +4216,11 @@ function createDroppedInputToolExecuteBeforeHook() {
   return async (input, output) => {
     const call = isRecord2(input) ? input : {};
     const args = isRecord2(output) ? output.args : undefined;
+    if (typeof call.sessionID === "string") {
+      const refusal = refuseBudgetedToolCall(call.sessionID);
+      if (refusal)
+        throw new Error(refusal.message);
+    }
     assertExecutableToolInput(guard, {
       sessionID: typeof call.sessionID === "string" ? call.sessionID : undefined,
       toolName: typeof call.tool === "string" ? call.tool : undefined,
@@ -5001,6 +5121,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);
@@ -5599,4 +5758,4 @@ function readRawSessionSeedTail2(sessionId, boundaryId, onQuery) {
   });
 }
 
-export { textMentionsRecentCommit, removeSystemReminders, hasMeaningfulUserText2, extractTexts2, extractToolCallSummaries, getTokenizerNativeMemoryStats, preloadTokenizer, estimateTokens, normalizeText, stableStringify, getCurrentToolSetHash, setDatabase, loadToolDefinitionMeasurements, recordToolDefinition, getMeasuredToolDefinitionTokens, CTX_REDUCE_KEEP, newestCtxReduceTagNumbers, stripWellFormedLeadingTagPrefix, stripPersistedAssistantText, byteSize, stripTagPrefix, peelLeadingMcTagNotation, prependTag, isThinkingPart, isRecord, estimateImageTokensFromDataUrl, estimateToolAttachmentImageTokens, normalizeTodoStateJson, buildSyntheticTodoPart, computeSyntheticCallId, isSyntheticTodoPart, countRealUserMessages, tailHygieneStructuralSignature, sameTailHygieneStructuralSignature, formatTailHygienePrefixMismatch, refreshTailHygieneBaseline, effectiveTailHygiene, assertTailHygieneContentUnchanged, CHANNEL1_SENTINEL, CHANNEL1_FLOOR_TOKENS, isDroppedToolOutput, toolOutputTokens, decideChannel1, evaluateChannel2, formatChannel1Evaluation, formatChannel2Evaluation, reclaimableToolOutputCount, buildChannel2Reminder, buildChannel1Reminder, estimateEmergencyDropReclaimTokens, planEmergencyDrop, measureEmergencyTag, getRecentTagOwnerMessageIds, AGE_RECLAIM_MIN_TOKENS, getOldestActiveUnprotectedToolTags, getActiveToolTagsForAgeReclaim, getTriggerTagTokenUpperBound, getActiveTagTokenTotalsByMessage, getAllStatusTagTokenTotalsFlat, tagTokenCountIsNull, backfillTagTokenCounts, insertTag, updateTagStatus, markWhitespaceAssistantTagInert, getInertWhitespaceAssistantTags, updateTagDropMode, updateCavemanDepth, updateTagMessageId, foldShrunkPartTags, markTagsCompactedByMessageIds, deleteTagsByMessageId, getMaxTagNumberBySession, getTagNumberByMessageId, getAssignableTagNumberByMessageId, deriveTagLoadFloor, TAG_SELECT_COLUMNS, getTagsBySession, getActiveTagsBySession, getTailHygieneTags, getDroppedTagsBySession, getTagsByNumbers, getDroppedTagsByNumbers, getMaxDroppedTagNumber, getTagById, getToolTagNumberByOwner, getNullOwnerToolTag, adoptNullOwnerToolTag, getCandidateToolOwners, pickNearestPriorOwner, markHostUnservedRow, retreatPastHostUnservedRows, snapTerminalCompartmentToServedRow, isHostRenderedSystemMessage, isStrictGapHealingMessage, isRawCompactionSummaryInfo, extractInMemoryMessageViews, buildInMemoryTailRawMessages, completedToolArcCrossesBoundary, estimateTrueRawMessageTokens, buildToolArcs, fenceBoundaryForCompletedToolArcs, fenceBoundaryForToolArcs, buildTrueRawTokenIndex, computeRawRangeFingerprint, invalidateTrueRawTokenCache, isTextPart, isToolPartWithOutput, isFilePart, buildFileSourceContent, recordToolParameters, createDroppedInputGuard, assertExecutableToolInput, createDroppedInputToolExecuteBeforeHook, isEditTool, describeFinalWireTail, estimateMessageTokens, estimateFinalWireInputTokens, SKELETON_REAL_INPUT_MAX_BYTES, partHasCompletedResult, extractToolCallObservation, ToolMutationBatch, createToolDropTarget, resolveHostServedBoundaryId2, hasRawMessageProvider2, setRawMessageProvider2, setBoundedRawMessageProvider2, cleanUserText2, withRawSessionMessageCache2, readRawSessionMessages2, readRawSessionMessagePage2, getRawSessionMessageOrdinalCount2, readRawSessionMessageRange2, primeTailRawMessageCache2, getCachedAbsoluteMessageCount2, primeInMemoryTailRawMessageCache2, readRawSessionMessageOrdinalPage2, getRawSessionStoredMessageCount2, readRawSessionMessageIdOrdinalsForRange2, readRawSessionMessagePartsById2, hasRawSessionMessageById2, readRawSessionMessageOrdinalById2, compareRawSessionMessageOrder2, readRawSessionMessageById2, getRawSessionMessageCount2, getRawSessionTagKeysThrough2, getLegacyProtectedTailStartOrdinal2, readSessionChunk2, readRawSessionSeedTail2 };
+export { textMentionsRecentCommit, removeSystemReminders, hasMeaningfulUserText2, extractTexts2, extractToolCallSummaries, getTokenizerNativeMemoryStats, preloadTokenizer, estimateTokens, normalizeText, stableStringify, getCurrentToolSetHash, setDatabase, loadToolDefinitionMeasurements, recordToolDefinition, getMeasuredToolDefinitionTokens, CTX_REDUCE_KEEP, newestCtxReduceTagNumbers, stripWellFormedLeadingTagPrefix, stripPersistedAssistantText, byteSize, stripTagPrefix, peelLeadingMcTagNotation, prependTag, isThinkingPart, isRecord, estimateImageTokensFromDataUrl, estimateToolAttachmentImageTokens, normalizeTodoStateJson, buildSyntheticTodoPart, computeSyntheticCallId, isSyntheticTodoPart, countRealUserMessages, tailHygieneStructuralSignature, sameTailHygieneStructuralSignature, formatTailHygienePrefixMismatch, refreshTailHygieneBaseline, effectiveTailHygiene, assertTailHygieneContentUnchanged, CHANNEL1_SENTINEL, CHANNEL1_FLOOR_TOKENS, isDroppedToolOutput, toolOutputTokens, decideChannel1, evaluateChannel2, formatChannel1Evaluation, formatChannel2Evaluation, reclaimableToolOutputCount, buildChannel2Reminder, buildChannel1Reminder, estimateEmergencyDropReclaimTokens, planEmergencyDrop, measureEmergencyTag, getRecentTagOwnerMessageIds, AGE_RECLAIM_MIN_TOKENS, getOldestActiveUnprotectedToolTags, getActiveToolTagsForAgeReclaim, getTriggerTagTokenUpperBound, getActiveTagTokenTotalsByMessage, getAllStatusTagTokenTotalsFlat, tagTokenCountIsNull, backfillTagTokenCounts, insertTag, updateTagStatus, markWhitespaceAssistantTagInert, getInertWhitespaceAssistantTags, updateTagDropMode, updateCavemanDepth, updateTagMessageId, foldShrunkPartTags, markTagsCompactedByMessageIds, deleteTagsByMessageId, getMaxTagNumberBySession, getTagNumberByMessageId, getAssignableTagNumberByMessageId, deriveTagLoadFloor, TAG_SELECT_COLUMNS, getTagsBySession, getActiveTagsBySession, getTailHygieneTags, getDroppedTagsBySession, getTagsByNumbers, getDroppedTagsByNumbers, getMaxDroppedTagNumber, getTagById, getToolTagNumberByOwner, getNullOwnerToolTag, adoptNullOwnerToolTag, getCandidateToolOwners, pickNearestPriorOwner, markHostUnservedRow, retreatPastHostUnservedRows, snapTerminalCompartmentToServedRow, isHostRenderedSystemMessage, isStrictGapHealingMessage, isRawCompactionSummaryInfo, extractInMemoryMessageViews, buildInMemoryTailRawMessages, completedToolArcCrossesBoundary, estimateTrueRawMessageTokens, buildToolArcs, fenceBoundaryForCompletedToolArcs, fenceBoundaryForToolArcs, buildTrueRawTokenIndex, computeRawRangeFingerprint, invalidateTrueRawTokenCache, isTextPart, isToolPartWithOutput, isFilePart, buildFileSourceContent, DreamTokenBudgetExceeded, TOKEN_BUDGET_FINALIZE_MESSAGE, registerBudgetFinalizeChild, releaseBudgetFinalizeChild, createDreamTokenBudget, recordToolParameters, createDroppedInputGuard, assertExecutableToolInput, createDroppedInputToolExecuteBeforeHook, isEditTool, describeFinalWireTail, estimateMessageTokens, estimateFinalWireInputTokens, SKELETON_REAL_INPUT_MAX_BYTES, partHasCompletedResult, extractToolCallObservation, ToolMutationBatch, createToolDropTarget, resolveHostServedBoundaryId2, hasRawMessageProvider2, setRawMessageProvider2, setBoundedRawMessageProvider2, cleanUserText2, withRawSessionMessageCache2, readRawSessionMessages2, readRawSessionMessagePage2, getRawSessionMessageOrdinalCount2, visitRawSessionMessages2, readRawSessionMessageRange2, primeTailRawMessageCache2, getCachedAbsoluteMessageCount2, primeInMemoryTailRawMessageCache2, readRawSessionMessageOrdinalPage2, getRawSessionStoredMessageCount2, readRawSessionMessageIdOrdinalsForRange2, readRawSessionMessagePartsById2, hasRawSessionMessageById2, readRawSessionMessageOrdinalById2, compareRawSessionMessageOrder2, readRawSessionMessageById2, getRawSessionMessageCount2, getRawSessionTagKeysThrough2, getLegacyProtectedTailStartOrdinal2, readSessionChunk2, readRawSessionSeedTail2 };