@henryqw/pi-footer 0.2.2 → 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,6 +53,19 @@ 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;
@@ -115,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.2",
+  "version": "0.3.0",
   "description": "Show concise repository, branch, and usage details in the Pi footer.",
   "keywords": [
     "pi-package",