bermudis-pi-goodies 0.1.0 → 0.2.0

package/README.md

@@ -1,7 +1,7 @@
 # bermudis-pi-goodies
 
 A bundle of small, frequently-used [Pi](https://github.com/earendil-works/pi)
-extensions. One entry point, eight independent features.
+extensions. One entry point, nine independent features.
 
 | Feature | Command / hook | What it does |
 |---------|----------------|--------------|
@@ -13,13 +13,14 @@ extensions. One entry point, eight independent features.
 | `fixed-defaults` | hook + `/fixed-defaults` | Keep the global startup provider, model, and thinking defaults fixed while allowing in-session model changes; `/fixed-defaults set` pins the current model and thinking level. |
 | `kilo` | provider | Access Kilo Gateway models via `/login kilo` or `KILO_API_KEY`. |
 | `provider-balance` | footer (no command) | Show remaining Kilo or OpenRouter credits, z.ai token-plan quota, or OpenAI Codex quota on the right side of the working-directory footer line. |
+| `tps` | hook (no command) | Notify tokens/sec and in/out/cache token usage at the end of each agent turn. |
 
 ## Install
 
 After publishing the package to npm:
 
 ```bash
-pi install npm:bermudis-pi-goodies@0.1.0
+pi install npm:bermudis-pi-goodies@0.2.0
 ```
 
 Remove any old `bermudis-pi-goodies.ts` symlink before reloading Pi. Each

package/index.ts

@@ -11,6 +11,7 @@
  *   - fixed-defaults   hook + command    keep startup defaults stable; /fixed-defaults set pins the current model
  *   - kilo             provider          access Kilo Gateway models
  *   - provider-balance footer            show Kilo credits or Codex quota in the footer
+ *   - tps              hook              notify tokens/sec and usage at each agent turn end
  */
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import copyWithModel from "./copy-with-model.ts";
@@ -18,6 +19,7 @@ import fixedDefaults from "./fixed-defaults.ts";
 import kilo from "./kilo.ts";
 import modelThinking from "./model-thinking.ts";
 import providerBalance from "./provider-balance.ts";
+import tps from "./tps.ts";
 import nameWithAi from "./name-with-ai.ts";
 import zed from "./zed.ts";
 import preferTools from "./prefer-tools.ts";
@@ -31,4 +33,5 @@ export default function bermudisPiGoodies(pi: ExtensionAPI): void {
   fixedDefaults(pi);
   providerBalance(pi);
   kilo(pi);
+  tps(pi);
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "bermudis-pi-goodies",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "private": false,
   "description": "A bundle of small, frequently-used Pi extensions.",
   "keywords": ["pi-package"],

package/tps.ts

@@ -0,0 +1,47 @@
+import type { AssistantMessage } from "@earendil-works/pi-ai";
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+
+function isAssistantMessage(message: unknown): message is AssistantMessage {
+  if (!message || typeof message !== "object") return false;
+  const role = (message as { role?: unknown }).role;
+  return role === "assistant";
+}
+
+export default function (pi: ExtensionAPI) {
+  let agentStartMs: number | null = null;
+
+  pi.on("agent_start", () => {
+    agentStartMs = Date.now();
+  });
+
+  pi.on("agent_end", (event, ctx) => {
+    if (!ctx.hasUI) return;
+    if (agentStartMs === null) return;
+
+    const elapsedMs = Date.now() - agentStartMs;
+    agentStartMs = null;
+    if (elapsedMs <= 0) return;
+
+    let input = 0;
+    let output = 0;
+    let cacheRead = 0;
+    let cacheWrite = 0;
+    let totalTokens = 0;
+
+    for (const message of event.messages) {
+      if (!isAssistantMessage(message)) continue;
+      input += message.usage.input || 0;
+      output += message.usage.output || 0;
+      cacheRead += message.usage.cacheRead || 0;
+      cacheWrite += message.usage.cacheWrite || 0;
+      totalTokens += message.usage.totalTokens || 0;
+    }
+
+    if (output <= 0) return;
+
+    const elapsedSeconds = elapsedMs / 1000;
+    const tokensPerSecond = output / elapsedSeconds;
+    const message = `TPS ${tokensPerSecond.toFixed(1)} tok/s. out ${output.toLocaleString()}, in ${input.toLocaleString()}, cache r/w ${cacheRead.toLocaleString()}/${cacheWrite.toLocaleString()}, total ${totalTokens.toLocaleString()}, ${elapsedSeconds.toFixed(1)}s`;
+    ctx.ui.notify(message, "info");
+  });
+}