@qmxme/pi-stats 0.2.0

package/README.md

@@ -0,0 +1,55 @@
+# @qmxme/pi-stats
+
+Stats widget extension for [pi](https://github.com/badlogic/pi) that displays token usage, throughput, and response duration.
+
+## Features
+
+- **Token throughput**: Tokens per second during response
+- **Token usage**: Input → Output tokens per message
+- **Cache tracking**: Cache read (↓) and write (↑) when applicable
+- **Total tokens**: Cumulative token count
+- **Duration**: Response time in seconds or minutes
+
+## Widget Display
+
+Example output in the UI status bar:
+
+```
+12.3 tok/s | 1.2k → 450 | cache 500↓ 300↑ | total 2.1k | 3.2s
+```
+
+## Installation
+
+```bash
+pi install npm:@qmxme/pi-stats
+```
+
+With a pinned version:
+
+```bash
+pi install npm:@qmxme/pi-stats@0.1.0
+```
+
+Project-local installation:
+
+```bash
+pi install npm:@qmxme/pi-stats -l
+```
+
+Try without installing:
+
+```bash
+pi -e npm:@qmxme/pi-stats
+```
+
+## Development
+
+```bash
+npm install      # install build dependencies
+npm run build    # compile to dist/
+npm run dev      # watch mode
+```
+
+## License
+
+MIT

package/package.json

@@ -0,0 +1,41 @@
+{
+  "name": "@qmxme/pi-stats",
+  "version": "0.2.0",
+  "description": "Stats widget extension for pi - shows token throughput, usage, and duration",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/qmx/pi-stats"
+  },
+  "type": "module",
+  "main": "./stats.ts",
+  "types": "./stats.ts",
+  "files": [
+    "stats.ts",
+    "README.md"
+  ],
+  "keywords": [
+    "pi",
+    "pi-package",
+    "extension",
+    "stats"
+  ],
+  "pi": {
+    "extensions": [
+      "./stats.ts"
+    ]
+  },
+  "scripts": {
+    "typecheck": "tsc --noEmit",
+    "dev": "tsc --noEmit --watch",
+    "prepublishOnly": "npm run typecheck"
+  },
+  "peerDependencies": {
+    "@mariozechner/pi-coding-agent": "*"
+  },
+  "devDependencies": {
+    "@mariozechner/pi-coding-agent": "^0.63.0",
+    "@types/node": "^22.10.7",
+    "typescript": "^5.7.3"
+  },
+  "license": "MIT"
+}

package/stats.ts

@@ -0,0 +1,65 @@
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+
+export default function (pi: ExtensionAPI) {
+	let streamingStart: number | undefined;
+	let initialOutputTokens: number = 0;
+	let widgetKey = "streaming-stats";
+
+	// Format numbers with K/M suffix
+	function formatNum(n: number): string {
+		if (n >= 1_000_000) return `${(n / 1_000_000).toFixed(1)}M`;
+		if (n >= 1_000) return `${(n / 1_000).toFixed(1)}k`;
+		return n.toString();
+	}
+
+	// Format duration
+	function formatDuration(ms: number): string {
+		const s = ms / 1000;
+		if (s < 60) return `${s.toFixed(1)}s`;
+		const m = Math.floor(s / 60);
+		const rem = (s % 60).toFixed(1);
+		return `${m}m${rem}s`;
+	}
+
+	pi.on("message_start", async (event, ctx) => {
+		if (event.message.role === "assistant") {
+			streamingStart = Date.now();
+			initialOutputTokens = event.message.usage.output;
+			ctx.ui.setWidget(widgetKey, undefined);
+		}
+	});
+
+	pi.on("message_end", async (event, ctx) => {
+		if (event.message.role === "assistant" && streamingStart) {
+			const elapsed = Date.now() - streamingStart;
+			const usage = event.message.usage;
+
+			// Calculate total output tokens generated in this turn
+			const outputTokens = usage.output - initialOutputTokens;
+			const tps = elapsed > 0 ? (outputTokens / elapsed) * 1000 : 0;
+
+			const total = usage.output + usage.input + usage.cacheRead + usage.cacheWrite;
+
+			// Build cache string, only showing non-zero values
+			let cacheStr = "";
+			const cacheParts: string[] = [];
+			if (usage.cacheRead > 0) cacheParts.push(`${formatNum(usage.cacheRead)}↓`);
+			if (usage.cacheWrite > 0) cacheParts.push(`${formatNum(usage.cacheWrite)}↑`);
+			if (cacheParts.length > 0) {
+				cacheStr = ` | cache ${cacheParts.join(" ")}`;
+			}
+
+			const stats = `${tps.toFixed(1)} tok/s | ${formatNum(usage.input)} → ${formatNum(usage.output)}${cacheStr} | total ${formatNum(total)} | ${formatDuration(elapsed)}`;
+
+			ctx.ui.setWidget(widgetKey, (tui, theme) => {
+				const lines = [theme.fg("dim", stats)];
+				return {
+					render: () => lines,
+					invalidate: () => {},
+				};
+			});
+
+			streamingStart = undefined;
+		}
+	});
+}