useful-pi-extensions 1.1.0 → 1.1.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "useful-pi-extensions",
-  "version": "1.1.0",
+  "version": "1.1.2",
   "description": "A small collection of pi extensions, installed with one command — a labelled status line with context pressure, cache, cost, effort, TTFT and tokens/sec.",
   "keywords": [
     "bun",

package/packages/statusline/extensions/statusline/index.ts

@@ -18,6 +18,7 @@
  * stepStartTime; decodeMs = completedTime - firstTokenTime; tok/s = usage.output / (decodeMs / 1000)
  */
 
+import { readFile, stat } from 'node:fs/promises'
 import { homedir } from 'node:os'
 import { join } from 'node:path'
 
@@ -38,6 +39,7 @@ import {
   pair,
   row,
   shortenPath,
+  ttftMs,
   USD,
   type Currency,
 } from './render.ts'
@@ -58,8 +60,19 @@ const CONFIG_PATH = join(homedir(), '.pi', 'agent', 'statusline.json')
  * bug. Editing the file therefore takes effect on `/reload`.
  */
 async function loadCurrency(notify: (message: string) => void): Promise<Currency> {
-  const file = Bun.file(CONFIG_PATH)
-  const text = (await file.exists()) ? await file.text() : null
+  try {
+    await stat(CONFIG_PATH)
+  } catch {
+    // No config file is the normal case, and it means USD.
+    return USD
+  }
+  let text: string
+  try {
+    text = await readFile(CONFIG_PATH, 'utf8')
+  } catch {
+    notify('statusline.json exists but could not be read, showing USD')
+    return USD
+  }
   const { currency, problem } = currencyFromConfig(text)
   if (problem !== null) notify(problem)
   return currency
@@ -173,7 +186,7 @@ export default function (pi: ExtensionAPI) {
   let reading: { rate: number; exact: boolean; ttftMs: number | null } | null = null
 
   let chars = 0
-  let stepStartedAt = 0
+  let requestAt: number | null = null
   let firstTokenAt: number | null = null
   let windowAt = 0
   let windowTokens = 0
@@ -187,8 +200,8 @@ export default function (pi: ExtensionAPI) {
     renderedAt = 0
   }
 
-  function publish(rate: number, exact: boolean, ttftMs: number | null): void {
-    reading = { rate, exact, ttftMs }
+  function publish(rate: number, exact: boolean, ttft: number | null): void {
+    reading = { rate, exact, ttftMs: ttft }
     requestRender?.()
   }
 
@@ -274,6 +287,7 @@ export default function (pi: ExtensionAPI) {
   pi.on('session_start', async (_event, ctx) => {
     ratio = seedRatio(ctx)
     reading = null
+    requestAt = null
     resetStream()
     currency = await loadCurrency((message) => ctx.ui.notify(message, 'warning'))
     installFooter(ctx)
@@ -282,7 +296,17 @@ export default function (pi: ExtensionAPI) {
   pi.on('message_start', async (event) => {
     if (event.message.role !== 'assistant') return
     resetStream()
-    stepStartedAt = Date.now()
+  })
+
+  // TTFT's request anchor, from the agent loop in pi's docs: turn_start opens the LLM call and
+  // before_provider_request is the last thing pi does before the wire. message_start is not that
+  // moment, and anchoring there made a first token that arrived early read as a clamped 0ms.
+  pi.on('turn_start', async () => {
+    requestAt = null
+  })
+
+  pi.on('before_provider_request', async () => {
+    requestAt = Date.now()
   })
 
   pi.on('message_update', async (event) => {
@@ -314,7 +338,7 @@ export default function (pi: ExtensionAPI) {
     const rate =
       windowMs > 0 ? ((tokens - windowTokens) / windowMs) * 1000 : (tokens / decodeMs) * 1000
     renderedAt = now
-    publish(rate, false, firstTokenAt - stepStartedAt)
+    publish(rate, false, ttftMs(requestAt, firstTokenAt))
   })
 
   pi.on('message_end', async (event) => {
@@ -329,10 +353,10 @@ export default function (pi: ExtensionAPI) {
     }
 
     const decodeMs = firstTokenAt !== null ? Date.now() - firstTokenAt : 0
-    const ttftMs = firstTokenAt !== null ? firstTokenAt - stepStartedAt : null
+    const measured = ttftMs(requestAt, firstTokenAt)
     const tokens = output > 0 ? output : totalChars * (ratio ?? FALLBACK_TOKENS_PER_CHAR)
-    if (ttftMs !== null && decodeMs >= MIN_SAMPLE_MS)
-      publish((tokens / decodeMs) * 1000, output > 0, ttftMs)
+    if (measured !== null && decodeMs >= MIN_SAMPLE_MS)
+      publish((tokens / decodeMs) * 1000, output > 0, measured)
     resetStream()
   })
 }

package/packages/statusline/extensions/statusline/render.ts

@@ -66,6 +66,21 @@ export function formatLatency(ms: number): string {
   return `${seconds < 10 ? Math.round(seconds * 10) / 10 : Math.round(seconds)}s`
 }
 
+/**
+ * TTFT from its two wall-clock anchors: the moment the request was dispatched, and the first
+ * streamed delta.
+ *
+ * The guard is the point. Both anchors come from extension events whose ordering around a stream is
+ * not guaranteed, and a first token that appears to precede its own request yields a negative that
+ * `formatLatency` would clamp to `0ms` — a confident fake where an honest absence belongs.
+ *
+ * @returns The milliseconds between the two, or null when they do not line up.
+ */
+export function ttftMs(requestAt: number | null, firstTokenAt: number | null): number | null {
+  if (requestAt === null || firstTokenAt === null) return null
+  return firstTokenAt > requestAt ? firstTokenAt - requestAt : null
+}
+
 /** Home-relative path, or the absolute path when it is outside the home directory. */
 export function formatCwd(cwd: string): string {
   const home = homedir()