@marshal/pi-turn-stats 0.1.6 → 0.1.8

package/README.md

@@ -7,10 +7,13 @@ A pi extension that automatically tracks per-exchange duration, token usage (inp
 - **Stats card in conversation stream** — After each reply, a stats card is appended to the stream (does not enter LLM context).
 - **Real-time status bar** — The bottom status bar shows the last exchange's duration, throughput, token count, and cost.
 - **`/turnstats` command** — Appends a session cumulative stats card (total exchanges, total LLM calls, total tokens, total cost).
+- **Auto i18n** — UI labels automatically switch between Chinese and English based on your system locale (`LANG` / `LC_ALL` / `Intl`).
 
 ## Demo
 
-![Turn Stats Demo](https://github.com/MarshalW/pi-turn-stats/raw/main/turn-stats.gif)
+**English locale (`LANG=en`)**
+
+![Turn Stats Demo](https://github.com/MarshalW/pi-turn-stats/raw/main/turn-stats_en.mp4)
 
 ## Screenshots
 
@@ -28,22 +31,22 @@ A pi extension that automatically tracks per-exchange duration, token usage (inp
 
 ```bash
 # Install directly with pi
-pi install -l npm:@marshal/pi-turn-stats
+pi install npm:@marshal/pi-turn-stats
 ```
 
 Or standard npm install:
 
 ```bash
-npm install @marshal/pi-turn-stats
+npm install -g @marshal/pi-turn-stats
 ```
 
-> Note: `pi install -l npm:...` is the pi-recommended local install method that registers the extension for the current project. Plain `npm install` only downloads the package without registering it as a pi extension.
+> Note: `pi install npm:...` registers the extension globally. Plain `npm install -g` only downloads the package without registering it as a pi extension.
 
 **Alternative**: Install from GitHub (for source access or custom modifications).
 
 ```bash
-# Install from GitHub repo, specifying tag (SSH)
-pi install -l git:git@github.com:MarshalW/pi-turn-stats@v0.1.1
+# Install from GitHub repo (SSH)
+pi install git:git@github.com:MarshalW/pi-turn-stats
 ```
 
 > Users in China are encouraged to use the npm method, as GitHub access may be unreliable.
@@ -66,6 +69,6 @@ pi install ./        # local install for testing
 git tag vX.Y.Z && git push origin main --tags
 # Publish to npm: npm version patch && npm publish --access public
 # Consumer install:
-#   npm (recommended): pi install -l npm:@marshal/pi-turn-stats
-#   git (fallback):    pi install -l git:git@github.com:MarshalW/pi-turn-stats@vX.Y.Z
+#   npm (recommended): pi install npm:@marshal/pi-turn-stats
+#   git (fallback):    pi install git:git@github.com:MarshalW/pi-turn-stats@vX.Y.Z
 ```

package/extensions/turn-stats.ts

@@ -164,6 +164,25 @@ function fmtCost(c: number): string {
 	return `$${c.toFixed(4)}`;
 }
 
+/**
+ * After session replacement (e.g. /new) or reload, pi invalidates the old
+ * extension runner, but a run's `finally` can still deliver agent_settled to
+ * it; any ctx/`pi` access on the stale runner throws. Treat those as
+ * "no UI / no session" and skip side effects instead of erroring.
+ */
+function isStaleCtxError(err: unknown): boolean {
+	return err instanceof Error && err.message.includes("extension ctx is stale");
+}
+
+/** Run a side effect against ctx/`pi`; silently skip when the ctx is stale. */
+function withStaleGuard(fn: () => void): void {
+	try {
+		fn();
+	} catch (err) {
+		if (!isStaleCtxError(err)) throw err;
+	}
+}
+
 export default function (pi: ExtensionAPI) {
 	// ---- session cumulative stats ----
 	const sessionTotals = {
@@ -248,8 +267,10 @@ export default function (pi: ExtensionAPI) {
 
 	// ===== Session start: init status bar =====
 	pi.on("session_start", (_event, ctx) => {
-		if (!ctx.hasUI) return;
-		ctx.ui.setStatus("turn-stats", ctx.ui.theme.fg("dim", t("statusWaiting")));
+		withStaleGuard(() => {
+			if (!ctx.hasUI) return;
+			ctx.ui.setStatus("turn-stats", ctx.ui.theme.fg("dim", t("statusWaiting")));
+		});
 	});
 
 	// ===== User submits: start timer =====
@@ -258,10 +279,12 @@ export default function (pi: ExtensionAPI) {
 		startTime = Date.now();
 		turnCount = 0;
 		accum = emptyAccum();
-		lastModel = ctx.model?.id ?? "unknown";
-		if (ctx.hasUI) {
-			ctx.ui.setStatus("turn-stats", ctx.ui.theme.fg("dim", t("statusRunning")));
-		}
+		withStaleGuard(() => {
+			lastModel = ctx.model?.id ?? "unknown";
+			if (ctx.hasUI) {
+				ctx.ui.setStatus("turn-stats", ctx.ui.theme.fg("dim", t("statusRunning")));
+			}
+		});
 	});
 
 	// ===== Each LLM turn ends: accumulate usage =====
@@ -297,27 +320,31 @@ export default function (pi: ExtensionAPI) {
 		sessionTotals.durationMs += durMs;
 
 		if (SHOW_CARD && turnCount > 0) {
-			pi.appendEntry<TurnStatsData>(ENTRY_TYPE, {
-				kind: "exchange",
-				startTime,
-				endTime,
-				turns: turnCount,
-				exchanges: 1,
-				...accum,
-				tokensPerSec: genTps,
-				model: lastModel,
+			withStaleGuard(() => {
+				pi.appendEntry<TurnStatsData>(ENTRY_TYPE, {
+					kind: "exchange",
+					startTime,
+					endTime,
+					turns: turnCount,
+					exchanges: 1,
+					...accum,
+					tokensPerSec: genTps,
+					model: lastModel,
+				});
 			});
 		}
 
-		if (ctx.hasUI) {
-			ctx.ui.setStatus(
-				"turn-stats",
-				ctx.ui.theme.fg(
-					"dim",
-					t("statusDone", fmtDuration(durMs), fmtThroughput(genTps), fmtTokens(accum.totalTokens), fmtCost(accum.cost)),
-				),
-			);
-		}
+		withStaleGuard(() => {
+			if (ctx.hasUI) {
+				ctx.ui.setStatus(
+					"turn-stats",
+					ctx.ui.theme.fg(
+						"dim",
+						t("statusDone", fmtDuration(durMs), fmtThroughput(genTps), fmtTokens(accum.totalTokens), fmtCost(accum.cost)),
+					),
+				);
+			}
+		});
 	});
 
 	// ===== /turnstats: append session cumulative stats card =====
@@ -326,20 +353,22 @@ export default function (pi: ExtensionAPI) {
 		handler: async () => {
 			const sessDurMs = sessionTotals.durationMs;
 			const sessGenTps = calcOutputPerSec(sessionTotals.output, sessDurMs);
-			pi.appendEntry<TurnStatsData>(ENTRY_TYPE, {
-				kind: "session",
-				startTime: sessDurMs > 0 ? Date.now() - sessDurMs : Date.now(),
-				endTime: Date.now(),
-				turns: sessionTotals.exchanges,
-				exchanges: sessionTotals.exchanges,
-				input: sessionTotals.input,
-				output: sessionTotals.output,
-				cacheRead: sessionTotals.cacheRead,
-				cacheWrite: sessionTotals.cacheWrite,
-				totalTokens: sessionTotals.totalTokens,
-				cost: sessionTotals.cost,
-				tokensPerSec: sessGenTps,
-				model: t("sessionModel"),
+			withStaleGuard(() => {
+				pi.appendEntry<TurnStatsData>(ENTRY_TYPE, {
+					kind: "session",
+					startTime: sessDurMs > 0 ? Date.now() - sessDurMs : Date.now(),
+					endTime: Date.now(),
+					turns: sessionTotals.exchanges,
+					exchanges: sessionTotals.exchanges,
+					input: sessionTotals.input,
+					output: sessionTotals.output,
+					cacheRead: sessionTotals.cacheRead,
+					cacheWrite: sessionTotals.cacheWrite,
+					totalTokens: sessionTotals.totalTokens,
+					cost: sessionTotals.cost,
+					tokensPerSec: sessGenTps,
+					model: t("sessionModel"),
+				});
 			});
 		},
 	});

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@marshal/pi-turn-stats",
-  "version": "0.1.6",
+  "version": "0.1.8",
   "description": "pi extension: per-exchange duration, token & cost stats for conversations (stream card + status bar + /turnstats command)",
   "keywords": [
     "pi-package"
@@ -12,11 +12,14 @@
     "README.md",
     "turn-stats.jpg",
     "turn-stats.gif",
-    "turn-stats_en.png"
+    "turn-stats_en.png",
+    "turn-stats_en.mp4"
   ],
   "pi": {
     "extensions": [
       "./extensions/turn-stats.ts"
-    ]
+    ],
+    "video": "https://github.com/MarshalW/pi-turn-stats/raw/main/turn-stats_en.mp4",
+    "image": "https://github.com/MarshalW/pi-turn-stats/raw/main/turn-stats_en.png"
   }
 }