@mrrlin-dev/mcp 0.4.141 → 0.4.142

package/dist/bin.cjs

@@ -43772,6 +43772,28 @@ function addTurnTokenUsage(total, turn) {
     outputTokens: add2(total.outputTokens, turn.outputTokens)
   };
 }
+var CONTEXT_WINDOW_TOKENS = 2e5;
+function extractContextWindow(method, params) {
+  if (method !== TOKEN_USAGE_NOTIFICATION_METHOD)
+    return null;
+  if (params === null || typeof params !== "object")
+    return null;
+  const usage = params.tokenUsage;
+  if (usage === null || typeof usage !== "object")
+    return null;
+  const window = readCount(usage, "modelContextWindow");
+  return window !== null && window > 0 ? window : null;
+}
+function isNearContextLimit(usage, contextWindow) {
+  if (!usage)
+    return false;
+  const contextTokens = (usage.inputTokens ?? 0) + (usage.outputTokens ?? 0);
+  if (contextTokens <= 0)
+    return false;
+  const window = contextWindow && contextWindow > 0 ? contextWindow : CONTEXT_WINDOW_TOKENS;
+  const remainingTokens = window - contextTokens;
+  return remainingTokens / window < 0.03 || remainingTokens < 4e3;
+}
 
 // ../../packages/agent-client/dist/providers/codex/event-mapper.js
 var CodexBridgeEventMapper = class {
@@ -43938,7 +43960,8 @@ var CodexBridgeEventMapper = class {
     const usage = extractTurnTokenUsage(TOKEN_USAGE_NOTIFICATION_METHOD, notification.params);
     if (!usage)
       return [];
-    return [{ type: "token_usage", usage }];
+    const contextWindow = extractContextWindow(TOKEN_USAGE_NOTIFICATION_METHOD, notification.params);
+    return [contextWindow === null ? { type: "token_usage", usage } : { type: "token_usage", usage, contextWindow }];
   }
   /** Bare `error` notifications carry model-stream failures ({ error: { message }, willRetry }).
    * Retryable ones (willRetry=true) are transport noise — codex is still driving the turn — so
@@ -63794,7 +63817,7 @@ Do NOT derail the current task - if the operator has an in-flight question, answ
 }
 
 // src/_generated/version.ts
-var PKG_VERSION = "0.4.141";
+var PKG_VERSION = "0.4.142";
 
 // src/bridge-allowed-origins.ts
 var DEFAULT_DIRECTOR_BRIDGE_ALLOWED_ORIGINS = [
@@ -67883,6 +67906,7 @@ async function driveTurn(deps, codex, projectSlug, runId, leaseId, threadId, inp
   let touch = () => {
   };
   let turnUsage = null;
+  let turnContextWindow = null;
   const promptBytes = Buffer.byteLength(inputText, "utf8");
   const turnStartedAt = Date.now();
   let lastActivityAt = turnStartedAt;
@@ -67914,6 +67938,7 @@ async function driveTurn(deps, codex, projectSlug, runId, leaseId, threadId, inp
       pending.push(persist(event));
       if (event.type === "token_usage") {
         turnUsage = event.usage;
+        turnContextWindow = event.contextWindow ?? turnContextWindow;
       } else if (event.type === "rate_limits") {
         const signal = detectExhaustedProviderLimit(event.snapshot, Date.now());
         if (signal) {
@@ -67975,7 +68000,7 @@ async function driveTurn(deps, codex, projectSlug, runId, leaseId, threadId, inp
       clearInterval(touchTimer);
       unsub();
     }
-    const telemetry = { promptBytes, usage: turnUsage };
+    const telemetry = { promptBytes, usage: turnUsage, contextWindow: turnContextWindow };
     return turnError ? { ok: false, error: turnError, interruptFailed, telemetry } : { ok: true, telemetry };
   });
 }
@@ -68238,14 +68263,6 @@ async function driveRun(deps, projectSlug, runId) {
     runTelemetry.usage = addTurnTokenUsage(runTelemetry.usage, outcome.telemetry.usage);
     return outcome;
   };
-  const isContextBudgetExhausted = (usage) => {
-    if (!usage) return false;
-    const contextTokens = (usage.inputTokens ?? 0) + (usage.outputTokens ?? 0);
-    if (contextTokens <= 0) return false;
-    const maxTokens = 2e5;
-    const remainingTokens = maxTokens - contextTokens;
-    return remainingTokens / maxTokens < 0.03 || remainingTokens < 4e3;
-  };
   const effortDecision = deriveEffort({
     impact: task.impact,
     type: task.type,
@@ -68589,7 +68606,7 @@ Continue executing task ${task.id}. Make the next concrete increment of progress
       await failRun(deps, projectSlug, task, runId, leaseId, outcome.error);
       return;
     }
-    if (isContextBudgetExhausted(outcome.telemetry.usage)) {
+    if (isNearContextLimit(outcome.telemetry.usage, outcome.telemetry.contextWindow)) {
       console.warn(
         JSON.stringify({
           event: "context_budget_gate_triggered",
@@ -78803,18 +78820,9 @@ function createBridgeMessageHandler(deps) {
               return;
             }
             if (event.type === "token_usage") {
-              const usage = event.usage;
+              const { usage, contextWindow } = event;
               turnTokenUsage = addTurnTokenUsage(turnTokenUsage, usage);
-              if (turnTokenUsage) {
-                const totalTokens = (turnTokenUsage.inputTokens ?? 0) + (turnTokenUsage.outputTokens ?? 0);
-                if (totalTokens > 0) {
-                  const maxTokens = 2e5;
-                  const remainingTokens = maxTokens - totalTokens;
-                  if (remainingTokens / maxTokens < 0.03 || remainingTokens < 4e3) {
-                    contextBudgetGateTriggered = true;
-                  }
-                }
-              }
+              if (isNearContextLimit(usage, contextWindow)) contextBudgetGateTriggered = true;
               return;
             }
             if (event.type === "turn_completed" || event.type === "rate_limits") {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mrrlin-dev/mcp",
-  "version": "0.4.141",
+  "version": "0.4.142",
   "type": "module",
   "bin": {
     "mrrlin-mcp": "dist/bin.cjs"