@pi-kaush/pi-tool-call-markers 0.2.10 → 0.2.11

package/CHANGELOG.md

@@ -13,10 +13,11 @@
 - Adopt Pi's native content-color split in collapsed rows: tool names ride
   `toolTitle` and call content/outcomes ride `toolOutput` (structural
   chrome stays muted; failed rows stay uniformly error).
-- Tint the collapsed "+ Thought" label with the session's active
-  thinking-level token (thinkingOff…thinkingMax), re-tinting on
+- Quiet the settled "+ Thought" label to the muted tone collapsed tool
+  calls use; only the live "⠋ Thinking…" spinner keeps the session's active
+  thinking-level tint (thinkingOff…thinkingMax), re-tinting on
   thinking_level_select; the expanded thinking block is untouched. The
-  PI_TOOL_CALL_MARKERS_THOUGHT_COLOR values are now `level` (default),
+  PI_TOOL_CALL_MARKERS_THOUGHT_COLOR values are `level` (default),
   `mdheading`, and `inherit`; the midpoint `gray` variant and its RGB
   color math are gone.
 - Paint user-run `!` bash block surfaces with the theme's

package/README.md

@@ -2,6 +2,14 @@
 
 Give Pi's collapsed tool calls a quiet, width-safe transcript shell while preserving native details under `Ctrl+O`. The package also includes a display-only thinking-block adapter.
 
+### Install
+
+```fish
+pi install npm:@pi-kaush/pi-tool-call-markers
+```
+
+Restart Pi or run `/reload`.
+
 ## What it changes
 
 Collapsed tool rows use semantic theme colors with no gear, background fill, box padding, or filled blank rows:
@@ -73,15 +81,9 @@ When Pi exposes its per-row hidden-thinking and streaming fields, hidden reasoni
 
 The live label samples Pi's native braille spinner sequence from the content updates Pi already renders; it does not add a timer. The adapter stores the first local streaming timestamp per assistant row in a `WeakMap`. A restored message or an older runtime with no streaming argument uses `+ Thought`. Visible-thinking mode remains native. There is no interval, timeout, render request, model call, or network work.
 
-## Install
+## Local development
 
-```bash
-pi install npm:@pi-kaush/pi-tool-call-markers
-```
-
-For local development:
-
-```bash
+```fish
 pi \
   -e ./extensions/pi-tool-call-markers/src/index.ts \
   -e ./extensions/pi-tool-call-markers/src/thinking-block-merger.ts

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-kaush/pi-tool-call-markers",
-  "version": "0.2.10",
+  "version": "0.2.11",
   "description": "Compact and group Pi tool calls, with a bundled extension for adjacent thinking blocks.",
   "license": "MIT",
   "author": "Kaushik Gopal",

package/src/index.ts

@@ -1660,6 +1660,14 @@ function uninstallPresentationPatch(
   delete proto[PRESENTATION_PATCHED];
 }
 
+// TODO(compaction cards): native/failed compaction cards in the transcript
+// are currently rendered by the pi-verbatim-compaction extension. Its
+// src/chat-log.ts is deliberately self-contained (public Pi imports only), so
+// porting here is: drop in that file, registerEntryRenderer(COMPACTION_LOG_TYPE),
+// and wire pi.on("session_compact") / pi.on("session_compact_failed") filtered
+// to fromExtension === false. Do this only if pi-verbatim-compaction is
+// removed or native cards should exist without it; while both are installed,
+// rendering here too would double-render every native compaction.
 export default function (pi: ExtensionAPI) {
   const patch = installPresentationPatch();
   const grouping = patch ? installGroupingPatch(patch) : undefined;

package/src/thinking-block-merger.ts

@@ -6,13 +6,16 @@ const THINKING_GROUPING_PATCHED = Symbol.for("kg.pi.thinkingGrouping.v2");
 const PI_SPINNER_FRAMES = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"];
 const SPINNER_INTERVAL_MS = 80;
 
-// Visual experiment for the settled "+ Thought" label. "inherit" keeps Pi's
-// native styling (italic + thinkingText); "mdheading" rides the theme's
-// mdHeading token; the default tints the label with the session's active
-// thinking-level color (thinkingOff…thinkingMax), so the collapsed label
-// quietly advertises the level Pi is reasoning at. Only the collapsed label
-// is restyled — the expanded thinking block keeps Pi's thinkingText. The env
-// var overrides the default so variants can be compared without republishing.
+// Visual treatment for the collapsed thinking label. The live "Thinking…"
+// spinner tints with the session's active thinking-level color
+// (thinkingOff…thinkingMax), so progress reads as activity; the settled
+// "+ Thought" row drops to the muted token collapsed tool calls use, so
+// finished reasoning stops advertising the level. "inherit" keeps Pi's
+// native styling (italic + thinkingText) for both states, and "mdheading"
+// rides the theme's mdHeading token for both. The env var overrides the
+// default so variants can be compared without republishing. Only the
+// collapsed label is restyled — the expanded thinking block keeps Pi's
+// thinkingText.
 const THOUGHT_LABEL_COLOR_ENV = "PI_TOOL_CALL_MARKERS_THOUGHT_COLOR";
 type ThoughtLabelColor = "inherit" | "level" | "mdheading";
 const DEFAULT_THOUGHT_LABEL_COLOR: ThoughtLabelColor = "level";
@@ -44,35 +47,48 @@ type ThemeDetail = {
 let activeTheme: ThemeDetail | undefined;
 let activeLevel: (() => string) | undefined;
 
-let thoughtLabelPrefix = "";
-let thoughtLabelSuffix = "";
+type LabelStyle = { prefix: string; suffix: string };
+let liveLabelStyle: LabelStyle | undefined;
+let settledLabelStyle: LabelStyle | undefined;
+
+function styledWith(token: string): LabelStyle | undefined {
+  const ansi = activeTheme?.getFgAnsi?.(token);
+  if (!ansi) return undefined;
+  // The styled label replaces Pi's italicized Text node wholesale (see
+  // restyleHiddenThinkingLabel). The leading italic-off still matters: the
+  // TUI's diff renderer can skip bytes shared with the previously drawn
+  // italic line, leaving the terminal in italic state otherwise.
+  return { prefix: `\x1b[23m${ansi}`, suffix: "\x1b[39m" };
+}
 
 function updateThoughtLabelStyle(): void {
-  thoughtLabelPrefix = "";
-  thoughtLabelSuffix = "";
+  liveLabelStyle = undefined;
+  settledLabelStyle = undefined;
   const choice = thoughtLabelColorChoice();
   if (choice === "inherit" || !activeTheme) return;
 
-  let ansi: string | undefined;
   const token =
     choice === "mdheading"
       ? "mdHeading"
       : (LEVEL_TOKEN[activeLevel?.() ?? "off"] ?? "thinkingOff");
-  ansi = activeTheme.getFgAnsi?.(token);
-  if (!ansi) return;
+  liveLabelStyle = styledWith(token);
+  settledLabelStyle =
+    choice === "mdheading" ? liveLabelStyle : styledWith("muted");
+}
 
-  // The styled label replaces Pi's italicized Text node wholesale (see
-  // restyleHiddenThinkingLabel). The leading italic-off still matters: the
-  // TUI's diff renderer can skip bytes shared with the previously drawn
-  // italic line, leaving the terminal in italic state otherwise.
-  thoughtLabelPrefix = `\x1b[23m${ansi}`;
-  thoughtLabelSuffix = "\x1b[39m";
+// The settled label is the finalized "+ Thought" row; every other label is
+// the live spinner.
+function isSettledThoughtLabel(label: string): boolean {
+  return label.startsWith("+ Thought");
 }
 
 export function visibleThoughtLabel(label: string): string {
-  if (!thoughtLabelPrefix) return label;
+  const style = isSettledThoughtLabel(label)
+    ? settledLabelStyle
+    : liveLabelStyle;
+  if (!style) return label;
   const raw = label.replace(/\x1b\[[0-9;]*m/g, "");
-  return `${thoughtLabelPrefix}${raw}${thoughtLabelSuffix}`;
+  return `${style.prefix}${raw}${style.suffix}`;
 }
 
 type AssistantMessageLike = {
@@ -98,7 +114,6 @@ type TextLikeChild = {
 // terminal. Swap the node's text for a self-contained styled version after
 // each native render pass instead.
 function restyleHiddenThinkingLabel(row: AssistantMessageRow): void {
-  if (!thoughtLabelPrefix) return;
   if (
     row.hideThinkingBlock !== true ||
     typeof row.hiddenThinkingLabel !== "string"
@@ -108,6 +123,7 @@ function restyleHiddenThinkingLabel(row: AssistantMessageRow): void {
   const label = row.hiddenThinkingLabel;
   const children = row.contentContainer?.children;
   if (!Array.isArray(children)) return;
+  if (!liveLabelStyle && !settledLabelStyle) return;
   const styled = visibleThoughtLabel(label);
   for (const child of children) {
     const textChild = child as TextLikeChild | undefined;
@@ -419,7 +435,7 @@ export default function (pi: ExtensionAPI) {
     if (patch && !uninstallThinkingGroupingPatch(patch)) return;
     activeTheme = undefined;
     activeLevel = undefined;
-    thoughtLabelPrefix = "";
-    thoughtLabelSuffix = "";
+    liveLabelStyle = undefined;
+    settledLabelStyle = undefined;
   });
 }