opencode-acp 1.16.0-pr.349.122 → 1.16.0-pr.350.123

package/dist/index.js

@@ -914,6 +914,7 @@ var VALID_CONFIG_KEYS = /* @__PURE__ */ new Set([
   "compress.reasoning",
   "compress.reasoning.drop",
   "compress.reasoning.threshold",
+  "compress.completionReserveTokens",
   "gc",
   "gc.algorithm",
   "gc.promotionThreshold",
@@ -965,7 +966,11 @@ function validateConfigTypes(config) {
     errors.push({ key: "storagePath", expected: "string", actual: typeof config.storagePath });
   }
   if (config.allowSubAgents !== void 0 && typeof config.allowSubAgents !== "boolean") {
-    errors.push({ key: "allowSubAgents", expected: "boolean", actual: typeof config.allowSubAgents });
+    errors.push({
+      key: "allowSubAgents",
+      expected: "boolean",
+      actual: typeof config.allowSubAgents
+    });
   }
   if (config.pruneNotification !== void 0) {
     const validValues = ["off", "minimal", "detailed"];
@@ -1294,6 +1299,20 @@ function validateConfigTypes(config) {
           }
         }
       }
+      if (compress.completionReserveTokens !== void 0 && typeof compress.completionReserveTokens !== "number") {
+        errors.push({
+          key: "compress.completionReserveTokens",
+          expected: "number",
+          actual: typeof compress.completionReserveTokens
+        });
+      }
+      if (typeof compress.completionReserveTokens === "number" && compress.completionReserveTokens < 0) {
+        errors.push({
+          key: "compress.completionReserveTokens",
+          expected: "non-negative number (>= 0)",
+          actual: `${compress.completionReserveTokens}`
+        });
+      }
       if (typeof compress.iterationNudgeThreshold === "number" && compress.iterationNudgeThreshold < 1) {
         errors.push({
           key: "compress.iterationNudgeThreshold",
@@ -1412,12 +1431,20 @@ function validateConfigTypes(config) {
             break;
           case "nudgeForce":
             if (value !== "strong" && value !== "soft") {
-              errors.push({ key, expected: "'strong' | 'soft'", actual: JSON.stringify(value) });
+              errors.push({
+                key,
+                expected: "'strong' | 'soft'",
+                actual: JSON.stringify(value)
+              });
             }
             break;
           case "stringArray":
             if (!Array.isArray(value) || !value.every((entry) => typeof entry === "string")) {
-              errors.push({ key, expected: "string[]", actual: JSON.stringify(value) });
+              errors.push({
+                key,
+                expected: "string[]",
+                actual: JSON.stringify(value)
+              });
             }
             break;
           case "reasoningConfig":
@@ -1452,7 +1479,11 @@ function validateConfigTypes(config) {
           return;
         }
         if (typeof overrides !== "object" || overrides === null || Array.isArray(overrides)) {
-          errors.push({ key: prefix, expected: "CompressModelOverrides", actual: typeof overrides });
+          errors.push({
+            key: prefix,
+            expected: "CompressModelOverrides",
+            actual: typeof overrides
+          });
           return;
         }
         const model = overrides;
@@ -1485,7 +1516,11 @@ function validateConfigTypes(config) {
         for (const [providerId, providerValue] of Object.entries(providers)) {
           const prefix = `compress.providers.${providerId}`;
           if (typeof providerValue !== "object" || providerValue === null || Array.isArray(providerValue)) {
-            errors.push({ key: prefix, expected: "ProviderOverrides", actual: typeof providerValue });
+            errors.push({
+              key: prefix,
+              expected: "ProviderOverrides",
+              actual: typeof providerValue
+            });
             continue;
           }
           const provider = providerValue;
@@ -1621,13 +1656,22 @@ function validateConfigTypes(config) {
           });
         } else {
           if (gc.batchCleanup.lowThreshold !== void 0) {
-            validateBatchThreshold("gc.batchCleanup.lowThreshold", gc.batchCleanup.lowThreshold);
+            validateBatchThreshold(
+              "gc.batchCleanup.lowThreshold",
+              gc.batchCleanup.lowThreshold
+            );
           }
           if (gc.batchCleanup.highThreshold !== void 0) {
-            validateBatchThreshold("gc.batchCleanup.highThreshold", gc.batchCleanup.highThreshold);
+            validateBatchThreshold(
+              "gc.batchCleanup.highThreshold",
+              gc.batchCleanup.highThreshold
+            );
           }
           if (gc.batchCleanup.forceThreshold !== void 0) {
-            validateBatchThreshold("gc.batchCleanup.forceThreshold", gc.batchCleanup.forceThreshold);
+            validateBatchThreshold(
+              "gc.batchCleanup.forceThreshold",
+              gc.batchCleanup.forceThreshold
+            );
           }
         }
       }
@@ -1912,7 +1956,8 @@ function mergeCompress(base, override) {
     reasoning: {
       drop: override.reasoning?.drop ?? base.reasoning?.drop ?? DEFAULT_COMPRESS_REASONING.drop,
       threshold: override.reasoning?.threshold ?? base.reasoning?.threshold ?? DEFAULT_COMPRESS_REASONING.threshold
-    }
+    },
+    completionReserveTokens: override.completionReserveTokens ?? base.completionReserveTokens
   };
 }
 function mergeCommands(base, override) {
@@ -1950,10 +1995,9 @@ function deepCloneConfig(config) {
             ...provider,
             ...provider.models ? {
               models: Object.fromEntries(
-                Object.entries(provider.models).map(([modelId, model]) => [
-                  modelId,
-                  { ...model }
-                ])
+                Object.entries(provider.models).map(
+                  ([modelId, model]) => [modelId, { ...model }]
+                )
               )
             } : {}
           }
@@ -2027,8 +2071,14 @@ function mergeLayer(config, data) {
     ],
     compress: mergeCompress(config.compress, data.compress),
     gc: mergeGC(config.gc, data.gc),
-    qualityGate: mergeQualityGate(config.qualityGate, data.qualityGate),
-    messageFilters: mergeMessageFilters(config.messageFilters, data.messageFilters)
+    qualityGate: mergeQualityGate(
+      config.qualityGate,
+      data.qualityGate
+    ),
+    messageFilters: mergeMessageFilters(
+      config.messageFilters,
+      data.messageFilters
+    )
   };
 }
 function scheduleParseWarning(ctx, title, message) {
@@ -4644,7 +4694,8 @@ function createSessionState() {
     modelID: void 0,
     systemPromptTokens: void 0,
     storageDir: void 0,
-    qualityGateRetryPending: false
+    qualityGateRetryPending: false,
+    noContextLimitWarned: false
   };
 }
 function resetSessionState(state) {
@@ -4687,6 +4738,7 @@ function resetSessionState(state) {
   state.systemPromptTokens = void 0;
   state.storageDir = void 0;
   state.qualityGateRetryPending = false;
+  state.noContextLimitWarned = false;
 }
 async function ensureSessionInitialized(client, state, sessionId, logger, messages, config, projectDir) {
   if (state.sessionId === sessionId) {
@@ -9579,7 +9631,7 @@ import { writeFile as writeFile2, mkdir as mkdir2 } from "fs/promises";
 import { join as join4 } from "path";
 import { existsSync as existsSync4 } from "fs";
 import { homedir as homedir3 } from "os";
-var LOG_VERSION = true ? "1.16.0-pr.349.122" : "dev";
+var LOG_VERSION = true ? "1.16.0-pr.350.123" : "dev";
 var LEVEL_RANK = {
   debug: 10,
   info: 20,
@@ -10484,6 +10536,153 @@ function truncateLargeToolOutputs(state, config, logger, messages) {
   }
 }
 
+// lib/messages/enforce-budget.ts
+var DEFAULT_COMPLETION_RESERVE_TOKENS = 32768;
+var TRUNCATION_MARKER2 = "[truncated for context space";
+var KEEP_PREFIX_CHARS2 = 2e3;
+var KEEP_SUFFIX_CHARS2 = 2e3;
+var PROTECT_RECENT_MESSAGES2 = 3;
+var MIN_CLEAR_TOKENS = 200;
+function resolveContextWindow(state) {
+  if (typeof state.modelContextLimit === "number" && state.modelContextLimit > 0) {
+    return state.modelContextLimit;
+  }
+  return void 0;
+}
+function estimateWireTokens(state, messages) {
+  const base = getCurrentTokenUsage(state, messages);
+  if (base > 0) {
+    let baseAssistant = -1;
+    for (let i = messages.length - 1; i >= 0; i--) {
+      if (messages[i].info.role !== "assistant") continue;
+      const tokens = messages[i].info.tokens;
+      if ((tokens?.input || 0) <= 0 && (tokens?.output || 0) <= 0) continue;
+      baseAssistant = i;
+      break;
+    }
+    if (baseAssistant >= 0) {
+      let additions = 0;
+      for (let i = baseAssistant + 1; i < messages.length; i++) {
+        additions += countAllMessageTokens(messages[i]);
+      }
+      return base + additions;
+    }
+  }
+  let total = 0;
+  for (const m of messages) total += countAllMessageTokens(m);
+  return total + (state.systemPromptTokens ?? 0);
+}
+function enforceContextBudget(state, config, logger, messages) {
+  const window = resolveContextWindow(state);
+  if (window === void 0) return void 0;
+  const configuredReserve = config.compress?.completionReserveTokens;
+  const reserve = typeof configuredReserve === "number" && Number.isFinite(configuredReserve) && configuredReserve >= 0 ? configuredReserve : DEFAULT_COMPLETION_RESERVE_TOKENS;
+  const budget = window - reserve;
+  if (budget <= 0) return void 0;
+  const estimatedTokens = estimateWireTokens(state, messages);
+  if (estimatedTokens <= budget) {
+    return {
+      applied: false,
+      window,
+      reserve,
+      budget,
+      estimatedTokens,
+      finalEstimate: estimatedTokens,
+      truncatedCount: 0,
+      clearedCount: 0
+    };
+  }
+  const protectedIndex = messages.length - PROTECT_RECENT_MESSAGES2;
+  const protectedTools = new Set(config.compress?.protectedTools ?? []);
+  const candidates = [];
+  for (let mi = 0; mi < protectedIndex; mi++) {
+    if (mi === 0 && messages[mi].info.role === "user") continue;
+    const msg = messages[mi];
+    const parts = Array.isArray(msg.parts) ? msg.parts : [];
+    for (const part of parts) {
+      if (part?.type !== "tool") continue;
+      if (part.state?.status !== "completed") continue;
+      if (part.tool === "compress") continue;
+      if (protectedTools.has(part.tool)) continue;
+      const content = extractCompletedToolOutput(part);
+      if (content === void 0) continue;
+      if (content === COMPACTED_TOOL_OUTPUT_PLACEHOLDER) continue;
+      const tokens = countTokens2(content);
+      if (tokens <= 0) continue;
+      candidates.push({ part, content, tokens, index: mi });
+    }
+  }
+  let saved = 0;
+  let truncatedCount = 0;
+  let clearedCount = 0;
+  const truncatable = candidates.filter(
+    (c) => !c.content.includes(TRUNCATION_MARKER2) && c.content.length > KEEP_PREFIX_CHARS2 + KEEP_SUFFIX_CHARS2
+  ).sort((a, b) => b.tokens - a.tokens);
+  for (const c of truncatable) {
+    if (estimatedTokens - saved <= budget) break;
+    const prefix = c.content.slice(0, KEEP_PREFIX_CHARS2);
+    const suffix = c.content.slice(-KEEP_SUFFIX_CHARS2);
+    const truncated = prefix + `
+
+...${TRUNCATION_MARKER2} \u2014 original ~${c.tokens} tokens]...
+
+` + suffix;
+    if (truncated.length >= c.content.length) continue;
+    c.part.state.output = truncated;
+    saved += c.tokens - countTokens2(truncated);
+    truncatedCount++;
+  }
+  if (estimatedTokens - saved > budget) {
+    const clearable = candidates.filter((c) => {
+      const out = extractCompletedToolOutput(c.part);
+      return out !== void 0 && out !== COMPACTED_TOOL_OUTPUT_PLACEHOLDER && countTokens2(out) > MIN_CLEAR_TOKENS;
+    }).sort((a, b) => a.index - b.index);
+    for (const c of clearable) {
+      if (estimatedTokens - saved <= budget) break;
+      const current = extractCompletedToolOutput(c.part);
+      if (current === void 0 || current === COMPACTED_TOOL_OUTPUT_PLACEHOLDER) continue;
+      c.part.state.output = COMPACTED_TOOL_OUTPUT_PLACEHOLDER;
+      saved += countTokens2(current) - countTokens2(COMPACTED_TOOL_OUTPUT_PLACEHOLDER);
+      clearedCount++;
+    }
+  }
+  const finalEstimate = Math.max(0, estimatedTokens - saved);
+  if (truncatedCount > 0 || clearedCount > 0) {
+    logger.warn("Context budget guard: pruned tool outputs to fit the request", {
+      session: state.sessionId,
+      estimatedTokens: Math.round(estimatedTokens),
+      budget,
+      window,
+      reserve,
+      truncatedCount,
+      clearedCount,
+      estimatedSavedTokens: Math.round(saved),
+      finalEstimate: Math.round(finalEstimate)
+    });
+  }
+  if (finalEstimate > budget) {
+    logger.warn(
+      "Context budget guard: still over budget after pruning all compressible tool outputs; the request may be rejected by the model",
+      {
+        session: state.sessionId,
+        finalEstimate: Math.round(finalEstimate),
+        budget,
+        window
+      }
+    );
+  }
+  return {
+    applied: truncatedCount > 0 || clearedCount > 0,
+    window,
+    reserve,
+    budget,
+    estimatedTokens,
+    finalEstimate,
+    truncatedCount,
+    clearedCount
+  };
+}
+
 // lib/commands/context.ts
 function analyzeTokens(state, messages) {
   const breakdown = {
@@ -11606,6 +11805,16 @@ function createChatMessageTransformHandler(client, registry4, logger, config, pr
         });
       }
       await updatePerTurnState(state, logger, messages);
+      if (state.modelContextLimit === void 0 && !state.noContextLimitWarned && requestModel?.providerID && requestModel?.modelID && registry4.resolveModelLimit(requestModel.providerID, requestModel.modelID) === void 0) {
+        state.noContextLimitWarned = true;
+        logger.warn(
+          'Model reports no context window and the catalog has no entry for it; all percentage thresholds (min/max/emergency, GC) and the context-budget guard are disabled. Set the model limit in opencode.json (e.g. "limit": {"context": 262144, "output": 16384}) to enable them (also fixes the 32000 max_tokens fallback); an absolute compress.maxContextLimit in acp.jsonc only enables proactive nudges, not the guard.',
+          {
+            session: state.sessionId,
+            model: `${requestModel.providerID}/${requestModel.modelID}`
+          }
+        );
+      }
     }
     syncCompressPermissionState(state, config, hostPermissions, output.messages);
     if (state.isSubAgent && !config.allowSubAgents) {
@@ -11655,6 +11864,7 @@ function createChatMessageTransformHandler(client, registry4, logger, config, pr
     const prePruneTokens = getCurrentTokenUsage(state, output.messages);
     prune(state, logger, config, output.messages);
     truncateLargeToolOutputs(state, config, logger, output.messages);
+    enforceContextBudget(state, config, logger, output.messages);
     hideConsumedCompressCalls(state, output.messages);
     assignMessageRefs(state, output.messages);
     const compressionPriorities = buildPriorityMap(config, state, output.messages);
@@ -11742,12 +11952,7 @@ function createCommandExecuteHandler(client, registry4, logger, config, workingD
         path: { id: input.sessionID }
       });
       const messages = filterMessages(messagesResponse.data || messagesResponse);
-      const state = await registry4.getOrCreate(
-        client,
-        input.sessionID,
-        messages,
-        config
-      );
+      const state = await registry4.getOrCreate(client, input.sessionID, messages, config);
       syncCompressPermissionState(state, config, hostPermissions, messages);
       const commandCtx = {
         client,
@@ -11769,13 +11974,7 @@ function createCommandExecuteHandler(client, registry4, logger, config, workingD
         throw new Error("__DCP_CONTEXT_HANDLED__");
       }
       if (sub === "help") {
-        await sendIgnoredMessage(
-          client,
-          input.sessionID,
-          buildHelpText(),
-          {},
-          logger
-        );
+        await sendIgnoredMessage(client, input.sessionID, buildHelpText(), {}, logger);
         throw new Error("__DCP_CONTEXT_HANDLED__");
       }
       await handleContextCommand(commandCtx);
@@ -11850,9 +12049,7 @@ function createEventHandler(registry4, logger) {
       return;
     }
     if (typeof part.callID === "string" && typeof part.messageID === "string") {
-      timing.startsByCallId.delete(
-        buildCompressionTimingKey(part.messageID, part.callID)
-      );
+      timing.startsByCallId.delete(buildCompressionTimingKey(part.messageID, part.callID));
     }
   };
 }
@@ -12129,7 +12326,7 @@ var server = (async (ctx) => {
   }
   const logger = new Logger(config.debug, config.debug ? "debug" : config.logLevel);
   logger.info("ACP plugin initialized", {
-    version: true ? "1.16.0-pr.349.122" : "dev",
+    version: true ? "1.16.0-pr.350.123" : "dev",
     workspace: ctx.directory,
     logLevel: logger.level,
     debug: config.debug,