pi-minimalist 0.3.1 → 0.4.0

package/index.ts

@@ -34,7 +34,11 @@ export default function (pi: ExtensionAPI) {
   // strand every row on stale state.
   const { config, grouping } = sharedState(loadSettings());
 
-  installBridges({ config, grouping });
+  // A timer (the post-stream thinking linger) must repaint outside any render.
+  // Extensions get no TUI handle except through a factory, so an invisible
+  // widget (zero lines, below the editor, no spacer) captures it.
+  let tui: { requestRender(): void } | undefined;
+  installBridges({ config, grouping, requestRender: () => tui?.requestRender() });
   // Idempotent, so a /reload only refreshes the Bridge the wrappers read.
   patchCore(core);
 
@@ -46,7 +50,19 @@ export default function (pi: ExtensionAPI) {
     // restart. Session overrides are re-applied by Config.replace().
     config.replace(loadSettings());
   };
-  pi.on("session_start", refresh);
+  pi.on("session_start", (_event, ctx) => {
+    refresh();
+    if (ctx.hasUI && ctx.mode === "tui") {
+      ctx.ui.setWidget(
+        "pi-minimalist-repaint",
+        (captured) => {
+          tui = captured;
+          return { render: () => [], invalidate() {} };
+        },
+        { placement: "belowEditor" },
+      );
+    }
+  });
   pi.on("turn_start", refresh);
   pi.on("agent_start", () => grouping.agentStarted());
   pi.on("agent_settled", () => grouping.agentSettled());

package/package.json

@@ -1,8 +1,10 @@
 {
   "name": "pi-minimalist",
-  "version": "0.3.1",
+  "version": "0.4.0",
   "description": "Compact one-line tool and thinking rows for Pi's interactive transcript",
-  "keywords": ["pi-package"],
+  "keywords": [
+    "pi-package"
+  ],
   "license": "MIT",
   "author": "Pinglei He",
   "repository": {
@@ -10,13 +12,20 @@
     "url": "git+https://github.com/EviHex/pi-minimalist.git"
   },
   "type": "module",
-  "files": ["index.ts", "src/", "README.md", "LICENSE"],
+  "files": [
+    "index.ts",
+    "src/",
+    "README.md",
+    "LICENSE"
+  ],
   "peerDependencies": {
     "@earendil-works/pi-coding-agent": "*",
     "@earendil-works/pi-tui": "*"
   },
   "pi": {
-    "extensions": ["./index.ts"],
+    "extensions": [
+      "./index.ts"
+    ],
     "image": "https://raw.githubusercontent.com/EviHex/pi-minimalist/main/docs/click.png"
   }
 }

package/src/bridge.ts

@@ -16,7 +16,7 @@ import { RunGrouping } from "./run-grouping.ts";
 import { Painter, type Row, type ThemeLike } from "./row.ts";
 import { createToolRenderer, type ToolRenderer } from "./tool-renderer.ts";
 import { createThinkingPreview, singleHueThinkingTheme, type ThinkingPreview } from "./thinking.ts";
-import type { Timers } from "./components.ts";
+import { realTimers, type Timers } from "./components.ts";
 
 export const BRIDGE_SYMBOLS = {
   /** The `Bridge` object: everything the prototype wrappers call. */
@@ -42,6 +42,11 @@ export type Bridge = {
   compactThinking: () => boolean;
   /** Keep a STREAMING thinking block expanded. */
   keepActiveThinkingExpanded: () => boolean;
+  /**
+   * Run `callback` once, `THINKING_LINGER_MS` after a streamed message ends,
+   * then repaint. Keeps a short thought on screen long enough to be read.
+   */
+  afterThinkingLinger: (callback: () => void) => void;
 
   /** Chronology hooks that keep thinking and prose in transcript order. */
   observeThinking: (owner: object, runIndex: number, streaming: boolean, hidden: boolean, timestamp?: number) => string;
@@ -103,8 +108,13 @@ export function sharedState(settings?: ConstructorParameters<typeof Config>[0]):
 export type InstallOptions = SharedState & {
   /** Injectable repaint scheduler; production uses Node's timers. */
   timers?: Timers;
+  /** Repaint the TUI from outside a render (a timer). No-op without a TUI. */
+  requestRender?: () => void;
 };
 
+/** How long expanded thinking stays open after its message stops streaming. */
+export const THINKING_LINGER_MS = 3000;
+
 /**
  * Install the bridge.
  *
@@ -112,7 +122,7 @@ export type InstallOptions = SharedState & {
  * from `session_shutdown`: reload ordering can let an old shutdown hook erase the
  * newly installed bridge. Process exit clears globalThis naturally.
  */
-export function installBridges({ config, grouping, timers }: InstallOptions): void {
+export function installBridges({ config, grouping, timers = realTimers, requestRender }: InstallOptions): void {
   const globals = globalThis as Globals;
   const painter = (theme: ThemeLike) => new Painter(theme, config);
 
@@ -122,6 +132,14 @@ export function installBridges({ config, grouping, timers }: InstallOptions): vo
     thinkingMarkdownTheme: (base, theme) => singleHueThinkingTheme(base, theme, config),
     compactThinking: () => config.get("thinkingAsToolCall") === true,
     keepActiveThinkingExpanded: () => config.get("keepActiveThinkingExpanded"),
+    afterThinkingLinger: (callback) => {
+      // Timers only has intervals; clearing on the first tick makes it one-shot.
+      const handle = timers.setInterval(() => {
+        timers.clearInterval(handle);
+        callback();
+        requestRender?.();
+      }, THINKING_LINGER_MS);
+    },
 
     // Chronology: core walks message content in transcript order, and run
     // folding depends on that order.

package/src/config-ui.ts

@@ -71,7 +71,7 @@ export const FIELDS: Field[] = [
   {
     key: "keepActiveThinkingExpanded",
     label: "Show thinking while generating",
-    description: "Show the full thinking block as it arrives, instead of a one-line preview.",
+    description: "Show the full thinking block as it arrives (and for 3s after), instead of a one-line preview.",
   },
   {
     key: "glyphStyle",

package/src/core-patch.ts

@@ -264,7 +264,10 @@ function isMouseRegion(value: any): boolean {
  *    `hideThinkingBlock` + the override map, then restoring both. Either way the
  *    click handler stores into the LIVE map, so an explicit Ctrl+T expansion
  *    always survives — the default only decides what happens when the user has
- *    not expressed a preference.
+ *    not expressed a preference. When a streamed message ends, the forced
+ *    expansion LINGERS for `THINKING_LINGER_MS`, then the message re-renders
+ *    collapsed: a one-phrase thought would otherwise vanish before it is read.
+ *    Replayed history never streamed, so it never lingers.
  *  - collapsed preview: swap the hidden run's `Text` for our compact line.
  *  - expanded theme: recolor the run's `Markdown` in place (its private `theme`
  *    field is read at render time, and this runs before the first render).
@@ -283,13 +286,26 @@ export function patchAssistantMessage(proto: AssistantProto): void {
     // its mouse layout, so Pi's y (relative to the lines we returned) would
     // miss the summary by that shift. Remember it to translate clicks back.
     const mouseShift = new WeakMap<AssistantProto, { shift: number; height: number }>();
+    // Messages seen streaming, and those still inside their post-stream linger.
+    const streamed = new WeakSet<AssistantProto>();
+    const lingering = new WeakSet<AssistantProto>();
 
     proto.updateContent = function (this: AssistantProto, message: any, isStreaming = this.isStreaming) {
       const savedHide = this.hideThinkingBlock;
       const savedOverrides = this.thinkingVisibilityOverrides;
+      const b = bridge();
+      const keepOpen = b?.keepActiveThinkingExpanded() === true;
+      if (isStreaming) streamed.add(this);
+      else if (streamed.delete(this) && keepOpen && b) {
+        lingering.add(this);
+        b.afterThinkingLinger(() => {
+          lingering.delete(this);
+          if (this.lastMessage) this.updateContent(this.lastMessage, false);
+        });
+      }
       // Default: a streaming block stays collapsed to its compact preview, which
       // already shows the newest text. Opt in to force it open instead.
-      const forceOpen = isStreaming && bridge()?.keepActiveThinkingExpanded() === true;
+      const forceOpen = keepOpen && (isStreaming || lingering.has(this));
       if (forceOpen) {
         this.hideThinkingBlock = false;
         this.thinkingVisibilityOverrides = new Map();