agy-statusblocks 0.1.5 → 0.1.6

package/dist/segments/context.js

@@ -7,21 +7,43 @@ function formatTokens(n) {
         return `${Math.round(safe / 1000)}K`;
     return String(safe);
 }
+function resolveContextUsage(total, rawPct, inTokens, outTokens) {
+    if (rawPct != null && rawPct > 0) {
+        const inPct = total > 0 ? (inTokens / total) * 100 : 0;
+        const sumPct = total > 0 ? ((inTokens + outTokens) / total) * 100 : 0;
+        const inDiff = Math.abs(inPct - rawPct);
+        const sumDiff = Math.abs(sumPct - rawPct);
+        let used;
+        if (inTokens > 0 && inDiff <= sumDiff && inDiff <= 1) {
+            // Antigravity sets total_input_tokens = estimated_tokens_used and
+            // used_percentage = (estimated_tokens_used / context_window_size) * 100,
+            // while leaving total_output_tokens as cumulative lifetime session output.
+            used = inTokens;
+        }
+        else if (inTokens + outTokens > 0 && sumDiff <= 1) {
+            // Fallback step-sum mode where used_percentage reflects inTokens + outTokens.
+            used = inTokens + outTokens;
+        }
+        else {
+            // Only used_percentage is reliable/present; derive used token count from it.
+            used = Math.round((rawPct / 100) * total);
+        }
+        const pct = Math.min(100, Math.max(0, Math.round(rawPct)));
+        return { pct, used };
+    }
+    const used = inTokens + outTokens;
+    const pct = total > 0 ? Math.min(100, Math.max(0, Math.round((used / total) * 100))) : 0;
+    return { pct, used };
+}
 export const contextSegment = {
     id: 'context',
     priority: 10,
     enabled: () => true,
     render(data) {
         const total = data.context_window.context_window_size || 1048576;
-        let pct = data.context_window.used_percentage != null
-            ? Math.floor(data.context_window.used_percentage)
-            : 0;
         const inTokens = data.context_window.total_input_tokens ?? 0;
         const outTokens = data.context_window.total_output_tokens ?? 0;
-        const used = inTokens + outTokens;
-        if (pct === 0 && used > 0 && total > 0) {
-            pct = Math.floor((used / total) * 100);
-        }
+        const { pct, used } = resolveContextUsage(total, data.context_window.used_percentage, inTokens, outTokens);
         const barColor = pctColor(pct);
         const info = `${color(`${pct}%`, barColor, c.bold)}${color(' · ', c.dim)}${color(`${formatTokens(used)}/${formatTokens(total)}`, c.dim)}`;
         const cardWidth = Math.max(16, visibleLength(info));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agy-statusblocks",
-  "version": "0.1.5",
+  "version": "0.1.6",
   "description": "Opinionated, adaptive block-based status line for Antigravity CLI (agy) with bin-packing layout, rate limit tracking, and live agent telemetry",
   "type": "module",
   "bin": {