pi-ui-extend 0.1.67 → 0.1.68

package/dist/app/session/lazy-session-manager.js

@@ -215,8 +215,8 @@ class LazySessionManager {
     createBranchedSession(leafId) {
         return this.hydrate().createBranchedSession(leafId);
     }
-    branchWithSummary(branchFromId, summary, details, fromHook) {
-        return this.hydrate().branchWithSummary(branchFromId, summary, details, fromHook);
+    branchWithSummary(branchFromId, summary, details, fromHook, usage) {
+        return this.hydrate().branchWithSummary(branchFromId, summary, details, fromHook, usage);
     }
     appendLabelChange(targetId, label) {
         if (this.hydrated)
@@ -248,14 +248,16 @@ class LazySessionManager {
             return this.hydrated.appendModelChange(provider, modelId);
         return this.appendEntry(this.newEntry("model_change", { provider, modelId }));
     }
-    appendCompaction(summary, firstKeptEntryId, tokensBefore, details, fromHook) {
+    appendCompaction(summary, firstKeptEntryId, tokensBefore, details, fromHook, usage) {
         if (this.hydrated)
-            return this.hydrated.appendCompaction(summary, firstKeptEntryId, tokensBefore, details, fromHook);
+            return this.hydrated.appendCompaction(summary, firstKeptEntryId, tokensBefore, details, fromHook, usage);
         const payload = { summary, firstKeptEntryId, tokensBefore };
         if (details !== undefined)
             payload.details = details;
         if (fromHook !== undefined)
             payload.fromHook = fromHook;
+        if (usage !== undefined)
+            payload.usage = usage;
         return this.appendEntry(this.newEntry("compaction", payload));
     }
     appendCustomEntry(customType, data) {

package/dist/app/session/session-stats.js

@@ -17,7 +17,18 @@ export function aggregateSessionStats(entries, base) {
     let cacheRead = 0;
     let cacheWrite = 0;
     let cost = 0;
+    function addUsage(usage) {
+        input += finiteNumber(usage?.input);
+        output += finiteNumber(usage?.output);
+        cacheRead += finiteNumber(usage?.cacheRead);
+        cacheWrite += finiteNumber(usage?.cacheWrite);
+        cost += finiteNumber(usage?.cost?.total);
+    }
     for (const entry of entries) {
+        if (entry.type === "compaction" || entry.type === "branch_summary") {
+            addUsage(entry.usage);
+            continue;
+        }
         if (entry.type !== "message")
             continue;
         totalMessages += 1;
@@ -28,6 +39,7 @@ export function aggregateSessionStats(entries, base) {
         }
         if (message.role === "toolResult") {
             toolResults += 1;
+            addUsage(message.usage);
             continue;
         }
         if (message.role !== "assistant")
@@ -36,12 +48,7 @@ export function aggregateSessionStats(entries, base) {
         if (Array.isArray(message.content)) {
             toolCalls += message.content.filter((content) => content.type === "toolCall").length;
         }
-        const usage = message.usage;
-        input += finiteNumber(usage?.input);
-        output += finiteNumber(usage?.output);
-        cacheRead += finiteNumber(usage?.cacheRead);
-        cacheWrite += finiteNumber(usage?.cacheWrite);
-        cost += finiteNumber(usage?.cost?.total);
+        addUsage(message.usage);
     }
     return {
         ...base,

package/external/pi-tools-suite/package.json

@@ -43,9 +43,9 @@
     "vscode-languageserver-protocol": "^3.17.5"
   },
   "peerDependencies": {
-    "@earendil-works/pi-ai": "0.80.10",
-    "@earendil-works/pi-coding-agent": "0.80.10",
-    "@earendil-works/pi-tui": "0.80.10",
+    "@earendil-works/pi-ai": "0.81.0",
+    "@earendil-works/pi-coding-agent": "0.81.0",
+    "@earendil-works/pi-tui": "0.81.0",
     "typebox": "*"
   },
   "devDependencies": {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-ui-extend",
-  "version": "0.1.67",
+  "version": "0.1.68",
   "description": "Pix: a workspace-first terminal UI for Pi with tabs, readable tool activity, voice input, and bundled agent tools.",
   "private": false,
   "repository": {
@@ -75,9 +75,9 @@
     "prepublishOnly": "npm run check && npm run build:pix && npm run generate-schemas"
   },
   "dependencies": {
-    "@earendil-works/pi-ai": "0.80.10",
-    "@earendil-works/pi-coding-agent": "0.80.10",
-    "@earendil-works/pi-tui": "0.80.10",
+    "@earendil-works/pi-ai": "0.81.0",
+    "@earendil-works/pi-coding-agent": "0.81.0",
+    "@earendil-works/pi-tui": "0.81.0",
     "@mariozechner/clipboard": "^0.3.9",
     "jsonc-parser": "3.3.1",
     "typebox": "1.1.38",