billion-context 0.1.86 → 0.1.87

package/dist/index.js

@@ -43476,10 +43476,6 @@ function defaultCountTokens(text) {
   const cjkCount = cjk?.length ?? 0;
   return cjkCount + Math.ceil((text.length - cjkCount) / 4);
 }
-function estimateTokensFast(text) {
-  if (!text) return 0;
-  return Math.ceil(text.length / 4);
-}
 var COMPRESS_PHILOSOPHY = `Compression Philosophy:
 - All compression serves the primary task, but be frugal.
 - Context capacity is precious. Save context by compressing consumed outputs, not by avoiding tools.
@@ -46339,7 +46335,7 @@ function formatTokens2(n) {
 }
 function pct(n, total) {
   if (n <= 0 || total <= 0) return 0;
-  return Math.max(1, Math.round(n / total * 100));
+  return Math.round(n / total * 100);
 }
 function numericPart2(blockId) {
   const match = /^b(\d+)$/.exec(blockId);
@@ -46378,12 +46374,18 @@ function collectVisible(messages, state, countTokens) {
     if (block.active) summaryTokens += summaryTokensOf(block, countTokens);
   }
   const visible = [];
+  const toolCallNames = /* @__PURE__ */ new Map();
+  for (const message of messages) {
+    if (message.contentType === "tool-call" && message.toolCallId && message.toolName) {
+      toolCallNames.set(message.toolCallId, message.toolName);
+    }
+  }
   messages.forEach((message, index) => {
     if (coveredIds.has(message.id)) return;
     const ref = refForRaw(state.messageRefs, message.id);
     if (!ref) return;
     const tokens = countTokens(message.text ?? "");
-    const tool = message.toolName ?? "text";
+    const tool = isToolMessage(message) ? message.toolName ?? (message.toolCallId ? toolCallNames.get(message.toolCallId) : void 0) ?? "tool" : "text";
     if (tokens > 0) visible.push({ ref, tokens, tool, index });
   });
   return { visible, summaryTokens };
@@ -46471,6 +46473,17 @@ function renderUncompressedRanges(visible) {
     const m2 = ref.match(/\d+/);
     return m2 ? parseInt(m2[0], 10) : 0;
   };
+  const dominantTool = (toolTokens) => {
+    let best = "text";
+    let bestN = -1;
+    for (const [tool, n] of toolTokens) {
+      if (n > bestN) {
+        best = tool;
+        bestN = n;
+      }
+    }
+    return best;
+  };
   const merged = [];
   for (const m2 of visible) {
     const num3 = refNum22(m2.ref);
@@ -46479,13 +46492,15 @@ function renderUncompressedRanges(visible) {
       last.endRef = m2.ref;
       last.count += 1;
       last.tokens += m2.tokens;
+      last.toolTokens.set(m2.tool, (last.toolTokens.get(m2.tool) ?? 0) + m2.tokens);
     } else {
-      merged.push({ startRef: m2.ref, endRef: m2.ref, startNum: num3, count: 1, tokens: m2.tokens, tool: m2.tool });
+      const toolTokens = /* @__PURE__ */ new Map([[m2.tool, m2.tokens]]);
+      merged.push({ startRef: m2.ref, endRef: m2.ref, startNum: num3, count: 1, tokens: m2.tokens, toolTokens });
     }
   }
   for (const r of merged.slice(0, 30)) {
     const range = r.count === 1 ? r.startRef : `${r.startRef}\u2013${r.endRef}`;
-    lines.push(`  ${range}  (${r.count} msgs, ${formatTokens2(r.tokens)}${r.count > 1 ? ` (${Math.round(r.tokens / r.count)}/msg)` : ""}) ${r.tool}`);
+    lines.push(`  ${range}  (${r.count} msgs, ${formatTokens2(r.tokens)}${r.count > 1 ? ` (${Math.round(r.tokens / r.count)}/msg)` : ""}) ${dominantTool(r.toolTokens)}`);
   }
   if (merged.length > 30) {
     lines.push(`  ... and ${merged.length - 30} more ranges`);
@@ -50754,7 +50769,7 @@ function executeAnthropicProxyTool(toolName, args, ctx) {
 ${lines.join("\n\n")}`;
   }
   if (toolName === "acp_status") {
-    return buildStatusReport(ctx.session.state, ctx.messages, estimateTokensFast);
+    return buildStatusReport(ctx.session.state, ctx.messages, defaultCountTokens);
   }
   return `[Unknown proxy tool: ${toolName}]`;
 }
@@ -51872,6 +51887,23 @@ function isLoopThinking(m2) {
 function stripLoopThinking(messages) {
   return messages.filter((m2) => !isLoopThinking(m2));
 }
+function canonicalArgs(args) {
+  try {
+    return JSON.stringify(sortKeys(JSON.parse(args)));
+  } catch {
+    return args;
+  }
+}
+function sortKeys(v2) {
+  if (Array.isArray(v2)) return v2.map(sortKeys);
+  if (v2 !== null && typeof v2 === "object") {
+    const src = v2;
+    const out = {};
+    for (const k2 of Object.keys(src).sort()) out[k2] = sortKeys(src[k2]);
+    return out;
+  }
+  return v2;
+}
 function executeProxyTool(toolName, args, ctx, callId) {
   if (toolName === "compress") {
     return applyRanges(parseCompressInput(args, callId), ctx);
@@ -51906,7 +51938,7 @@ function handleAcpStatus(args, ctx) {
   const tool = typeof args.tool === "string" ? args.tool : void 0;
   const sort = typeof args.sort === "string" ? args.sort : void 0;
   const limit = typeof args.limit === "number" ? args.limit : void 0;
-  const base = buildStatusReport(ctx.session.state, ctx.messages, estimateTokensFast, { scope, view, tool, sort, limit });
+  const base = buildStatusReport(ctx.session.state, ctx.messages, defaultCountTokens, { scope, view, tool, sort, limit });
   if (scope) return base;
   const nudge = ctx.nudge;
   const ranges = nudge?.compressibleRanges ?? [];
@@ -51974,6 +52006,7 @@ async function* runCompressLoop(upstream, ctx, requestBody, requestOptions, adap
       log("warn", `[acp-loop] upstream rejected replay (HTTP ${info.status}); retrying in ${info.delayMs}ms (attempt ${info.attempt}/${info.maxAttempts})${lc}`);
     }
   );
+  const failedSignatures = /* @__PURE__ */ new Set();
   try {
     for (let round = 1; round <= MAX_LOOP_ROUNDS; round++) {
       if (signal?.aborted) break;
@@ -52229,6 +52262,19 @@ async function* runCompressLoop(upstream, ctx, requestBody, requestOptions, adap
         yield adapter.emitCompletion({ finishReason: "length", usage });
         return;
       }
+      const failedMutating = proxyResults.filter(
+        (pr2) => (pr2.name === "compress" || pr2.name === "decompress") && pr2.result.includes("FAILED")
+      );
+      if (reRequest && failedMutating.length > 0 && failedMutating.length === proxyResults.length) {
+        const repeated = failedMutating.some((pr2) => failedSignatures.has(`${pr2.name}\0${canonicalArgs(pr2.arguments)}`));
+        if (repeated) {
+          ctx.log(`[acp-loop] round ${round}: model re-submitted an already-failed compress with identical arguments; stopping after ${round} round(s) instead of ${MAX_LOOP_ROUNDS}`);
+          log("warn", `[acp-loop] repeated identical compress failure at round ${round}; breaking early (#156)`);
+          yield adapter.emitCompletion({ finishReason: "length", usage });
+          return;
+        }
+      }
+      for (const pr2 of failedMutating) failedSignatures.add(`${pr2.name}\0${canonicalArgs(pr2.arguments)}`);
       ctx.log(`[acp-loop] round ${round}: proxy tool executed; re-requesting so the model sees the result`);
       if (signal?.aborted) break;
       let newBody = adapter.buildRequest(coreMessages, systemPrompt, requestBody);
@@ -53755,7 +53801,7 @@ function executeProxyTool2(toolName, args, ctx) {
 ${lines.join("\n\n")}`;
   }
   if (toolName === "acp_status") {
-    return buildStatusReport(ctx.session.state, ctx.messages, estimateTokensFast);
+    return buildStatusReport(ctx.session.state, ctx.messages, defaultCountTokens);
   }
   return `[Unknown proxy tool: ${toolName}]`;
 }
@@ -54334,14 +54380,14 @@ var MIN_TAIL_MATCH = 3;
 var MAX_STORED_ITEMS = 128;
 var MIN_FORK_PREFIX = 3;
 function stableStringify(value) {
-  return JSON.stringify(sortKeys(value));
+  return JSON.stringify(sortKeys2(value));
 }
-function sortKeys(value) {
-  if (Array.isArray(value)) return value.map(sortKeys);
+function sortKeys2(value) {
+  if (Array.isArray(value)) return value.map(sortKeys2);
   if (value && typeof value === "object") {
     const record = value;
     const out = {};
-    for (const key of Object.keys(record).sort()) out[key] = sortKeys(record[key]);
+    for (const key of Object.keys(record).sort()) out[key] = sortKeys2(record[key]);
     return out;
   }
   return value;