@vtstech/pi-status 1.1.3 → 1.1.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vtstech/pi-status",
-  "version": "1.1.3",
+  "version": "1.1.4",
   "description": "System monitor / status bar extension for Pi Coding Agent",
   "main": "status.js",
   "keywords": ["pi-extensions"],
@@ -14,7 +14,7 @@
     "url": "https://github.com/VTSTech/pi-coding-agent"
   },
   "dependencies": {
-    "@vtstech/pi-shared": "1.1.3"
+    "@vtstech/pi-shared": "1.1.4"
   },
   "peerDependencies": {
     "@mariozechner/pi-coding-agent": ">=0.66"

package/status.js

@@ -1,5 +1,6 @@
 // .build-npm/status/status.temp.ts
 import * as fs from "node:fs";
+import { execSync } from "node:child_process";
 import os from "node:os";
 import { getOllamaBaseUrl, fetchModelContextLength, readModelsJson } from "@vtstech/pi-shared/ollama";
 import { fmtBytes, fmtDur } from "@vtstech/pi-shared/format";
@@ -13,6 +14,7 @@ function status_temp_default(pi) {
   let toolTimerInterval = null;
   let currentCtx = null;
   let ctxUi = null;
+  let ctxTheme = null;
   let prevCpuInfo = getCpuSnapshot();
   let lastPayload = null;
   let cpuUsage = 0;
@@ -22,10 +24,11 @@ function status_temp_default(pi) {
   let swapTotal = 0;
   let hasSwap = false;
   let footerModel = "";
-  let footerThinking = "";
   let footerNativeCtx = "";
   let nativeCtxModel = "";
   let isLocalProvider = true;
+  let versionsText = "";
+  let cachedPromptText = null;
   let securityFlashTool = "";
   let securityFlashUntil = 0;
   let activeTool = "";
@@ -121,8 +124,6 @@ function status_temp_default(pi) {
     if (payload.temperature !== void 0) params.push(`temp:${payload.temperature}`);
     if (payload.top_p !== void 0) params.push(`top_p:${payload.top_p}`);
     if (payload.top_k !== void 0) params.push(`top_k:${payload.top_k}`);
-    if (payload.max_completion_tokens !== void 0) params.push(`max:${payload.max_completion_tokens}`);
-    else if (payload.max_tokens !== void 0) params.push(`max:${payload.max_tokens}`);
     if (payload.num_predict !== void 0) params.push(`predict:${payload.num_predict}`);
     if (payload.num_ctx !== void 0) params.push(`ctx:${payload.num_ctx}`);
     if (payload.reasoning_effort !== void 0) params.push(`think:${payload.reasoning_effort}`);
@@ -130,44 +131,58 @@ function status_temp_default(pi) {
   }
   function flushStatus() {
     if (!ctxUi) return;
-    ctxUi.setStatus("status-cpu", isLocalProvider ? `CPU ${cpuUsage.toFixed(0)}%` : void 0);
-    ctxUi.setStatus("status-ram", isLocalProvider ? `RAM ${fmtBytes(memUsed)}/${fmtBytes(memTotal)}` : void 0);
+    const theme = ctxTheme;
+    const dim2 = (s) => theme?.fg?.("dim", s) ?? s;
+    const green2 = (s) => theme?.fg?.("success", s) ?? s;
+    ctxUi.setStatus("status-cpu", isLocalProvider ? `${dim2("CPU")} ${green2(cpuUsage.toFixed(0) + "%")}` : void 0);
+    ctxUi.setStatus("status-ram", isLocalProvider ? `${dim2("RAM")} ${green2(fmtBytes(memUsed) + "/" + fmtBytes(memTotal))}` : void 0);
     ctxUi.setStatus(
       "status-swap",
-      isLocalProvider && hasSwap && swapUsed > 0 ? `Swap ${fmtBytes(swapUsed)}/${fmtBytes(swapTotal)}` : void 0
+      isLocalProvider && hasSwap && swapUsed > 0 ? `${dim2("Swap")} ${green2(fmtBytes(swapUsed) + "/" + fmtBytes(swapTotal))}` : void 0
     );
     ctxUi.setStatus(
       "status-native-ctx",
-      isLocalProvider && footerNativeCtx ? `M:${footerNativeCtx}` : void 0
-    );
-    ctxUi.setStatus(
-      "status-thinking",
-      footerThinking && footerThinking !== "off" ? footerThinking : void 0
+      footerNativeCtx ? `${dim2("CtxMax:")}${green2(footerNativeCtx)}` : void 0
     );
+    if (lastPayload) {
+      const rawMax = lastPayload.max_completion_tokens ?? lastPayload.max_tokens;
+      if (rawMax !== void 0) {
+        const formatted = rawMax >= 1e3 ? `${(rawMax / 1e3).toFixed(rawMax % 1e3 === 0 ? 0 : 1)}k` : String(rawMax);
+        ctxUi.setStatus("status-resp-max", `${dim2("RespMax:")}${green2(formatted)}`);
+      } else {
+        ctxUi.setStatus("status-resp-max", void 0);
+      }
+    } else {
+      ctxUi.setStatus("status-resp-max", void 0);
+    }
     ctxUi.setStatus(
       "status-resp",
-      lastResponseTime !== null ? `Resp ${fmtDur(lastResponseTime)}` : void 0
+      lastResponseTime !== null ? `${dim2("Resp")} ${green2(fmtDur(lastResponseTime))}` : void 0
     );
     if (lastPayload) {
       const params = extractParams(lastPayload);
-      ctxUi.setStatus("status-params", params.length > 0 ? params.join(" ") : void 0);
+      ctxUi.setStatus("status-params", params.length > 0 ? dim2(params.join(" ")) : void 0);
     } else {
       ctxUi.setStatus("status-params", void 0);
     }
     const now = Date.now();
     if (securityFlashTool && now < securityFlashUntil) {
-      ctxUi.setStatus("status-sec", `SEC:${blockedCount} (blocked: ${securityFlashTool})`);
+      ctxUi.setStatus("status-sec", `${dim2("SEC:")}${green2(String(blockedCount))} ${dim2("(blocked: " + securityFlashTool + ")")}`);
     } else if (blockedCount > 0) {
-      ctxUi.setStatus("status-sec", `SEC:${blockedCount}`);
+      ctxUi.setStatus("status-sec", `${dim2("SEC:")}${green2(String(blockedCount))}`);
     } else {
       ctxUi.setStatus("status-sec", void 0);
     }
     if (activeTool && activeToolStart > 0) {
       const elapsed = performance.now() - activeToolStart;
-      ctxUi.setStatus("status-tool", `> ${activeTool}: ${fmtDur(elapsed)}`);
+      ctxUi.setStatus("status-tool", `${green2(">")} ${dim2(activeTool + ":")} ${green2(fmtDur(elapsed))}`);
     } else {
       ctxUi.setStatus("status-tool", void 0);
     }
+    ctxUi.setStatus("system-prompt", cachedPromptText ?? void 0);
+    if (versionsText) {
+      ctxUi.setStatus("status-versions", dim2(versionsText));
+    }
   }
   function updateMetrics() {
     cpuUsage = getCpuUsage();
@@ -186,9 +201,8 @@ function status_temp_default(pi) {
     isLocalProvider = modelsJson ? detectLocalProvider(modelsJson) : false;
     if (currentCtx) {
       footerModel = currentCtx.model?.id || "";
-      footerThinking = pi.getThinkingLevel?.() ?? "";
       const modelId = currentCtx.model?.id || "";
-      if (modelId && isLocalProvider) {
+      if (modelId) {
         getNativeModelCtx(modelId);
       }
     }
@@ -197,7 +211,13 @@ function status_temp_default(pi) {
   pi.on("session_start", async (_event, ctx) => {
     currentCtx = ctx;
     ctxUi = ctx.ui;
+    ctxTheme = ctx.ui.theme;
     prevCpuInfo = getCpuSnapshot();
+    try {
+      const out = execSync("pi -v 2>&1", { encoding: "utf-8", timeout: 5e3 }).trim();
+      if (out) versionsText = `pi:${out}`;
+    } catch {
+    }
     updateMetrics();
     if (updateInterval) clearInterval(updateInterval);
     updateInterval = setInterval(updateMetrics, STATUS_UPDATE_INTERVAL_MS);
@@ -219,12 +239,13 @@ function status_temp_default(pi) {
       ui.setStatus("status-ram", void 0);
       ui.setStatus("status-swap", void 0);
       ui.setStatus("status-native-ctx", void 0);
-      ui.setStatus("status-thinking", void 0);
       ui.setStatus("status-resp", void 0);
+      ui.setStatus("status-resp-max", void 0);
       ui.setStatus("status-params", void 0);
       ui.setStatus("system-prompt", void 0);
       ui.setStatus("status-sec", void 0);
       ui.setStatus("status-tool", void 0);
+      ui.setStatus("status-versions", void 0);
     }
     securityFlashTool = "";
     securityFlashUntil = 0;
@@ -233,6 +254,8 @@ function status_temp_default(pi) {
     blockedCount = 0;
     lastResponseTime = null;
     lastPayload = null;
+    versionsText = "";
+    cachedPromptText = null;
   });
   pi.on("before_provider_request", (event) => {
     lastPayload = event.payload;
@@ -241,14 +264,12 @@ function status_temp_default(pi) {
     agentStartTime = performance.now();
     try {
       const prompt = ctx.getSystemPrompt();
-      const theme = ctx.ui.theme;
       const chr = prompt.length;
       const tok = prompt.split(/\s+/).filter(Boolean).length;
-      const label = theme.fg("dim", "Prompt:");
-      const value = theme.fg("success", `${chr} chr ${tok} tok`);
-      ctxUi?.setStatus("system-prompt", `${label} ${value}`);
+      cachedPromptText = `${dim("Prompt:")} ${green(`${chr} chr ${tok} tok`)}`;
     } catch {
     }
+    flushStatus();
   });
   pi.on("agent_end", async () => {
     if (agentStartTime !== null) {