@fosterg4/pi-subagent 1.0.2 → 1.0.3

package/index.ts

@@ -35,12 +35,15 @@ import {
 	formatAgentList,
 } from "./agents.ts";
 import { type ValidationResult, validateSchema } from "./validate.ts";
+import { fmt, usageLine, sumUsage } from "./utils.ts";
 
 const MAX_PARALLEL_TASKS = 8;
 const MAX_CONCURRENCY = 4;
 const PER_TASK_OUTPUT_CAP = 50 * 1024;
 
+/* eslint-disable @typescript-eslint/no-unused-vars */
 
+interface UsageStats {
 	input: number;
 	output: number;
 	cacheRead: number;
@@ -954,11 +957,13 @@ export default function (pi: ExtensionAPI) {
 				return theme.fg("error", "\u2717");
 			};
 
-			// --- Single mode ---
+		// --- Single mode ---
 			if (details.mode === "single" && details.results.length === 1) {
 				const r = details.results[0];
 				const status = getStatusText(r);
 				const finalOutput = getFinalOutputText(r.messages);
+				const usage = usageLine(r.usage);
+				const usg = usage ? theme.fg("dim", usage) : "";
 
 				if (expanded) {
 					const container = new Container();
@@ -969,6 +974,10 @@ export default function (pi: ExtensionAPI) {
 						container.addChild(new Spacer(1));
 						container.addChild(new Markdown(finalOutput.trim(), 0, 0, mdTheme));
 					}
+					if (usg) {
+						container.addChild(new Spacer(1));
+						container.addChild(new Text(usg, 0, 0));
+					}
 					return container;
 				}
 
@@ -981,6 +990,7 @@ export default function (pi: ExtensionAPI) {
 				} else {
 					text += `\n${theme.fg("muted", "(no output)")}`;
 				}
+				if (usg) text += `\n${usg}`;
 				return new Text(text, 0, 0);
 			}
 
@@ -991,6 +1001,8 @@ export default function (pi: ExtensionAPI) {
 				const allOk = details.results.every((r) => r.exitCode === 0);
 				const icon = allOk ? theme.fg("success", "\u2713") : theme.fg("error", "\u2717");
 				const steps = details.results.map((r) => r.agent).join(" \u2192 ");
+				const total = sumUsage(details.results);
+				const totalUsg = usageLine(total);
 
 				if (expanded) {
 					const container = new Container();
@@ -999,18 +1011,19 @@ export default function (pi: ExtensionAPI) {
 					);
 					for (const r of details.results) {
 						const out = getFinalOutputText(r.messages);
+						const stepUsage = usageLine(r.usage);
+						const stepUsg = stepUsage ? theme.fg("dim", stepUsage) : "";
+						const label = `${getStatusText(r)} ${theme.fg("accent", r.agent)}${r.model ? theme.fg("muted", ` \u00B7 ${r.model}`) : ""}${stepUsg ? "  " + stepUsg : ""}`;
 						if (out) {
 							container.addChild(new Spacer(1));
-							container.addChild(
-								new Text(
-									`${getStatusText(r)} ${theme.fg("accent", r.agent)}${r.model ? theme.fg("muted", ` \u00B7 ${r.model}`) : ""}`,
-									0,
-									0,
-								),
-							);
+							container.addChild(new Text(label, 0, 0));
 							container.addChild(new Markdown(out.trim(), 0, 0, mdTheme));
 						}
 					}
+					if (totalUsg && details.results.length > 1) {
+						container.addChild(new Spacer(1));
+						container.addChild(new Text(totalUsg, 0, 0));
+					}
 					return container;
 				}
 
@@ -1023,6 +1036,7 @@ export default function (pi: ExtensionAPI) {
 				} else {
 					text += `\n${theme.fg("muted", "(no output)")}`;
 				}
+				if (totalUsg) text += `\n${totalUsg}`;
 				return new Text(text, 0, 0);
 			}
 
@@ -1039,6 +1053,8 @@ export default function (pi: ExtensionAPI) {
 
 				if (expanded && running === 0) {
 					const container = new Container();
+					const total = sumUsage(details.results);
+					const totalUsg = usageLine(total);
 					container.addChild(
 						new Text(
 							`${icon} ${theme.fg("accent", `${done} tasks`)}`,
@@ -1048,11 +1064,13 @@ export default function (pi: ExtensionAPI) {
 					);
 					for (const r of details.results) {
 						const out = getFinalOutputText(r.messages);
+						const stepUsage = usageLine(r.usage);
+						const stepUsg = stepUsage ? theme.fg("dim", stepUsage) : "";
 						if (out) {
 							container.addChild(new Spacer(1));
 							container.addChild(
 								new Text(
-									`${getStatusText(r)} ${theme.fg("accent", r.agent)}`,
+									`${getStatusText(r)} ${theme.fg("accent", r.agent)}${stepUsg ? "  " + stepUsg : ""}`,
 									0,
 									0,
 								),
@@ -1060,11 +1078,19 @@ export default function (pi: ExtensionAPI) {
 							container.addChild(new Markdown(out.trim(), 0, 0, mdTheme));
 						}
 					}
+					if (totalUsg && details.results.length > 1) {
+						container.addChild(new Spacer(1));
+						container.addChild(new Text(totalUsg, 0, 0));
+					}
 					return container;
 				}
 
+				const usg = usageLine(sumUsage(details.results));
 				let text = `${icon} ${theme.fg("accent", `${done}/${details.results.length} tasks`)}`;
-				if (running === 0 && !expanded) text += theme.fg("muted", " (Ctrl+O to expand)");
+				if (running === 0) {
+					if (usg) text += `\n${usg}`;
+					if (!expanded) text += theme.fg("muted", " (Ctrl+O to expand)");
+				}
 				return new Text(text, 0, 0);
 			}
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@fosterg4/pi-subagent",
-  "version": "1.0.2",
+  "version": "1.0.3",
   "description": "Delegate tasks to specialized subagents with isolated context windows, structured JSON handoff, contract schemas, and live TUI streaming",
   "keywords": [
     "pi-package",

package/utils.ts

@@ -0,0 +1,37 @@
+/**
+ * Shared formatting helpers — testable without pi runtime deps
+ */
+
+import type { UsageStats } from "./index.ts";
+
+export function fmt(n: number): string {
+	if (n < 1000) return n.toString();
+	if (n < 10000) return (n / 1000).toFixed(1) + "k";
+	if (n < 1000000) return Math.round(n / 1000) + "k";
+	return (n / 1000000).toFixed(1) + "M";
+}
+
+export function usageLine(u: UsageStats): string {
+	const parts: string[] = [];
+	if (u.input) parts.push("\u2191" + fmt(u.input));
+	if (u.output) parts.push("\u2193" + fmt(u.output));
+	if (u.cacheRead) {
+		parts.push("R" + fmt(u.cacheRead));
+		const total = u.input + u.cacheRead;
+		if (total > 0) parts.push("CH" + ((u.cacheRead / total) * 100).toFixed(1) + "%");
+	}
+	if (u.cost) parts.push("$" + u.cost.toFixed(4));
+	return parts.join(" ");
+}
+
+export function sumUsage(results: ReadonlyArray<{ usage: UsageStats }>): UsageStats {
+	const u: UsageStats = { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, cost: 0, contextTokens: 0, turns: 0 };
+	for (const r of results) {
+		u.input += r.usage.input;
+		u.output += r.usage.output;
+		u.cacheRead += r.usage.cacheRead;
+		u.cacheWrite += r.usage.cacheWrite;
+		u.cost += r.usage.cost;
+	}
+	return u;
+}