@firstpick/pi-extension-git-footer-status 0.1.9 → 0.2.1

package/README.md

@@ -8,7 +8,7 @@ Enhanced Pi footer with git health and model/token telemetry.
 
 - Shows compact runtime metrics in the footer:
   - input/output/cache tokens
-  - prompt-injection estimate (`PI: X tok`, compacted as `k` for thousands)
+  - export-backed initial prompt estimate (`PI: X tok`, same estimator as `/stats-pi`, compacted as `k` for thousands; falls back to live context data if Pi HTML export is unavailable)
   - live output token counter + token output speed (`tok/s`) measured from assistant streaming lifecycle events, with a session-history fallback
   - cost + context-window usage
   - current model and reasoning level

package/index.ts

@@ -3,7 +3,13 @@ import { homedir } from "node:os";
 import { isAbsolute, resolve, sep } from "node:path";
 import type { AssistantMessage } from "@earendil-works/pi-ai";
 import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
-import { formatTokens, estimatePromptInjectionTokens, estimateTokensFromCharCount } from "@firstpick/pi-utils";
+import {
+  collectInitialPromptCalibration,
+  estimateInitialPromptForPiContext,
+  estimateInitialPromptFromPiExport,
+  estimateTokensFromCharCount,
+  formatTokens,
+} from "@firstpick/pi-utils";
 import { truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
 
 type GitSnapshot = {
@@ -90,6 +96,11 @@ type LiveTokenSample = {
   tokens: number;
 };
 
+type PromptEstimateCache = {
+  key: string;
+  tokens: number;
+};
+
 function formatTokenSpeed(tokensPerSecond: number): string {
   if (tokensPerSecond < 100) {
     if (tokensPerSecond >= 10) return tokensPerSecond.toFixed(1);
@@ -355,6 +366,57 @@ export default function gitFooterStatus(pi: ExtensionAPI) {
   let currentAssistantLiveTokenSpeed: number | null = null;
   let currentAssistantTokenSamples: LiveTokenSample[] = [];
   let latestMeasuredTokenSpeed: number | null = null;
+  let promptEstimateCache: PromptEstimateCache | null = null;
+  let promptEstimateRequestId = 0;
+  let requestFooterRender: (() => void) | null = null;
+
+  const getPromptCalibration = (ctx: ExtensionContext) => collectInitialPromptCalibration(ctx.sessionManager.getSessionDir());
+
+  const getFallbackPromptEstimate = (ctx: ExtensionContext) => {
+    const calibration = getPromptCalibration(ctx);
+    const estimate = estimateInitialPromptForPiContext(pi, ctx.getSystemPrompt(), calibration);
+    const key = [
+      estimate.uncalibratedTotal,
+      estimate.promptText,
+      estimate.toolSchemas,
+      estimate.framing,
+      estimate.toolCount,
+      estimate.calibrationMultiplier,
+      estimate.calibrationSamples,
+      estimate.low,
+      estimate.high,
+    ].join(":");
+    return { calibration, estimate, key };
+  };
+
+  const refreshPromptInjectionEstimate = async (ctx: ExtensionContext) => {
+    const requestId = ++promptEstimateRequestId;
+    const fallback = getFallbackPromptEstimate(ctx);
+    promptEstimateCache = { key: fallback.key, tokens: fallback.estimate.total };
+    requestFooterRender?.();
+
+    try {
+      const result = await estimateInitialPromptFromPiExport(pi, ctx, fallback.calibration);
+      const latestFallback = getFallbackPromptEstimate(ctx);
+      if (requestId !== promptEstimateRequestId || latestFallback.key !== fallback.key) return;
+      promptEstimateCache = { key: fallback.key, tokens: result.estimate.total };
+    } catch {
+      if (requestId !== promptEstimateRequestId) return;
+      const latestFallback = getFallbackPromptEstimate(ctx);
+      promptEstimateCache = { key: latestFallback.key, tokens: latestFallback.estimate.total };
+    } finally {
+      if (requestId === promptEstimateRequestId) requestFooterRender?.();
+    }
+  };
+
+  const getFooterPromptInjectionTokens = (ctx: ExtensionContext): number => {
+    const fallback = getFallbackPromptEstimate(ctx);
+    if (!promptEstimateCache || promptEstimateCache.key !== fallback.key) {
+      promptEstimateCache = { key: fallback.key, tokens: fallback.estimate.total };
+      void refreshPromptInjectionEstimate(ctx);
+    }
+    return promptEstimateCache.tokens;
+  };
 
   const recordAssistantSpeed = (message: AssistantMessage, endMs = Date.now()): boolean => {
     const outputTokens = message.usage?.output ?? 0;
@@ -413,11 +475,19 @@ export default function gitFooterStatus(pi: ExtensionAPI) {
   };
 
   pi.on("session_start", async (_event, ctx) => {
+    promptEstimateCache = null;
+    void refreshPromptInjectionEstimate(ctx);
+
     ctx.ui.setFooter((tui, theme, footerData) => {
-      const unsub = footerData.onBranchChange(() => tui.requestRender());
+      const render = () => tui.requestRender();
+      requestFooterRender = render;
+      const unsub = footerData.onBranchChange(render);
 
       return {
-        dispose: unsub,
+        dispose() {
+          unsub();
+          if (requestFooterRender === render) requestFooterRender = null;
+        },
         invalidate() {},
         render(width: number): string[] {
           let totalInput = 0;
@@ -483,7 +553,7 @@ export default function gitFooterStatus(pi: ExtensionAPI) {
             latestTokenSpeed = historicalTokenSpeed;
           }
 
-          const promptInjectionTokens = estimatePromptInjectionTokens(ctx.getSystemPrompt());
+          const currentPromptInjectionTokens = getFooterPromptInjectionTokens(ctx);
 
           const contextUsage = ctx.getContextUsage();
           const contextWindow = contextUsage?.contextWindow ?? ctx.model?.contextWindow ?? 0;
@@ -521,7 +591,7 @@ export default function gitFooterStatus(pi: ExtensionAPI) {
           const segments: string[] = [];
           if (ioItems.length > 0) segments.push(`${theme.fg("muted", "🪙")} ${ioItems.join(` ${itemSep} `)}`);
           if (cacheItems.length > 0) segments.push(`${theme.fg("muted", "💾")} ${cacheItems.join(` ${itemSep} `)}`);
-          segments.push(`PI: ${formatTokens(promptInjectionTokens)} tok`);
+          segments.push(`PI: ${formatTokens(currentPromptInjectionTokens)} tok`);
           if (latestTokenSpeed !== null) {
             const livePrefix = liveOutputTokens > 0 ? `${formatTokens(liveOutputTokens)} tok @ ` : "";
             segments.push(`⚡ ${livePrefix}${formatTokenSpeed(latestTokenSpeed)} tok/s`);
@@ -606,6 +676,10 @@ export default function gitFooterStatus(pi: ExtensionAPI) {
     await refresh(ctx);
   });
 
+  pi.on("agent_start", async (_event, ctx) => {
+    void refreshPromptInjectionEstimate(ctx);
+  });
+
   pi.on("message_start", (event) => {
     if (event.message.role === "assistant") {
       currentAssistantStartMs = Date.now();
@@ -656,6 +730,7 @@ export default function gitFooterStatus(pi: ExtensionAPI) {
       recordAssistantSpeed(event.message as AssistantMessage);
       resetLiveAssistantState();
     }
+    void refreshPromptInjectionEstimate(ctx);
     await refresh(ctx);
   });
 
@@ -667,6 +742,7 @@ export default function gitFooterStatus(pi: ExtensionAPI) {
   pi.registerCommand("git-footer-refresh", {
     description: "Refresh git footer information",
     handler: async (_args, ctx) => {
+      await refreshPromptInjectionEstimate(ctx);
       await refresh(ctx);
       ctx.ui.notify("Git footer refreshed", "info");
     },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@firstpick/pi-extension-git-footer-status",
-  "version": "0.1.9",
+  "version": "0.2.1",
   "description": "Enhanced Pi footer with git status, token usage, context usage, and model telemetry.",
   "license": "MIT",
   "keywords": [