@bacnh85/pi-sub 0.1.10 → 0.1.12

package/README.md

@@ -34,25 +34,29 @@ The footer status appears after Pi's built-in status/token usage line and includ
 Example subscription line:
 
 ```text
-(user@example.com) R:15%/2H W:20%/3D
+(user@example.com) R:15%/2H W:20%/3D 42 tok/s
 ```
 
 ### OpenCode Go
 
-OpenCode Go does not expose a public usage-window API, so the footer shows the active account/key label and accumulated session cost:
+OpenCode Go does not expose a public usage-window API, so the footer shows the active account/key label, accumulated session cost, and last response speed:
 
 ```text
-OpenCode Go (OpenCode Go key#1a2b3c4d) $0.23
+OpenCode Go (OpenCode Go key#1a2b3c4d) $0.23 42 tok/s
 ```
 
 ### Z.ai
 
-Z.ai (GLM Coding Plan) shows the active account/key label plus 5-hour rolling and weekly remaining quota with reset countdowns:
+Z.ai (GLM Coding Plan) shows the active account/key label, 5-hour rolling and weekly remaining quota with reset countdowns, and last response speed:
 
 ```text
-(Z.ai key#1a2b3c4d) R:55%/2H W:80%/3D
+(Z.ai key#1a2b3c4d) R:55%/2H W:80%/3D 42 tok/s
 ```
 
+### Tokens per second
+
+`pi-sub` tracks each response's tokens-per-second (tok/s) speed by measuring the time from provider request to message completion against the response's output token count. The last response's speed is shown in the footer next to usage data. The `/sub` detail view shows both the last response speed and the session-wide average.
+
 When the current model provider is not supported, `pi-sub` clears its subscription line and does not refresh subscription data.
 
 ## Commands
@@ -63,18 +67,23 @@ When the current model provider is not supported, `pi-sub` clears its subscripti
 | `/sub status` | Same as `/sub`. |
 | `/sub refresh` | Force a usage refresh, then show details. |
 
-When Pi OpenAI Codex auth is available, `/sub` shows the active account usage:
+When Pi OpenAI Codex auth is available, `/sub` shows the active account usage and speed:
 
 ```text
+Provider: Codex · Model: o4-mini · Fetched: 14:23
+Session cost: $0.12
+Last response: 42 tok/s · Session avg: 39 tok/s
+
 ACCOUNT                 PLAN  ROLLING  WEEKLY  LAST ACTIVITY
 * user@example.com     Plus  15%/2H   20%/3D  Now
 ```
 
-For OpenCode Go, `/sub` shows the provider/model, active account/key label, and session cost:
+For OpenCode Go, `/sub` shows the provider/model, active account/key label, session cost, and speed:
 
 ```text
 Provider: OpenCode Go · Model: kimi-k2.6 · Fetched: 14:23
 Session cost: $0.23
+Last response: 42 tok/s · Session avg: 39 tok/s
 
   ACCOUNT                         PLAN  LAST ACTIVITY
 ------------------------------------------------------
@@ -83,10 +92,12 @@ Session cost: $0.23
 OpenCode Go does not expose usage windows.
 ```
 
-For Z.ai, `/sub` shows the rolling and weekly quota windows:
+For Z.ai, `/sub` shows the rolling and weekly quota windows and speed:
 
 ```text
 Provider: Z.ai · Model: glm-5.2 · Fetched: 14:23
+Session cost: $0.05
+Last response: 42 tok/s · Session avg: 39 tok/s
 
   ACCOUNT             PLAN  ROLLING        WEEKLY             LAST ACTIVITY
   ------------------------------------------------------------------------

package/extensions/index.ts

@@ -1,5 +1,4 @@
 import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
-import { Box, Text } from "@earendil-works/pi-tui";
 import { createHash } from "node:crypto";
 import fs from "node:fs/promises";
 import os from "node:os";
@@ -86,6 +85,10 @@ interface State {
 	inFlight?: Promise<SubscriptionUsageSnapshot>;
 	refreshTimer?: NodeJS.Timeout;
 	debounceTimer?: NodeJS.Timeout;
+	responseStartTime?: number;
+	lastTokPerSec?: number;
+	cumulativeOutput: number;
+	cumulativeDurationMs: number;
 }
 
 function isCodexModel(model: ModelLike): boolean {
@@ -532,6 +535,7 @@ function renderSubscriptionLine(ctx: ExtensionContext, state: State): void {
 		const cost = snapshot.cost;
 		const hasWindows = windowParts.length > 0;
 		if (cost !== undefined && cost > 0) segments.push(`$${cost.toFixed(2)}`);
+		if (state.lastTokPerSec !== undefined) segments.push(`${state.lastTokPerSec} tok/s`);
 		if (segments.length === 0) {
 			line = `Sub ${state.adapter.displayName}`;
 		} else if (!hasWindows) {
@@ -657,7 +661,13 @@ function buildDetails(snapshot: SubscriptionUsageSnapshot | undefined, state: St
 	});
 
 	const costLine = snapshot.cost !== undefined ? `\nSession cost: $${snapshot.cost.toFixed(2)}` : "";
-	const lines = [`Provider: ${snapshot.providerDisplayName} · Model: ${state.model?.id ?? "unknown-model"} · Fetched: ${new Date(snapshot.fetchedAt).toLocaleTimeString()}${costLine}`, "", header, sep, ...body];
+	const tokPerSecLine = state.lastTokPerSec !== undefined
+		? `\nLast response: ${state.lastTokPerSec} tok/s` +
+			(state.cumulativeDurationMs > 0
+				? ` · Session avg: ${Math.round(state.cumulativeOutput / (state.cumulativeDurationMs / 1000))} tok/s`
+				: "")
+		: "";
+	const lines = [`Provider: ${snapshot.providerDisplayName} · Model: ${state.model?.id ?? "unknown-model"} · Fetched: ${new Date(snapshot.fetchedAt).toLocaleTimeString()}${costLine}${tokPerSecLine}`, "", header, sep, ...body];
 	if (!hasFiveHour && !hasWeekly) {
 		lines.push("", `${snapshot.providerDisplayName} does not expose usage windows.`);
 	}
@@ -665,13 +675,7 @@ function buildDetails(snapshot: SubscriptionUsageSnapshot | undefined, state: St
 }
 
 export default function (pi: ExtensionAPI) {
-	const state: State = { lastRefreshAt: 0, refreshGeneration: 0 };
-
-	pi.registerMessageRenderer(MESSAGE_TYPE, (message, _options, theme) => {
-		const box = new Box(1, 1, (text) => theme.bg("customMessageBg", text));
-		box.addChild(new Text(String(message.content ?? ""), 0, 0));
-		return box;
-	});
+	const state: State = { lastRefreshAt: 0, refreshGeneration: 0, cumulativeOutput: 0, cumulativeDurationMs: 0 };
 
 	pi.on("session_start", async (_event, ctx) => {
 		updateActiveAdapter(ctx, state, ctx.model);
@@ -683,6 +687,24 @@ export default function (pi: ExtensionAPI) {
 		if (state.adapter) void refreshUsage(ctx, state, true);
 	});
 
+	pi.on("before_provider_request", async (_event, _ctx) => {
+		state.responseStartTime = Date.now();
+	});
+
+	pi.on("message_end", async (event, ctx) => {
+		if (event.message.role === "assistant" && state.responseStartTime) {
+			const output = (event.message.usage as any)?.output ?? 0;
+			const elapsed = Date.now() - state.responseStartTime;
+			state.responseStartTime = undefined;
+			if (elapsed > 0 && output > 0) {
+				state.lastTokPerSec = Math.round(output / (elapsed / 1000));
+				state.cumulativeOutput += output;
+				state.cumulativeDurationMs += elapsed;
+				if (state.adapter) renderSubscriptionLine(ctx, state);
+			}
+		}
+	});
+
 	pi.on("after_provider_response", async (_event, ctx) => {
 		if (state.adapter) scheduleRefresh(ctx, state);
 	});

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bacnh85/pi-sub",
-  "version": "0.1.10",
+  "version": "0.1.12",
   "description": "Pi extension showing subscription usage for supported model providers.",
   "type": "module",
   "license": "MIT",
@@ -26,7 +26,8 @@
   ],
   "files": [
     "README.md",
-    "extensions/"
+    "extensions/index.ts",
+    "extensions/package.json"
   ],
   "pi": {
     "extensions": [
@@ -37,7 +38,6 @@
     "node": ">=20.3.0"
   },
   "peerDependencies": {
-    "@earendil-works/pi-coding-agent": "*",
-    "@earendil-works/pi-tui": "*"
+    "@earendil-works/pi-coding-agent": "*"
   }
 }