@aexol/spectral 0.2.18 → 0.2.20

package/dist/server/pi-bridge.js

@@ -656,6 +656,10 @@ export class PiBridge {
         if (this.disposed)
             return;
         this.disposed = true;
+        // Persist the current in-flight message (if any) before tearing down.
+        // Without this, cancelling mid-turn loses the message along with its
+        // token_usage event, which breaks context-window tracking on reconnect.
+        this.finalizePendingMessage();
         try {
             this.unsubscribe?.();
         }
@@ -829,8 +833,17 @@ export class PiBridge {
                 // Emit token usage for this assistant message. pi provides token
                 // counts via ev.message.usage; cost is computed from the model's
                 // configured pricing (or null when unavailable).
+                //
+                // Skip zero-total-usage events — they happen when a turn is
+                // cancelled before the provider starts streaming, and emitting
+                // inputTokens:0 would mislead the UI into showing 0% context window
+                // usage after reconnect.
                 const usage = ev.message.usage;
-                if (usage) {
+                const totalTokens = (usage?.input ?? 0) +
+                    (usage?.output ?? 0) +
+                    (usage?.cacheRead ?? 0) +
+                    (usage?.cacheWrite ?? 0);
+                if (usage && totalTokens > 0) {
                     const usageEvent = {
                         type: "token_usage",
                         messageId,
@@ -839,7 +852,7 @@ export class PiBridge {
                             outputTokens: usage.output ?? 0,
                             cacheReadTokens: usage.cacheRead ?? 0,
                             cacheWriteTokens: usage.cacheWrite ?? 0,
-                            totalTokens: usage.totalTokens ?? 0,
+                            totalTokens: usage.totalTokens ?? totalTokens,
                             cost: usage.cost?.total ?? null,
                         },
                     };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aexol/spectral",
-  "version": "0.2.18",
+  "version": "0.2.20",
   "description": "Always-on coding agent for Aexol — branded pi wrapper with relay-based browser access.",
   "type": "module",
   "private": false,