@pi-unipi/unipi 2.0.13 → 2.1.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-unipi/unipi",
-  "version": "2.0.13",
+  "version": "2.1.1",
   "description": "All-in-one extension suite for Pi coding agent",
   "type": "module",
   "license": "MIT",
@@ -59,38 +59,38 @@
     ]
   },
   "peerDependencies": {
-    "@earendil-works/pi-ai": "^0.78.0",
-    "@earendil-works/pi-coding-agent": "^0.78.0",
-    "@earendil-works/pi-tui": "^0.78.0",
+    "@earendil-works/pi-ai": "^0.80.0",
+    "@earendil-works/pi-coding-agent": "^0.80.0",
+    "@earendil-works/pi-tui": "^0.80.0",
     "typebox": "^1.1.38"
   },
   "dependencies": {
-    "@pi-unipi/ask-user": "2.0.13",
-    "@pi-unipi/btw": "2.0.13",
-    "@pi-unipi/compactor": "2.0.13",
-    "@pi-unipi/notify": "2.0.13",
-    "@pi-unipi/command-enchantment": "2.0.13",
-    "@pi-unipi/core": "2.0.13",
-    "@pi-unipi/info-screen": "2.0.13",
-    "@pi-unipi/mcp": "2.0.13",
-    "@pi-unipi/memory": "2.0.13",
-    "@pi-unipi/ralph": "2.0.13",
-    "@pi-unipi/subagents": "2.0.13",
-    "@pi-unipi/utility": "2.0.13",
-    "@pi-unipi/milestone": "2.0.13",
-    "@pi-unipi/kanboard": "2.0.13",
-    "@pi-unipi/web-api": "2.0.13",
-    "@pi-unipi/workflow": "2.0.13",
-    "@pi-unipi/footer": "2.0.13",
-    "@pi-unipi/updater": "2.0.13",
-    "@pi-unipi/input-shortcuts": "2.0.13",
-    "@pi-unipi/cocoindex": "2.0.13"
+    "@pi-unipi/ask-user": "2.1.1",
+    "@pi-unipi/btw": "2.1.1",
+    "@pi-unipi/compactor": "2.1.1",
+    "@pi-unipi/notify": "2.1.1",
+    "@pi-unipi/command-enchantment": "2.1.1",
+    "@pi-unipi/core": "2.1.1",
+    "@pi-unipi/info-screen": "2.1.1",
+    "@pi-unipi/mcp": "2.1.1",
+    "@pi-unipi/memory": "2.1.1",
+    "@pi-unipi/ralph": "2.1.1",
+    "@pi-unipi/subagents": "2.1.1",
+    "@pi-unipi/utility": "2.1.1",
+    "@pi-unipi/milestone": "2.1.1",
+    "@pi-unipi/kanboard": "2.1.1",
+    "@pi-unipi/web-api": "2.1.1",
+    "@pi-unipi/workflow": "2.1.1",
+    "@pi-unipi/footer": "2.1.1",
+    "@pi-unipi/updater": "2.1.1",
+    "@pi-unipi/input-shortcuts": "2.1.1",
+    "@pi-unipi/cocoindex": "2.1.1"
   },
   "devDependencies": {
-    "@earendil-works/pi-agent-core": "^0.78.0",
-    "@earendil-works/pi-ai": "^0.78.0",
-    "@earendil-works/pi-coding-agent": "^0.78.0",
-    "@earendil-works/pi-tui": "^0.78.0",
+    "@earendil-works/pi-agent-core": "^0.80.0",
+    "@earendil-works/pi-ai": "^0.80.0",
+    "@earendil-works/pi-coding-agent": "^0.80.0",
+    "@earendil-works/pi-tui": "^0.80.0",
     "@types/node": "^25.6.0",
     "tsx": "^4.21.0",
     "typescript": "^6.0.0"

package/packages/ask-user/index.ts

@@ -5,6 +5,8 @@
  * multi-select, and freeform modes. Includes bundled skill for agent guidance.
  */
 
+import { dirname } from "node:path";
+import { fileURLToPath } from "node:url";
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import {
   UNIPI_EVENTS,
@@ -17,7 +19,7 @@ import { registerAskUserTools } from "./tools.js";
 import { registerAskUserCommands } from "./commands.js";
 
 /** Package version */
-const VERSION = getPackageVersion(new URL(".", import.meta.url).pathname);
+const VERSION = getPackageVersion(dirname(fileURLToPath(import.meta.url)));
 
 export default function (pi: ExtensionAPI) {
 

package/packages/ask-user/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@pi-unipi/ask-user",
-  "version": "2.0.13",
-  "description": "Structured user input tool for Pi coding agent \u2014 single-select, multi-select, freeform",
+  "version": "2.1.1",
+  "description": "Structured user input tool for Pi coding agent — single-select, multi-select, freeform",
   "type": "module",
   "main": "index.ts",
   "license": "MIT",
@@ -40,11 +40,11 @@
     "access": "public"
   },
   "dependencies": {
-    "@pi-unipi/core": "2.0.13"
+    "@pi-unipi/core": "2.1.1"
   },
   "peerDependencies": {
-    "@earendil-works/pi-coding-agent": "^0.78.0",
-    "@earendil-works/pi-tui": "^0.78.0",
+    "@earendil-works/pi-coding-agent": "^0.80.0",
+    "@earendil-works/pi-tui": "^0.80.0",
     "typebox": "^1.1.38"
   },
   "pi": {

package/packages/ask-user/tools.ts

@@ -11,6 +11,7 @@ import {
   COMPACTOR_INSTRUCTION,
   UNIPI_EVENTS,
   emitEvent,
+  withHerdrBlocked,
 } from "@pi-unipi/core";
 import type { NormalizedOption, AskUserResponse, SessionLauncherResult } from "./types.js";
 import { renderAskUI, createRenderCall, createRenderResult } from "./ask-ui.js";
@@ -272,15 +273,19 @@ export function registerAskUserTools(pi: ExtensionAPI): void {
       }
 
       // Render interactive UI
-      const result = await ctx.ui.custom<{ response: AskUserResponse } | null>(
-        renderAskUI({
-          question,
-          context,
-          options: normalizedOptions,
-          allowMultiple,
-          allowFreeform,
-          timeout,
-        }),
+      const result = await withHerdrBlocked(
+        pi,
+        "ask_user",
+        () => ctx.ui.custom<{ response: AskUserResponse } | null>(
+          renderAskUI({
+            question,
+            context,
+            options: normalizedOptions,
+            allowMultiple,
+            allowFreeform,
+            timeout,
+          }),
+        ),
       );
 
       // Handle cancel
@@ -346,8 +351,12 @@ export function registerAskUserTools(pi: ExtensionAPI): void {
       // Session launcher intercept: when user selects new_session, offer compact/direct/cancel
       if (response.kind === "new_session") {
         const prefill = response.prefill || "";
-        const launcherResult = await ctx.ui.custom<SessionLauncherResult | null>(
-          renderLauncherUI({ prefill }),
+        const launcherResult = await withHerdrBlocked(
+          pi,
+          "ask_user: launch",
+          () => ctx.ui.custom<SessionLauncherResult | null>(
+            renderLauncherUI({ prefill }),
+          ),
         );
 
         if (!launcherResult || launcherResult.action === "cancel") {

package/packages/autocomplete/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@pi-unipi/command-enchantment",
-  "version": "2.0.13",
-  "description": "Enhanced TUI autocomplete for /unipi:* commands \u2014 colored, sorted, and grouped by package",
+  "version": "2.1.1",
+  "description": "Enhanced TUI autocomplete for /unipi:* commands — colored, sorted, and grouped by package",
   "type": "module",
   "main": "src/index.ts",
   "license": "MIT",
@@ -27,8 +27,8 @@
     "README.md"
   ],
   "peerDependencies": {
-    "@earendil-works/pi-coding-agent": "^0.78.0",
-    "@earendil-works/pi-tui": "^0.78.0"
+    "@earendil-works/pi-coding-agent": "^0.80.0",
+    "@earendil-works/pi-tui": "^0.80.0"
   },
   "scripts": {
     "test": "vitest run"

package/packages/btw/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@pi-unipi/btw",
-  "version": "2.0.13",
-  "description": "A pi extension for parallel side conversations with /unipi:btw \u2014 part of the Unipi suite",
+  "version": "2.1.1",
+  "description": "A pi extension for parallel side conversations with /unipi:btw — part of the Unipi suite",
   "type": "module",
   "main": "extensions/btw.ts",
   "license": "MIT",
@@ -32,12 +32,12 @@
     "access": "public"
   },
   "peerDependencies": {
-    "@earendil-works/pi-ai": "^0.78.0",
-    "@earendil-works/pi-coding-agent": "^0.78.0",
-    "@earendil-works/pi-tui": "^0.78.0"
+    "@earendil-works/pi-ai": "^0.80.0",
+    "@earendil-works/pi-coding-agent": "^0.80.0",
+    "@earendil-works/pi-tui": "^0.80.0"
   },
   "dependencies": {
-    "@pi-unipi/core": "2.0.13"
+    "@pi-unipi/core": "2.1.1"
   },
   "pi": {
     "extensions": [],

package/packages/cocoindex/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@pi-unipi/cocoindex",
-  "version": "2.0.13",
-  "description": "CocoIndex integration for Pi \u2014 AST-aware content indexing, semantic vector search, and incremental pipeline management",
+  "version": "2.1.1",
+  "description": "CocoIndex integration for Pi — AST-aware content indexing, semantic vector search, and incremental pipeline management",
   "type": "module",
   "main": "index.ts",
   "types": "index.ts",
@@ -19,13 +19,13 @@
     "typecheck": "tsc --noEmit"
   },
   "peerDependencies": {
-    "@earendil-works/pi-coding-agent": "^0.78.0",
-    "@earendil-works/pi-ai": "^0.78.0",
-    "@earendil-works/pi-tui": "^0.78.0",
+    "@earendil-works/pi-coding-agent": "^0.80.0",
+    "@earendil-works/pi-ai": "^0.80.0",
+    "@earendil-works/pi-tui": "^0.80.0",
     "typebox": "^1.1.38"
   },
   "dependencies": {
-    "@pi-unipi/core": "2.0.13"
+    "@pi-unipi/core": "2.1.1"
   },
   "optionalDependencies": {
     "@lancedb/lancedb": "^0.21.0"

package/packages/compactor/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@pi-unipi/compactor",
-  "version": "2.0.13",
-  "description": "Context engine for Pi \u2014 zero-LLM compaction, session continuity, sandbox execution, and tool display optimization",
+  "version": "2.1.1",
+  "description": "Context engine for Pi — zero-LLM compaction, session continuity, sandbox execution, and tool display optimization",
   "type": "module",
   "main": "src/index.ts",
   "license": "MIT",
@@ -34,13 +34,13 @@
     "access": "public"
   },
   "dependencies": {
-    "@pi-unipi/core": "2.0.13",
-    "@pi-unipi/info-screen": "2.0.13",
-    "@earendil-works/pi-agent-core": "^0.78.0"
+    "@pi-unipi/core": "2.1.1",
+    "@pi-unipi/info-screen": "2.1.1",
+    "@earendil-works/pi-agent-core": "^0.80.0"
   },
   "peerDependencies": {
-    "@earendil-works/pi-coding-agent": "^0.78.0",
-    "@earendil-works/pi-tui": "^0.78.0",
+    "@earendil-works/pi-coding-agent": "^0.80.0",
+    "@earendil-works/pi-tui": "^0.80.0",
     "typebox": "^1.1.38"
   },
   "devDependencies": {

package/packages/core/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-unipi/core",
-  "version": "2.0.13",
+  "version": "2.1.1",
   "description": "Shared utilities, event types, and constants for Unipi extension suite",
   "type": "module",
   "license": "MIT",
@@ -23,11 +23,11 @@
     "access": "public"
   },
   "peerDependencies": {
-    "@earendil-works/pi-coding-agent": "^0.78.0",
+    "@earendil-works/pi-coding-agent": "^0.80.0",
     "typebox": "^1.1.38"
   },
   "devDependencies": {
-    "@earendil-works/pi-coding-agent": "^0.78.0",
+    "@earendil-works/pi-coding-agent": "^0.80.0",
     "typebox": "^1.1.38",
     "@types/node": "^22.0.0"
   },

package/packages/core/utils.ts

@@ -240,3 +240,29 @@ export function emitEvent(
     return false;
   }
 }
+
+/**
+ * Wrap an awaitable blocking-UI operation with herdr `blocked` state reporting.
+ *
+ * Emits `herdr:blocked` active (with a label) before awaiting `fn`, and
+ * inactive (with matching label) afterwards — including on throw. This lets
+ * the herdr integration surface `blocked` agent status for ask_user and other
+ * stop-and-continue UIs instead of `working`.
+ *
+ * The label should match what's shown to the user so herdr's sidebar reads
+ * meaningfully (e.g. "ask_user", "helper viewer").
+ */
+export async function withHerdrBlocked<T>(
+  pi: { events: { emit: (name: string, payload: unknown) => void } },
+  label: string,
+  fn: () => Promise<T>,
+): Promise<T> {
+  const event = (active: boolean) =>
+    emitEvent(pi, "herdr:blocked", { active, label });
+  event(true);
+  try {
+    return await fn();
+  } finally {
+    event(false);
+  }
+}

package/packages/footer/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@pi-unipi/footer",
-  "version": "2.0.13",
-  "description": "Persistent status bar for Unipi \u2014 subscribes to UNIPI_EVENTS and renders key stats from all unipi packages",
+  "version": "2.1.1",
+  "description": "Persistent status bar for Unipi — subscribes to UNIPI_EVENTS and renders key stats from all unipi packages",
   "type": "module",
   "main": "src/index.ts",
   "license": "MIT",
@@ -32,11 +32,11 @@
     "access": "public"
   },
   "dependencies": {
-    "@pi-unipi/core": "2.0.13"
+    "@pi-unipi/core": "2.1.1"
   },
   "peerDependencies": {
-    "@earendil-works/pi-coding-agent": "^0.78.0",
-    "@earendil-works/pi-tui": "^0.78.0"
+    "@earendil-works/pi-coding-agent": "^0.80.0",
+    "@earendil-works/pi-tui": "^0.80.0"
   },
   "devDependencies": {
     "@types/node": "^25.6.0",

package/packages/footer/src/index.ts

@@ -96,6 +96,14 @@ export default function footerExtension(pi: ExtensionAPI): void {
     state.registry.registerGroup(group);
   }
 
+  // ─── TPS streaming-event hooks (registered once) ────────────────────────
+  // pi.on() has no unsubscribe, so we register these exactly once at factory
+  // time (not per session_start) to avoid duplicate handlers accumulating
+  // across session restarts. The streamingIndex counter is reset on each
+  // session_shutdown. These hooks feed the TPS tracker in real time; the
+  // 1s branch-scan in the refresh timer only reconciles persisted messages.
+  wireTpsStreamingEvents(pi);
+
   // ─── Session lifecycle ──────────────────────────────────────────────────
 
   pi.on("session_start", async (_event, ctx) => {
@@ -127,6 +135,7 @@ export default function footerExtension(pi: ExtensionAPI): void {
     }
     state.tuiRef = null;
     tpsTracker.reset();
+    resetTpsStreamingIndex();
   });
 
   // ─── Register commands ──────────────────────────────────────────────────
@@ -152,10 +161,14 @@ function setupFooterUI(pi: ExtensionAPI, ctx: ExtensionContext, state: FooterSta
   ctx.ui.setFooter((tui, _theme, footerData) => {
     state.tuiRef = tui;
 
-    // Start periodic refresh for time-sensitive segments (e.g. clock)
+    // Start periodic refresh for time-sensitive segments (e.g. clock, TPS)
     if (!state.refreshTimer) {
       state.refreshTimer = setInterval(() => {
-        // Feed TPS tracker with per-message data
+        // Re-seed TPS tracker from the session branch on each tick.
+        // Streaming events (message_start/update/end) handle live updates
+        // in real time; this scan reconciles the tracker with persisted
+        // messages after compactions, branch switches, or session reloads
+        // where in-flight streaming state may have been lost.
         try {
           const piCtx = state.piContext as Record<string, unknown> | undefined;
           if (piCtx?.sessionManager) {
@@ -168,9 +181,10 @@ function setupFooterUI(pi: ExtensionAPI, ctx: ExtensionContext, state: FooterSta
               const m = e.message;
               if (!m || m.role !== "assistant") continue;
               if (m.stopReason === "error" || m.stopReason === "aborted") continue;
-              const output = m.usage?.output ?? 0;
               const hasStop = !!m.stopReason;
-              tpsTracker.onMessageUpdate(msgIndex, output, hasStop);
+              // Pass the whole message: TPS tracker counts tokens from content,
+              // not from usage.output (which is 0 during streaming).
+              tpsTracker.onMessageUpdate(msgIndex, m, hasStop);
               msgIndex++;
             }
           }
@@ -251,3 +265,56 @@ function setupFooterUI(pi: ExtensionAPI, ctx: ExtensionContext, state: FooterSta
     };
   }, { placement: "belowEditor" });
 }
+
+// ─── TPS streaming-event hooks ──────────────────────────────────────────────
+
+/**
+ * Sequential index of the currently-streaming assistant message. Tracked
+ * locally because pi does not expose a stable message index on streaming
+ * events, and the TPS tracker keys records off this index.
+ */
+let tpsStreamingIndex = -1;
+
+/** Reset the streaming index (called on session_shutdown). */
+function resetTpsStreamingIndex(): void {
+  tpsStreamingIndex = -1;
+}
+
+/**
+ * Subscribe to pi's message streaming events and feed the TPS tracker in real
+ * time. This complements the 1s branch-scan in the refresh timer, which only
+ * sees persisted (completed) messages. Without these hooks the tracker would
+ * never observe an in-flight assistant message, so live TPS would stay frozen
+ * at the last completed message's value.
+ *
+ * Registered once at extension-factory time (pi.on has no unsubscribe, so we
+ * must not re-register per session_start or handlers would accumulate).
+ */
+function wireTpsStreamingEvents(pi: ExtensionAPI): void {
+  const safe = (fn: () => void) => {
+    try { fn(); } catch { /* TPS is best-effort */ }
+  };
+
+  pi.on("message_start", ((event: { message: unknown }) => safe(() => {
+    const m = event.message as Record<string, unknown> | undefined;
+    if (!m || m.role !== "assistant") return;
+    if (m.stopReason === "error" || m.stopReason === "aborted") return;
+    tpsStreamingIndex++;
+    tpsTracker.onMessageUpdate(tpsStreamingIndex, m, false);
+  })) as (event: unknown) => void);
+
+  pi.on("message_update", ((event: { message: unknown }) => safe(() => {
+    if (tpsStreamingIndex < 0) return;
+    const m = event.message as Record<string, unknown> | undefined;
+    if (!m || m.role !== "assistant") return;
+    tpsTracker.onMessageUpdate(tpsStreamingIndex, m, false);
+  })) as (event: unknown) => void);
+
+  pi.on("message_end", ((event: { message: unknown }) => safe(() => {
+    if (tpsStreamingIndex < 0) return;
+    const m = event.message as Record<string, unknown> | undefined;
+    if (!m || m.role !== "assistant") return;
+    if (m.stopReason === "error" || m.stopReason === "aborted") return;
+    tpsTracker.onMessageUpdate(tpsStreamingIndex, m, true);
+  })) as (event: unknown) => void);
+}