pi-context 1.1.2 → 1.1.4

package/dist/index.js

@@ -281,7 +281,7 @@ export default function (pi) {
             // --- Context Dashboard (HUD) ---
             const usage = await ctx.getContextUsage();
             let usageStr = "Unknown";
-            if (usage) {
+            if (usage && usage.percent != null && usage.tokens != null && usage.contextWindow != null) {
                 usageStr = `${usage.percent.toFixed(1)}% (${formatTokens(usage.tokens)}/${formatTokens(usage.contextWindow)})`;
             }
             // Find the distance to the nearest tag

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-context",
-  "version": "1.1.2",
+  "version": "1.1.4",
   "description": "Agentic Context Management for the Pi",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -25,7 +25,7 @@
   "devDependencies": {
     "@sinclair/typebox": "^0.34.13",
     "typescript": "^5.7.2",
-    "@mariozechner/pi-coding-agent": "latest"
+    "@earendil-works/pi-coding-agent": "latest"
   },
   "pi": {
     "extensions": [

package/src/context.ts

@@ -62,6 +62,11 @@ export default function (pi: ExtensionAPI) {
       const totalActual = usage.tokens;
       const limit = usage.contextWindow;
 
+      if (totalActual == null || limit == null || usage.percent == null) {
+        ctx.ui.notify("Context usage info not available.", "warning");
+        return;
+      }
+
       const totalRaw = systemTokensRaw + toolDefTokensRaw + msgTokensRaw + toolUseTokensRaw + toolResultTokensRaw;
       const ratio = totalRaw > 0 ? (totalActual / totalRaw) : 1;
 

package/src/index.ts

@@ -82,11 +82,18 @@ export default function (pi: ExtensionAPI) {
     });
 
     // Helper: Check if a tag name already exists in the tree
+    // Iterative DFS to avoid call stack overflows on deep histories.
+    // Push children in reverse order to preserve left-to-right pre-order semantics.
     const findTagInTree = (sm: SessionManager, nodes: SessionTreeNode[], tagName: string): string | null => {
-        for (const n of nodes) {
+        const stack: SessionTreeNode[] = [...nodes].reverse();
+        while (stack.length > 0) {
+            const n = stack.pop()!;
             if (sm.getLabel(n.entry.id) === tagName) return n.entry.id;
-            const r = findTagInTree(sm, n.children, tagName);
-            if (r) return r;
+            if (n.children?.length) {
+                for (let i = n.children.length - 1; i >= 0; i--) {
+                    stack.push(n.children[i]);
+                }
+            }
         }
         return null;
     };

package/src/utils.ts

@@ -1,4 +1,5 @@
-export const formatTokens = (n: number) => {
+export const formatTokens = (n: number | null | undefined) => {
+  if (n == null) return "N/A";
   if (n >= 1_000_000) return (n / 1_000_000).toFixed(1) + "M";
   if (n >= 1_000) return Math.round(n / 1_000) + "k";
   return n.toString();