@henryqw/pi-footer 0.2.1 → 0.3.0

package/README.md

@@ -23,13 +23,13 @@ pi install npm:@henryqw/pi-footer
 
 ```text
 pi-packages · clear-field-f8d2 · PR #123 · approved
-↑ 12.4k · ↓ 2.1k · ↺ 84.3% · $ 0.127 · ◔ 36.8%    gpt-5.6-luna • high
+↑ 12.4k · ↓ 2.1k · ↺ 84.3% · ⚡ 87.4 t/s · $ 0.127 · ◔ 36.8%    gpt-5.6-luna • high
 Codex #1 · 50% · 7d 1d 1h 22m
 ```
 
 First line shows repository, branch, and pull request status (`pi-pr`) after clickable checkout link. Linked-worktree branches drop generated `worktree/` prefix.
 
-Second line shows cumulative input tokens, output tokens, latest cache-hit rate, estimated cost, and context usage. Unavailable values render as `—` without a misleading percent sign. Active model and thinking level are right-aligned.
+Second line shows cumulative input tokens, output tokens, latest cache-hit rate, tokens per second of the most recent assistant response, estimated cost, and context usage. Unavailable values render as `—` without a misleading percent sign. Active model and thinking level are right-aligned.
 
 `off` uses the same dim grey as the model name. Active levels use an ANSI-256 gradient: green `minimal`, yellow-green `low`, lime `medium`, yellow `high`, orange `xhigh`, and red `max`. `ultra` renders as a rainbow when the runtime supplies it. Pi 0.84.2 does not yet accept `ultra`, so that footer path remains unreachable until Pi adds it.
 

package/extensions/footer.ts

@@ -53,34 +53,62 @@ export default function footerExtension(pi: ExtensionAPI): void {
 		const commonName = commonDir && basename(commonDir) === ".git" ? basename(dirname(commonDir)) : undefined;
 		const repo = git.code === 0 ? commonName && commonName !== rootName ? commonName : rootName : basename(ctx.cwd);
 
+		let tps: number | undefined;
+		let assistantStartedAt: number | undefined;
+		pi.on("message_start", async (event) => {
+			if (event.message.role === "assistant") assistantStartedAt = performance.now();
+		});
+		pi.on("message_end", async (event) => {
+			if (event.message.role !== "assistant") return;
+			const output = event.message.usage?.output ?? 0;
+			const seconds = assistantStartedAt === undefined ? 0 : (performance.now() - assistantStartedAt) / 1000;
+			assistantStartedAt = undefined;
+			tps = seconds > 0 ? output / seconds : undefined;
+		});
+
+		// ponytail: keyed on length + last entry (sessions are append-only); revisit if entries ever mutate in place.
+		let usageKey: string | undefined;
+		let input = 0;
+		let output = 0;
+		let cost = 0;
+		let cacheRate: number | undefined;
+		const computeUsage = () => {
+			input = 0;
+			output = 0;
+			cost = 0;
+			cacheRate = undefined;
+			const add = (usage: Usage | undefined) => {
+				if (!usage) return;
+				input += usage.input;
+				output += usage.output;
+				cost += usage.cost.total;
+			};
+
+			for (const entry of ctx.sessionManager.getEntries()) {
+				if (entry.type === "message" && entry.message.role === "assistant") {
+					const usage = entry.message.usage;
+					const prompt = usage.input + usage.cacheRead + usage.cacheWrite;
+					cacheRate = prompt ? usage.cacheRead / prompt * 100 : 0;
+					add(usage);
+				} else if (entry.type === "message" && entry.message.role === "toolResult") {
+					add(entry.message.usage);
+				} else if (entry.type === "branch_summary" || entry.type === "compaction") {
+					add(entry.usage);
+				}
+			}
+		};
+
 		ctx.ui.setFooter((tui, theme, data) => {
 			const unsubscribe = data.onBranchChange(() => tui.requestRender());
 			return {
 				dispose: unsubscribe,
 				invalidate() { },
 				render(width: number): string[] {
-					let input = 0;
-					let output = 0;
-					let cost = 0;
-					let cacheRate: number | undefined;
-					const add = (usage: Usage | undefined) => {
-						if (!usage) return;
-						input += usage.input;
-						output += usage.output;
-						cost += usage.cost.total;
-					};
-
-					for (const entry of ctx.sessionManager.getEntries()) {
-						if (entry.type === "message" && entry.message.role === "assistant") {
-							const usage = entry.message.usage;
-							const prompt = usage.input + usage.cacheRead + usage.cacheWrite;
-							cacheRate = prompt ? usage.cacheRead / prompt * 100 : 0;
-							add(usage);
-						} else if (entry.type === "message" && entry.message.role === "toolResult") {
-							add(entry.message.usage);
-						} else if (entry.type === "branch_summary" || entry.type === "compaction") {
-							add(entry.usage);
-						}
+					const entries = ctx.sessionManager.getEntries();
+					const key = `${entries.length}:${entries.at(-1)?.type}`;
+					if (key !== usageKey) {
+						usageKey = key;
+						computeUsage();
 					}
 
 					const branch = data.getGitBranch()?.replace(/^worktree\//, "");
@@ -100,7 +128,8 @@ export default function footerExtension(pi: ExtensionAPI): void {
 						`↑ ${formatTokens(input)}`,
 						`↓ ${formatTokens(output)}`,
 						`↺ ${cacheRate === undefined ? "—" : `${cacheRate.toFixed(1)}%`}`,
-						`$ ${cost.toFixed(3)}`,
+						`⚡ ${tps === undefined ? "—" : `${tps.toFixed(1)} t/s`}`,
+					`$ ${cost.toFixed(3)}`,
 						`◔ ${context == null ? "—" : `${context.toFixed(1)}%`}`,
 					].join(" · "));
 					const thinkingText = thinking === "ultra"

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@henryqw/pi-footer",
-  "version": "0.2.1",
+  "version": "0.3.0",
   "description": "Show concise repository, branch, and usage details in the Pi footer.",
   "keywords": [
     "pi-package",