pi-thread-engine 0.4.5 → 0.4.6

package/PLAN.md

@@ -1,6 +1,6 @@
 # pi-thread-engine
 
-## Current State (v0.4.4)
+## Current State (v0.4.5)
 
 ### Working
 - All 7 thread types: Base (via /pthread), P (parallel), C (chained), B (branch/meta), F (fusion), Z (zero-touch), L (long)
@@ -15,7 +15,7 @@
 
 ## Next Steps (ranked by impact/effort)
 
-### P0 — Required for completeness
+### ✅ P0 — Required for completeness
 1. **Live progress with % + ETA** — Wire event emitter from executor to dashboard for real-time progress
 2. **Token/cost counters** — Hook into pi's usage tracking per thread (use `pi.exec` output parsing)
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-thread-engine",
-  "version": "0.4.5",
+  "version": "0.4.6",
   "description": "Thread-Based Engineering for pi — all 7 thread types + stories + fusion + zero-touch + TUI dashboard. Based on @IndyDevDan framework from agenticengineer.com.",
   "type": "module",
   "license": "MIT",

package/src/core/executor.ts

@@ -31,7 +31,25 @@ export class ThreadExecutor {
 			});
 
 			if (result.code === 0) {
-				this.registry.completeTask(thread.id, task.id, result.stdout);
+				this.registry.completeTask(thread.id, task.id, result.stdout);
+			// Parse --mode json output for token/cost data
+			try {
+				const jsonLines = result.stdout.split("\n").filter(l => l.startsWith("{\"type\":\"message_end\"}"));
+				const last = jsonLines[jsonLines.length - 1];
+				if (last) {
+					const msg = JSON.parse(last);
+					if (msg.message?.usage) {
+						const u = msg.message.usage;
+						task.usage = {
+							inputTokens: u.input ?? 0,
+							outputTokens: u.output ?? 0,
+							totalTokens: u.totalTokens ?? 0,
+							cacheRead: u.cacheRead ?? 0,
+							cost: (u.cost?.input ?? 0) + (u.cost?.output ?? 0),
+						};
+					}
+				}
+			} catch {}
 			} else {
 				this.registry.failTask(thread.id, task.id, result.stderr || `Exit code: ${result.code}`);
 			}

package/src/core/registry.ts

@@ -199,6 +199,8 @@ export class ThreadRegistry {
 			state: thread.state,
 			progress,
 			elapsed,
+		totalTokens: thread.tasks.reduce((a, tk) => a + (tk.usage?.totalTokens ?? 0), 0),
+		totalCost: thread.tasks.reduce((a, tk) => a + (tk.usage?.cost ?? 0), 0),
 			backend: thread.config.backend,
 		};
 	}

package/src/core/types.ts

@@ -9,6 +9,7 @@ export type ExecutionBackend = "subagent" | "native";
 
 /** A single unit of work within a thread */
 export interface ThreadTask {
+	usage?: { inputTokens: number; outputTokens: number; cacheRead: number; totalTokens: number; cost: number };
 	id: string;
 	label: string;
 	prompt: string;
@@ -85,6 +86,8 @@ export interface ThreadSummary {
 	state: ThreadState;
 	progress: string;
 	elapsed: string;
+	totalTokens: number;
+	totalCost: number;
 	backend: ExecutionBackend;
 }
 

package/src/dashboard.ts

@@ -194,7 +194,11 @@ export function createDashboard(
 				const icon = stateIcon(task.state);
 				const color = stateColor(task.state);
 				lines.push(theme.fg(color, `${indent}${icon} ${task.id}: ${truncateToWidth(task.label, maxW)}`));
-				if (task.model) lines.push(theme.fg("dim", `${indent}  model: ${task.model}`));
+				if (task.model) lines.push(theme.fg("dim", `${indent}  model: ${task.model}`));
+				if (task.usage) {
+					const costStr = task.usage.cost > 0 ? ` ${task.usage.cost.toFixed(4)}` : "";
+					lines.push(theme.fg("dim", `${indent}  tokens: ${task.usage.totalTokens}${costStr}`));
+				}
 				if (task.result) {
 					const preview = task.result.replace(/\n/g, " ").slice(0, 200);
 					lines.push(theme.fg("muted", `${indent}  → ${truncateToWidth(preview, maxW)}`));