billion-context-pi 0.1.40 → 0.1.41

package/dist/index.js

@@ -2828,6 +2828,19 @@ function clamp(v, lo, hi) {
   return v < lo ? lo : v > hi ? hi : v;
 }
 
+// src/tag-tokens.ts
+function formatTokens3(tokens) {
+  if (tokens < 1e3) return String(tokens);
+  if (tokens < 1e4) return `${(tokens / 1e3).toFixed(1)}K`;
+  return `${Math.round(tokens / 1e3)}K`;
+}
+function stableTagTokens(text) {
+  return formatTokens3(defaultCountTokens(text));
+}
+function rewriteTagTokens(tag, body) {
+  return tag.replace(/tokens="[^"]*"/, `tokens="${stableTagTokens(body)}"`);
+}
+
 // src/messages.ts
 var REF_TAG_SOURCE = "(?:<acp\\s[^>]*>m\\d{5}</acp>|\\[m\\d{1,5}\\])";
 var REF_TAG = new RegExp(`^${REF_TAG_SOURCE}\\s?\\n?`);
@@ -3037,6 +3050,7 @@ function reconstructToolCallMessage(original, firstCore, survivingCallIds) {
     return true;
   });
   const peeled = peelRefTagBlocks(filtered);
+  const stableTag = rewriteTagTokens(tag, coreBodyOf(firstCore.text ?? "", tag));
   const lastTextIdx = [...peeled].reverse().findIndex((b) => b.type === "text");
   if (lastTextIdx >= 0) {
     const idx = peeled.length - 1 - lastTextIdx;
@@ -3044,10 +3058,16 @@ function reconstructToolCallMessage(original, firstCore, survivingCallIds) {
     const baseText = lastBlock.text ?? "";
     peeled[idx] = { ...lastBlock, text: baseText.length > 0 ? `${baseText}
 
-${tag}` : tag };
+${stableTag}` : stableTag };
     return { ...original, content: peeled };
   }
-  return { ...original, content: [{ type: "text", text: tag }, ...peeled] };
+  return { ...original, content: [{ type: "text", text: stableTag }, ...peeled] };
+}
+function coreBodyOf(coreText, tag) {
+  const tagCore = tag.replace(/\s+$/, "");
+  let bodyStart = tagCore.length;
+  if (coreText.charAt(bodyStart) === "\n") bodyStart += 1;
+  return coreText.slice(bodyStart);
 }
 function patchRefTag(original, core) {
   const match = core.text ? core.text.match(REF_TAG) : null;
@@ -3055,15 +3075,13 @@ function patchRefTag(original, core) {
   if (!tag) return original;
   const base = original;
   if (base.role === "assistant") return original;
-  const tagCore = tag.replace(/\s+$/, "");
-  let bodyStart = tagCore.length;
-  if (core.text && core.text.charAt(bodyStart) === "\n") bodyStart += 1;
-  const coreBody = core.text ? core.text.slice(bodyStart) : "";
+  const coreBody = coreBodyOf(core.text ?? "", tag);
   const originalBody = extractText(base.content);
   const trimEnd = (s) => s.replace(/\s+$/, "");
   if (coreBody && trimEnd(coreBody) !== trimEnd(originalBody)) {
-    return rebuildBodyFromCore(original, coreBody, tag);
+    return rebuildBodyFromCore(original, coreBody, rewriteTagTokens(tag, coreBody));
   }
+  const stableTag = rewriteTagTokens(tag, originalBody);
   const rawBlocks = Array.isArray(base.content) ? base.content : typeof base.content === "string" ? [{ type: "text", text: base.content }] : [];
   const peeled = peelRefTagBlocks(rawBlocks);
   const newBlocks = [...peeled];
@@ -3074,7 +3092,7 @@ function patchRefTag(original, core) {
       const baseText = b.text.replace(/\n*$/, "");
       newBlocks[i] = { ...b, text: `${baseText}
 
-${tag}` };
+${stableTag}` };
       injected = true;
       break;
     }
@@ -3084,7 +3102,7 @@ ${tag}` };
   }
   return {
     ...original,
-    content: [...peeled, { type: "text", text: tag }]
+    content: [...peeled, { type: "text", text: stableTag }]
   };
 }
 function rebuildBodyFromCore(original, coreBody, tag) {
@@ -10088,7 +10106,7 @@ async function statusReport(runtime, ctx) {
   const modelId = ctx.model?.id ?? "default";
   const sentTokens = estimateTokens(coreMessages, coveredIds) + systemPromptTokens;
   const turn = runtime.core.processTurn({ messages: coreMessages, state, config, tokenCount: calibrateTokens(sentTokens, runtime.density.densityFor(modelId)) });
-  const versionStr = "0.1.40" ? `billion-context-pi@${"0.1.40"}` : void 0;
+  const versionStr = "0.1.41" ? `billion-context-pi@${"0.1.41"}` : void 0;
   let text = buildStatusPanel({
     version: versionStr,
     tokenCount: sessionTokens,
@@ -10410,7 +10428,7 @@ async function checkForUpdate(autoUpdate, notify) {
     const data = await res.json();
     const latest = data.version;
     if (!latest) return;
-    const current = runtimeVersion ?? "0.1.40";
+    const current = runtimeVersion ?? "0.1.41";
     const hasUpdate = isNewer(latest, current);
     debug.event("update-check", {
       current,
@@ -10607,7 +10625,7 @@ function wireSessionLifecycle(pi, runtime) {
     const sid = ctx.sessionManager.getSessionId();
     runtime.clearSessionTracking(sid);
     const modelInfo = ctx.model;
-    logInfo("session", { event: "start", sid, cwd: ctx.cwd, debug: runtime.adapter.debug ?? null, version: true ? "0.1.40" : null, model: modelInfo?.id ?? null, modelApi: modelInfo?.api ?? null, contextWindow: modelInfo?.contextWindow ?? null });
+    logInfo("session", { event: "start", sid, cwd: ctx.cwd, debug: runtime.adapter.debug ?? null, version: true ? "0.1.41" : null, model: modelInfo?.id ?? null, modelApi: modelInfo?.api ?? null, contextWindow: modelInfo?.contextWindow ?? null });
     try {
       await runtime.reloadConfig(ctx.cwd);
       setDelegateDisplayUsage(resolveDelegate(runtime.adapter).displayUsage);