zames_pro 2.19.0 → 2.20.0

package/README.md

@@ -158,8 +158,9 @@ Codex CLI:
 
 The token context is also shown live: the status line above the input has the
 spinner/text on the left and the context on the right (e.g. `125k · 13%`,
-percent of a 1M context). It comes from DeepSeek's `accumulated_token_usage`
-and is hidden until the first answer delivers it.
+percent of a 1M context). It is COLORED by fill level: green below 50%,
+yellow 50-80%, red above 80%. It comes from DeepSeek's
+`accumulated_token_usage` and is hidden until the first answer delivers it.
 
 
 ## Project context, skills and memory

package/dist/browser.js

@@ -1196,13 +1196,14 @@ export class DeepSeekBrowser {
         // replied to the wrong thing (or nothing), and the loop looked stalled.
         if (norm(got) !== norm(text)) {
             await input.click();
-            await this.page.keyboard.press("Control+A");
-            await this.page.keyboard.press("Delete");
+            await this.page.keyboard.press('Control+A');
+            await this.page.keyboard.press('Delete');
             await this.page.keyboard.insertText(text);
             const got2 = await input.evaluate((el) => {
-                if (el.tagName.toLowerCase() === "textarea" || el.tagName.toLowerCase() === "input")
+                if (el.tagName.toLowerCase() === 'textarea' ||
+                    el.tagName.toLowerCase() === 'input')
                     return el.value;
-                return el.innerText || el.textContent || "";
+                return el.innerText || el.textContent || '';
             });
             if (norm(got2) !== norm(text)) {
                 throw new Error(this._t('ds.input_partial', {
@@ -1366,7 +1367,14 @@ export class DeepSeekBrowser {
         // the "new answer" check never fired — the source of the
         // ds.send_no_new_answer flapping.
         const beforeText = await this._readLastAnswerTextCleanDom().catch(() => '');
-        this._askDebug('SEND agent=' + agent + ' len=' + prompt.length + ' beforeLen=' + beforeText.length + ' beforeHead=' + JSON.stringify(beforeText.slice(0, 60)));
+        this._askDebug('SEND agent=' +
+            agent +
+            ' len=' +
+            prompt.length +
+            ' beforeLen=' +
+            beforeText.length +
+            ' beforeHead=' +
+            JSON.stringify(beforeText.slice(0, 60)));
         await this._waitForSendSlot(agent);
         // Esc/Ctrl+C pressed during the pause — do not send anything.
         if (this._abort)
@@ -1459,7 +1467,12 @@ export class DeepSeekBrowser {
             const netStarted = !!this._netCapture && this._netCaptureAt >= this._lastSentAt;
             if (changed || netStarted || bodyLen > startBodyLen) {
                 started = true;
-                this._askDebug('STARTED changed=' + changed + ' netStarted=' + netStarted + ' bodyGrew=' + (bodyLen > startBodyLen));
+                this._askDebug('STARTED changed=' +
+                    changed +
+                    ' netStarted=' +
+                    netStarted +
+                    ' bodyGrew=' +
+                    (bodyLen > startBodyLen));
                 break;
             }
             // Fallback for an echo: the text equals beforeText, so it is the OLD
@@ -1480,7 +1493,18 @@ export class DeepSeekBrowser {
                 settledTicks = 0;
             }
             lastStartCur = cur;
-            this._askDebug('START-loop curLen=' + cur.length + ' changed=' + changed + ' netStarted=' + netStarted + ' bodyLen=' + bodyLen + ' settled=' + settledTicks + ' generating=' + notGenerating);
+            this._askDebug('START-loop curLen=' +
+                cur.length +
+                ' changed=' +
+                changed +
+                ' netStarted=' +
+                netStarted +
+                ' bodyLen=' +
+                bodyLen +
+                ' settled=' +
+                settledTicks +
+                ' generating=' +
+                notGenerating);
             await this.page.waitForTimeout(300);
         }
         if (!started) {
@@ -1490,7 +1514,10 @@ export class DeepSeekBrowser {
             // previous tool call.
             const cur = await this._readLastAnswerTextCleanDom().catch(() => '');
             const fresh = !!this._netCapture && this._netCaptureAt >= this._lastSentAt;
-            if (cur && cur.trim() && normText(cur) !== normText(beforeText) && !(await this._isGenerating())) {
+            if (cur &&
+                cur.trim() &&
+                normText(cur) !== normText(beforeText) &&
+                !(await this._isGenerating())) {
                 return cur;
             }
             if (fresh) {
@@ -1550,14 +1577,20 @@ export class DeepSeekBrowser {
                 }
             }
             const netFresh = !!this._netCapture && this._netCaptureAt >= this._lastSentAt;
-            const cur = await this._readLastAnswerTextCleanDom().catch(() => '');
+            // The DOM may be empty while the answer is already in the network
+            // capture (a resumed chat re-renders its history, the answer selector
+            // may lag, or page.evaluate times out). In that case fall back to the
+            // raw capture: it is the CURRENT answer and must not be ignored just
+            // because the DOM has not caught up.
+            let cur = await this._readLastAnswerTextCleanDom().catch(() => '');
+            if (!cur && netFresh)
+                cur = this._cleanAnswer(this._netCapture);
             // Ignore an "answer" that is identical to what was on the page BEFORE we
             // sent the message: that is the previous answer, not a new one. Returning
             // it would make the agent re-process the old tool call (or silently
             // stop). We keep waiting instead. A fresh network capture is exempt: it
             // belongs to the CURRENT send even if the DOM still shows the old text.
-            const isNew = !!cur &&
-                (netFresh || normText(cur) !== normText(beforeText));
+            const isNew = !!cur && (netFresh || normText(cur) !== normText(beforeText));
             // An echo/stale answer equals beforeText, so isNew stays false and the
             // old loop waited until the full timeout — the "agent stopped after a
             // tool call" hang. If generation has clearly ENDED (no Stop button) and
@@ -1581,7 +1614,18 @@ export class DeepSeekBrowser {
             }
             if (isNew)
                 last = cur;
-            this._askDebug('FIN-loop isNew=' + isNew + ' sameAsBefore=' + sameAsBefore + ' stable=' + stable + ' curLen=' + cur.length + ' lastLen=' + last.length + ' netFresh=' + netFresh);
+            this._askDebug('FIN-loop isNew=' +
+                isNew +
+                ' sameAsBefore=' +
+                sameAsBefore +
+                ' stable=' +
+                stable +
+                ' curLen=' +
+                cur.length +
+                ' lastLen=' +
+                last.length +
+                ' netFresh=' +
+                netFresh);
             await this.page.waitForTimeout(Math.max(0, this.stabilityDelayMs));
         }
         if (last && (normText(last) !== normText(beforeText) || this._netCapture)) {
@@ -1593,7 +1637,10 @@ export class DeepSeekBrowser {
         if (this._netCapture && this._netCaptureAt >= this._lastSentAt) {
             return this._netCapture;
         }
-        this._askDebug('THROW no-new-answer lastLen=' + last.length + ' beforeLen=' + beforeText.length);
+        this._askDebug('THROW no-new-answer lastLen=' +
+            last.length +
+            ' beforeLen=' +
+            beforeText.length);
         throw new Error(this._t('ds.send_no_new_answer'));
     }
     async dumpDom(filePath) {

package/dist/input.js

@@ -174,6 +174,24 @@ export function layoutInput(promptStr, buf, cursor, cols) {
 // percentage of CONTEXT_LIMIT. Exported so it is unit-tested without a live
 // editor. A null/undefined/NaN count renders an empty string (no status).
 export const CONTEXT_LIMIT = 1_000_000;
+// Thresholds for coloring the context counter. Below YELLOW the value is
+// green (plenty of room), between YELLOW and RED it is yellow (getting full),
+// above RED it is red (nearly exhausted). Percent of CONTEXT_LIMIT.
+export const CONTEXT_YELLOW_PCT = 50;
+export const CONTEXT_RED_PCT = 80;
+// The color role for a context fill level, as a stable string so it can be
+// unit-tested without a terminal: 'ok' | 'warn' | 'high'. Null tokens -> null.
+export function tokenStatusLevel(tokens, limit = CONTEXT_LIMIT) {
+    if (typeof tokens !== 'number' || !Number.isFinite(tokens) || tokens < 0) {
+        return null;
+    }
+    const pct = (tokens / limit) * 100;
+    if (pct >= CONTEXT_RED_PCT)
+        return 'high';
+    if (pct >= CONTEXT_YELLOW_PCT)
+        return 'warn';
+    return 'ok';
+}
 export function formatTokenStatus(tokens, limit = CONTEXT_LIMIT) {
     if (typeof tokens !== 'number' || !Number.isFinite(tokens) || tokens < 0) {
         return '';
@@ -247,6 +265,8 @@ export class LineEditor {
     // percentage of the context limit, right-aligned above the input line. Null
     // hides it. Updated by the caller from browser.getLastTokenUsage().
     contextStatus;
+    // The raw token count behind contextStatus (for the color level).
+    contextTokens;
     // A callback the editor calls to fetch the CURRENT token count before every
     // status render, so the status line stays fresh without the caller polling.
     onContextQuery;
@@ -283,6 +303,7 @@ export class LineEditor {
         this.onClipboard = null;
         this.locked = false;
         this.contextStatus = null;
+        this.contextTokens = null;
         this.onContextQuery = null;
     }
     // The token status for the CURRENT render: refreshed from onContextQuery
@@ -291,7 +312,9 @@ export class LineEditor {
         if (this.onContextQuery) {
             try {
                 const n = this.onContextQuery();
-                this.contextStatus = n === null || n === undefined ? null : formatTokenStatus(n) || null;
+                this.contextTokens = typeof n === 'number' ? n : null;
+                this.contextStatus =
+                    n === null || n === undefined ? null : formatTokenStatus(n) || null;
             }
             catch {
                 // A broken callback must never break the render.
@@ -299,6 +322,19 @@ export class LineEditor {
         }
         return this.contextStatus;
     }
+    // The context text colored by fill level: green (ok), yellow (warn),
+    // red (high).
+    _contextText() {
+        const ctx = this._contextForRender();
+        if (!ctx)
+            return '';
+        const level = tokenStatusLevel(this.contextTokens);
+        if (level === 'high')
+            return theme.error(ctx);
+        if (level === 'warn')
+            return theme.warn(ctx);
+        return theme.success(ctx);
+    }
     // Read the OS clipboard for an image and insert its marker. Used when the
     // terminal sends no usable paste data (Ctrl+V / right-click / empty paste).
     async _tryClipboard() {
@@ -420,6 +456,8 @@ export class LineEditor {
     }
     // Update the token-context text shown at the right of the status line.
     setContextStatus(tokens) {
+        this.contextTokens =
+            typeof tokens === 'number' && Number.isFinite(tokens) ? tokens : null;
         this.contextStatus = formatTokenStatus(tokens) || null;
         this._render();
     }
@@ -459,19 +497,31 @@ export class LineEditor {
         // token context right-aligned on the SAME row (10k · 12%). The context is
         // refreshed from onContextQuery() on every render, so it follows the live
         // DeepSeek counter without a polling timer of its own.
-        const ctx = this._contextForRender();
-        const ctxText = ctx ? theme.dim(ctx) : '';
+        const ctxText = this._contextText();
         if (this.statusText) {
             if (ctxText) {
-                const pad = Math.max(1, cols - visLen(this.statusText) - visLen(ctxText));
-                out += this.statusText + ' '.repeat(pad) + ctxText + NL;
+                // Right-align the context on the SAME row as the status. When the
+                // status is too long to leave room, do NOT cram them together (that
+                // pushed the trailing `%` past the right edge and it got truncated):
+                // put the context on its own line instead.
+                const space = cols - visLen(this.statusText) - visLen(ctxText);
+                if (space >= 2) {
+                    out += this.statusText + ' '.repeat(space) + ctxText + NL;
+                    top = visRows(this.statusText + ' '.repeat(2) + ctxText, cols);
+                }
+                else {
+                    out += this.statusText + NL;
+                    top = visRows(this.statusText, cols);
+                    out += ' '.repeat(Math.max(0, cols - visLen(ctxText))) + ctxText + NL;
+                    top += visRows(ctxText, cols);
+                }
             }
             else {
                 out += this.statusText + NL;
+                // The status may wrap onto several lines — we account for this,
+                // otherwise the block erase misses and statuses pile up.
+                top = visRows(this.statusText, cols);
             }
-            // The status may wrap onto several lines — we account for this,
-            // otherwise the block erase misses and statuses pile up.
-            top = visRows(this.statusText + (ctxText ? ' '.repeat(2) + ctxText : ''), cols);
         }
         else if (ctxText) {
             // Idle: no spinner, but the context still belongs on its own line just

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "zames_pro",
-  "version": "2.19.0",
+  "version": "2.20.0",
   "description": "Terminal coding agent over chat.deepseek.com via Playwright",
   "type": "module",
   "bin": {
@@ -15,7 +15,7 @@
   ],
   "scripts": {
     "start": "node ./dist/index.js",
-    "dev": "ZAMES_NET_DEBUG=1 tsx ./src/index.ts --dev",
+    "dev": "ZAMES_NET_DEBUG=1 tsx ./src/index.ts --dev --headed",
     "build": "tsc -p tsconfig.build.json",
     "typecheck": "tsc --noEmit",
     "prepublishOnly": "npm run build",