@wrongstack/webui 0.4.1 → 0.5.2

package/dist/index.html

@@ -5,7 +5,7 @@
     <link rel="icon" type="image/svg+xml" href="/wrongstack.svg" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <title>WrongStack WebUI</title>
-    <script type="module" crossorigin src="/assets/index-Do84Z2Fv.js"></script>
+    <script type="module" crossorigin src="/assets/index-q1lcmxqy.js"></script>
     <link rel="modulepreload" crossorigin href="/assets/vendor-Dff2jyfM.js">
     <link rel="stylesheet" crossorigin href="/assets/index-B-u6omip.css">
   </head>

package/dist/index.js

@@ -795,15 +795,22 @@ var useSessionStore = create2()(
       todos: [],
       setSession: (session) => set({ session }),
       updateUsage: (usage) => set((state) => {
-        const totalInput = (usage.input ?? 0) + (usage.cacheRead ?? 0) + (usage.cacheWrite ?? 0);
+        const inputDelta = usage.input + (usage.cacheRead ?? 0) + (usage.cacheWrite ?? 0);
+        const cacheReadDelta = usage.cacheRead ?? 0;
+        const cacheWriteDelta = usage.cacheWrite ?? 0;
         return {
           totalTokens: {
             input: state.totalTokens.input + usage.input,
             output: state.totalTokens.output + usage.output,
-            cacheRead: (state.totalTokens.cacheRead ?? 0) + (usage.cacheRead ?? 0),
-            cacheWrite: (state.totalTokens.cacheWrite ?? 0) + (usage.cacheWrite ?? 0)
+            cacheRead: (state.totalTokens.cacheRead ?? 0) + cacheReadDelta,
+            cacheWrite: (state.totalTokens.cacheWrite ?? 0) + cacheWriteDelta
           },
-          lastInputTokens: totalInput || state.lastInputTokens
+          // lastInputTokens = the single most-recent turn's effective input
+          // (fresh + cached). This drives the ctx % chip in the topbar and
+          // is NOT additive — it's the latest provider response's token count.
+          // Use cacheWrite too because cache-write tokens were part of this
+          // turn's context cost (written to build the cache for next turn).
+          lastInputTokens: inputDelta || state.lastInputTokens
         };
       }),
       addCost: (cost) => set((state) => ({ cost: state.cost + cost })),
@@ -1055,6 +1062,15 @@ function installHandlers(ws) {
     });
     useSessionStore.setState({ lastInputTokens: payload.after });
   });
+  on("provider.response", (msg) => {
+    const payload = msg.payload;
+    const u = payload?.usage;
+    if (!u) return;
+    const delta = (u.input ?? 0) + (u.cacheWrite ?? 0) - (u.cacheRead ?? 0);
+    if (delta > 0) {
+      useSessionStore.setState({ lastInputTokens: delta });
+    }
+  });
   on("context.repaired", (msg) => {
     const payload = msg.payload;
     const removed = payload.removedToolUses.length + payload.removedToolResults.length + payload.removedMessages;