@vtstech/pi-status 1.0.5 → 1.0.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vtstech/pi-status",
-  "version": "1.0.5",
+  "version": "1.0.6",
   "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.0.5"
+    "@vtstech/pi-shared": "1.0.6"
   },
   "peerDependencies": {
     "@mariozechner/pi-coding-agent": ">=0.66"

package/status.js

@@ -26,6 +26,7 @@ function status_temp_default(pi) {
   let footerModel = "";
   let footerThinking = "";
   let footerCtxPct = "";
+  let isLocalProvider = true;
   let securityFlashTool = "";
   let securityFlashUntil = 0;
   let activeTool = "";
@@ -75,6 +76,26 @@ function status_temp_default(pi) {
   let ollamaLoadedCache = "";
   let ollamaLoadedLastCheck = 0;
   const OLLAMA_LOADED_INTERVAL = 15e3;
+  function detectLocalProvider(modelsJson) {
+    try {
+      for (const provider of Object.values(modelsJson.providers || {})) {
+        const url = provider.baseUrl || "";
+        const hasLocalUrl = url.includes("localhost") || url.includes("127.0.0.1") || url.includes("0.0.0.0");
+        const modelId = footerModel || "";
+        if (modelId && (provider.models || []).some((m) => m.id === modelId)) {
+          return hasLocalUrl;
+        }
+      }
+      for (const [name, provider] of Object.entries(modelsJson.providers || {})) {
+        const url = provider.baseUrl || "";
+        if (url.includes("localhost") || url.includes("127.0.0.1") || url.includes("0.0.0.0") || name === "ollama") {
+          return true;
+        }
+      }
+    } catch {
+    }
+    return false;
+  }
   function getOllamaLoadedModel() {
     const now = Date.now();
     if (now - ollamaLoadedLastCheck < OLLAMA_LOADED_INTERVAL) return ollamaLoadedCache;
@@ -151,34 +172,37 @@ function status_temp_default(pi) {
       hasSwap = false;
     }
     ollamaLoaded = getOllamaLoadedModel();
+    let modelsJson = null;
+    try {
+      const raw = fs.readFileSync(
+        path.join(os.homedir(), ".pi", "agent", "models.json"),
+        "utf-8"
+      );
+      modelsJson = JSON.parse(raw);
+    } catch {
+    }
     if (currentCtx) {
       footerModel = currentCtx.model?.id || "";
       footerThinking = pi.getThinkingLevel?.() ?? "";
       const usage = currentCtx.getContextUsage?.();
       if (usage && usage.contextWindow > 0) {
-        const pct = (usage.tokens / usage.contextWindow * 100).toFixed(1);
-        footerCtxPct = `${pct}%/${(usage.contextWindow / 1e3).toFixed(0)}k`;
+        const pctVal = (usage.tokens / usage.contextWindow * 100).toFixed(1);
+        footerCtxPct = `${pctVal}%/${(usage.contextWindow / 1e3).toFixed(0)}k`;
       } else {
         footerCtxPct = "";
       }
       const modelId = currentCtx.model?.id || "";
-      if (modelId && !footerCtxPct) {
-        try {
-          const modelsJson = JSON.parse(fs.readFileSync(
-            path.join(os.homedir(), ".pi", "agent", "models.json"),
-            "utf-8"
-          ));
-          for (const prov of Object.values(modelsJson.providers || {})) {
-            const match = (prov.models || []).find((m) => m.id === modelId);
-            if (match?.contextLength) {
-              footerCtxPct = `${(match.contextLength / 1e3).toFixed(0)}k ctx`;
-              break;
-            }
+      if (modelId && !footerCtxPct && modelsJson) {
+        for (const prov of Object.values(modelsJson.providers || {})) {
+          const match = (prov.models || []).find((m) => m.id === modelId);
+          if (match?.contextLength) {
+            footerCtxPct = `${(match.contextLength / 1e3).toFixed(0)}k ctx`;
+            break;
           }
-        } catch {
         }
       }
     }
+    isLocalProvider = modelsJson ? detectLocalProvider(modelsJson) : false;
     refreshBlockedCount();
   }
   pi.on("session_start", async (_event, ctx) => {
@@ -207,10 +231,12 @@ function status_temp_default(pi) {
           if (footerModel) parts.push(dim(footerModel));
           if (footerThinking && footerThinking !== "off") parts.push(dim(footerThinking));
           if (footerCtxPct) parts.push(footerCtxPct);
-          parts.push(dim(`CPU ${cpuUsage.toFixed(0)}%`));
-          parts.push(`RAM ${fmtBytes(memUsed)}/${fmtBytes(memTotal)}`);
-          if (hasSwap && swapUsed > 0) {
-            parts.push(`Swap ${fmtBytes(swapUsed)}/${fmtBytes(swapTotal)}`);
+          if (isLocalProvider) {
+            parts.push(dim(`CPU ${cpuUsage.toFixed(0)}%`));
+            parts.push(`RAM ${fmtBytes(memUsed)}/${fmtBytes(memTotal)}`);
+            if (hasSwap && swapUsed > 0) {
+              parts.push(`Swap ${fmtBytes(swapUsed)}/${fmtBytes(swapTotal)}`);
+            }
           }
           if (ollamaLoaded) parts.push(`${ollamaLoaded}`);
           if (lastResponseTime !== null) parts.push(`Resp ${fmtDur(lastResponseTime)}`);