@vanillagreen/pi-qol 1.7.2 → 1.7.4

package/README.md

@@ -76,7 +76,7 @@ Glyph style: each package exposes `glyphStyle` (`unicode` default, `ascii` for t
 | Show compact statusline | Render or disable the QOL statusline row. |
 | Replace built-in footer | Hide Pi's default footer while the QOL statusline is enabled. |
 | Use π prompt editor | Use the compact prompt editor. |
-| Show session name title | Show the session name above the prompt and in the tmux pane title. |
+| Show session name title | Show the session name above the prompt and in the tmux pane title; refreshes as soon as Pi reports a session metadata change. |
 | Sync session name to tmux window name | Rename the tmux window to `π <session>`. |
 | Input bottom padding | Blank lines below the prompt. |
 | Show dirty marker | Append `*` to the branch when the worktree is dirty. |
@@ -85,8 +85,8 @@ Glyph style: each package exposes `glyphStyle` (`unicode` default, `ascii` for t
 
 | Setting | What it does |
 | --- | --- |
-| Newline-insert binding | Insert a newline instead of submitting. |
-| Fallback newline binding | Alternate binding for terminals that can't send the primary one. |
+| shift+enter inserts newline | Insert a newline instead of submitting. |
+| Fallback newline key | Alternate binding for terminals that can't send the primary one. |
 | Style pending queue preview | Highlight Pi's pending-queue preview with a green left bar. |
 | Style image chips | Render `[Image #N]` placeholders as distinct chips. |
 | Show attachment count | Show a status badge when the draft has image placeholders. |
@@ -129,7 +129,7 @@ Advanced: input cap, title length, output tokens, timeout, custom prompt templat
 | Enable session search | Register `/search` and the overlay. |
 | Session search shortcut | Configurable; set to `none` to disable. |
 | Result limit | Max matching prompts returned. |
-| Visible session rows | Rows shown before scrolling. |
+| Visible session rows | Rows shown before scrolling; defaults to `8`. |
 | Preview snippets | Matching snippets shown on the preview screen. |
 | Session cache TTL | Seconds before the session list refreshes; `0` keeps it until you run `/search:refresh`. |
 
@@ -169,8 +169,8 @@ Off by default. When enabled, non-interactive matches are blocked.
 
 | Setting | What it does |
 | --- | --- |
-| Custom compaction summaries | Use QOL summaries instead of Pi's default. |
-| Compaction model | Summarizer model. Defaults to `current`, meaning Pi's active model; set a provider/model when you want a dedicated larger-context summarizer. |
+| Custom compaction summaries | Use QOL summaries instead of Pi's default, with Pi-standard transient provider retries when available. |
+| Compaction model | Summarizer model. Defaults to `current`, meaning Pi's active model; set a provider/model when you want a dedicated larger-context summarizer. Thinking suffixes through `:max` are accepted, and Pi-resolved header/environment authentication is forwarded. |
 | Compaction detail profile | `concise`, `balanced`, or `exhaustive`. |
 | Include previous summary | Pass the previous summary for iterative continuity. |
 | Fallback to Pi default compaction | Run Pi's default compaction if QOL's fails. |

package/extensions/qol/compaction.ts

@@ -1,5 +1,6 @@
 import type { AgentMessage } from "@earendil-works/pi-agent-core";
-import { complete, type Message } from "@earendil-works/pi-ai";
+import type { Message } from "@earendil-works/pi-ai";
+import { complete, retryAssistantCallCompat } from "./pi-ai-compat.js";
 import { convertToLlm, serializeConversation, type ExtensionContext } from "@earendil-works/pi-coding-agent";
 import {
 	DEFAULT_BUDGET_GUARD_PERCENT,
@@ -115,7 +116,7 @@ async function summarizeWithRemote(endpoint: string, systemPrompt: string, promp
 
 export function resolveConfiguredModel(ctx: ExtensionContext, configured: string): any | undefined {
 	if (!configured || configured.trim().toLowerCase() === "current") return ctx.model;
-	const withoutThinking = configured.replace(/:(off|minimal|low|medium|high|xhigh)$/i, "");
+	const withoutThinking = configured.replace(/:(off|minimal|low|medium|high|xhigh|max)$/i, "");
 	const slash = withoutThinking.indexOf("/");
 	if (slash > 0) return ctx.modelRegistry.find(withoutThinking.slice(0, slash), withoutThinking.slice(slash + 1));
 	const providers = [ctx.model?.provider, "google", "openai", "anthropic", "mistral", "moonshot", "cloudflare-ai-gateway", "cloudflare-workers-ai"].filter((value): value is string => typeof value === "string");
@@ -163,23 +164,34 @@ async function singleShotSummary(ctx: ExtensionContext, request: SummarizeReques
 	if (!model) throw new Error(`Summary model not found: ${configuredModel}`);
 	const auth = await ctx.modelRegistry.getApiKeyAndHeaders(model);
 	if (!auth.ok) throw new Error(auth.error);
-	if (!auth.apiKey) throw new Error(`No API key for ${model.provider}`);
 
 	const message: Message = {
 		content: [{ text: promptText, type: "text" }],
 		role: "user",
 		timestamp: Date.now(),
 	};
-	const response = await complete(
-		model,
-		{ messages: [message], systemPrompt: QOL_COMPACTION_SYSTEM_PROMPT },
-		{ apiKey: auth.apiKey, headers: auth.headers, maxTokens: options.maxTokens, signal: options.signal },
+	const response = await retryAssistantCallCompat(
+		() => complete(
+			model,
+			{ messages: [message], systemPrompt: QOL_COMPACTION_SYSTEM_PROMPT },
+			{ apiKey: auth.apiKey, env: auth.env, headers: auth.headers, maxTokens: options.maxTokens, signal: options.signal },
+		),
+		options.signal,
+		{
+			onRetryScheduled: (attempt, maxAttempts, delayMs, errorMessage) => {
+				compactionNotify(ctx, `QOL compaction retry ${attempt}/${maxAttempts} in ${Math.ceil(delayMs / 1000)}s: ${errorMessage}`, "warning");
+			},
+		},
 	);
+	if (response.stopReason === "error" || response.stopReason === "aborted") {
+		throw new Error(response.errorMessage || `Summary request stopped: ${response.stopReason}`);
+	}
 	const summary = response.content
 		.filter((content): content is { type: "text"; text: string } => content.type === "text")
 		.map((content) => content.text)
 		.join("\n")
 		.trim();
+	if (!summary) throw new Error("Summary model returned no text");
 	return { model: modelLabel(model), summary, via: "model" };
 }
 

package/extensions/qol/handoff.ts

@@ -1,5 +1,6 @@
 import type { AgentMessage } from "@earendil-works/pi-agent-core";
-import { complete, type Message } from "@earendil-works/pi-ai";
+import type { Message } from "@earendil-works/pi-ai";
+import { complete } from "./pi-ai-compat.js";
 import { BorderedLoader, type ExtensionCommandContext } from "@earendil-works/pi-coding-agent";
 import { acquireVstackModalLock } from "./bridges.js";
 import { serializeMessagesForSummary } from "./compaction.js";
@@ -49,9 +50,7 @@ export async function runHandoff(args: string, ctx: ExtensionCommandContext): Pr
 
 			const doGenerate = async () => {
 				const auth = await ctx.modelRegistry.getApiKeyAndHeaders(ctx.model!);
-				if (!auth.ok || !auth.apiKey) {
-					throw new Error(auth.ok ? `No API key for ${ctx.model!.provider}` : auth.error);
-				}
+				if (!auth.ok) throw new Error(auth.error);
 
 				const userMessage: Message = {
 					role: "user",
@@ -67,7 +66,7 @@ export async function runHandoff(args: string, ctx: ExtensionCommandContext): Pr
 				const response = await complete(
 					ctx.model!,
 					{ systemPrompt: HANDOFF_SYSTEM_PROMPT, messages: [userMessage] },
-					{ apiKey: auth.apiKey, headers: auth.headers, signal: loader.signal },
+					{ apiKey: auth.apiKey, env: auth.env, headers: auth.headers, signal: loader.signal },
 				);
 
 				if (response.stopReason === "aborted") return null;

package/extensions/qol/pi-ai-compat.ts

@@ -0,0 +1,37 @@
+import * as piAi from "@earendil-works/pi-ai";
+
+const dynamicImport = (specifier: string) => import(specifier);
+
+export interface PiAiCompatDeps {
+	root?: any;
+	loadCompat?: () => Promise<any>;
+}
+
+type RetryAssistantCall = (
+	produce: () => Promise<any>,
+	policy: { enabled: boolean; maxRetries: number; baseDelayMs: number },
+	signal: AbortSignal | undefined,
+	callbacks?: {
+		onRetryScheduled?: (attempt: number, maxAttempts: number, delayMs: number, errorMessage: string) => void | Promise<void>;
+		onRetryAttemptStart?: () => void | Promise<void>;
+		onRetryFinished?: (success: boolean, attempt: number, finalError?: string) => void | Promise<void>;
+	},
+) => Promise<any>;
+
+export async function complete(model: unknown, context: unknown, options?: unknown, deps: PiAiCompatDeps = {}): Promise<any> {
+	const rootComplete = (deps.root ?? piAi as any).complete;
+	if (typeof rootComplete === "function") return await rootComplete(model, context, options);
+	const compat = deps.loadCompat ? await deps.loadCompat() : await dynamicImport("@earendil-works/pi-ai/compat") as any;
+	return await compat.complete(model, context, options);
+}
+
+export async function retryAssistantCallCompat(
+	produce: () => Promise<any>,
+	signal: AbortSignal | undefined,
+	callbacks?: Parameters<RetryAssistantCall>[3],
+	deps: { root?: any } = {},
+): Promise<any> {
+	const candidate = (deps.root ?? piAi as any).retryAssistantCall as RetryAssistantCall | undefined;
+	if (typeof candidate !== "function") return produce();
+	return candidate(produce, { enabled: true, maxRetries: 3, baseDelayMs: 2000 }, signal, callbacks);
+}

package/extensions/qol/session-rename.ts

@@ -1,4 +1,5 @@
-import { complete, type Message } from "@earendil-works/pi-ai";
+import type { Message } from "@earendil-works/pi-ai";
+import { complete } from "./pi-ai-compat.js";
 import type { ExtensionContext, SessionEntry } from "@earendil-works/pi-coding-agent";
 import { textFromContent } from "./agent-end.js";
 import { oneLine } from "./ansi.js";
@@ -20,6 +21,7 @@ type AutoRenameFallbackMode = "none" | "truncate" | "words";
 
 interface AutoRenameAuth {
 	apiKey?: string;
+	env?: Record<string, string>;
 	headers?: Record<string, string>;
 	label: string;
 	model: any;
@@ -262,7 +264,7 @@ async function cheapestAvailableAutoRenameModel(ctx: ExtensionContext): Promise<
 		try {
 			const auth = await ctx.modelRegistry.getApiKeyAndHeaders(model);
 			const headers = headerRecord(auth.headers);
-			if (auth.ok && (auth.apiKey || headers)) return { apiKey: auth.apiKey, headers, label: modelLabel(model), model, source: "cheapest" };
+			if (auth.ok) return { apiKey: auth.apiKey, env: auth.env, headers, label: modelLabel(model), model, source: "cheapest" };
 		} catch (error) {
 			if (isStaleCtxError(error)) return undefined;
 			// Try the next available model.
@@ -286,11 +288,7 @@ async function resolveAutoRenameModel(ctx: ExtensionContext, configured: string)
 				autoRenameDebug(ctx, `auth unavailable for ${modelLabel(model)}: ${auth.error}`, "warning");
 				return undefined;
 			}
-			if (!auth.apiKey && !headers) {
-				autoRenameDebug(ctx, `no auth for ${modelLabel(model)}; use /login or models.json`, "warning");
-				return undefined;
-			}
-			return { apiKey: auth.apiKey, headers, label: modelLabel(model), model, source: configured };
+			return { apiKey: auth.apiKey, env: auth.env, headers, label: modelLabel(model), model, source: configured };
 		} catch (error) {
 			if (isStaleCtxError(error)) return undefined;
 			autoRenameDebug(ctx, `auth failed for ${modelLabel(model)}: ${stringifyError(error)}`, "warning");
@@ -330,7 +328,7 @@ export async function generateAutoRenameName(query: string, ctx: ExtensionContex
 				const response = await complete(
 					resolved.model,
 					{ messages: [message], systemPrompt: AUTO_RENAME_SYSTEM_PROMPT },
-					{ apiKey: resolved.apiKey, headers: resolved.headers, maxTokens, signal: controller.signal },
+					{ apiKey: resolved.apiKey, env: resolved.env, headers: resolved.headers, maxTokens, signal: controller.signal },
 				);
 				if (response.stopReason === "error") {
 					autoRenameDebug(ctx, `${resolved.label} failed: ${response.errorMessage ?? "unknown error"}`, "warning");

package/extensions/qol/statusline.ts

@@ -39,17 +39,19 @@ function formatModelName(ctx: ExtensionContext): string {
 	return name;
 }
 
-type ThinkingLevel = "off" | "minimal" | "low" | "medium" | "high" | "xhigh";
-const THINKING_TOKEN: Record<ThinkingLevel, "thinkingOff" | "thinkingMinimal" | "thinkingLow" | "thinkingMedium" | "thinkingHigh" | "thinkingXhigh"> = {
+type ThinkingLevel = "off" | "minimal" | "low" | "medium" | "high" | "xhigh" | "max";
+type ThinkingThemeToken = "thinkingOff" | "thinkingMinimal" | "thinkingLow" | "thinkingMedium" | "thinkingHigh" | "thinkingXhigh" | "thinkingMax";
+const THINKING_TOKEN: Record<ThinkingLevel, ThinkingThemeToken> = {
 	off: "thinkingOff",
 	minimal: "thinkingMinimal",
 	low: "thinkingLow",
 	medium: "thinkingMedium",
 	high: "thinkingHigh",
 	xhigh: "thinkingXhigh",
+	max: "thinkingMax",
 };
 
-function normalizeThinkingLevel(value: string | undefined): ThinkingLevel {
+export function normalizeThinkingLevel(value: string | undefined): ThinkingLevel {
 	switch ((value ?? "").toLowerCase()) {
 		case "off": return "off";
 		case "minimal": return "minimal";
@@ -57,10 +59,15 @@ function normalizeThinkingLevel(value: string | undefined): ThinkingLevel {
 		case "medium": return "medium";
 		case "high": return "high";
 		case "xhigh": return "xhigh";
+		case "max": return "max";
 		default: return "off";
 	}
 }
 
+export function thinkingThemeToken(value: string | undefined): ThinkingThemeToken {
+	return THINKING_TOKEN[normalizeThinkingLevel(value)];
+}
+
 function formatWindow(tokens: number | undefined): string {
 	if (!tokens || tokens <= 0) return "?";
 	if (tokens >= 1_000_000) {
@@ -207,9 +214,10 @@ export function renderStatusLine(width: number, ctx: ExtensionContext, git: GitS
 	const rightPlain = subagentMarker ? `${percentPlain} ${subagentMarker.plain}` : percentPlain;
 	const percentColor = percent === null ? "muted" : percent <= 15 ? "error" : percent <= 30 ? "warning" : "success";
 	const separatorColored = theme.fg("muted", statusSeparator);
+	const thinkingToken = thinkingThemeToken(thinkingLevel) as Parameters<Theme["fg"]>[0];
 	const leftColored = caveman
-		? `${theme.fg("accent", projectChunk)}${separatorColored}${theme.fg(THINKING_TOKEN[thinkingLevel], thinkingChunk)}${separatorColored}${theme.fg(cavemanTone, cavemanGlyph)}${theme.fg("muted", contextSeparator)}${theme.fg("accent", contextChunk.trimStart())}`
-		: `${theme.fg("accent", projectChunk)}${separatorColored}${theme.fg(THINKING_TOKEN[thinkingLevel], thinkingChunk)}${theme.fg("accent", contextChunk)}`;
+		? `${theme.fg("accent", projectChunk)}${separatorColored}${theme.fg(thinkingToken, thinkingChunk)}${separatorColored}${theme.fg(cavemanTone, cavemanGlyph)}${theme.fg("muted", contextSeparator)}${theme.fg("accent", contextChunk.trimStart())}`
+		: `${theme.fg("accent", projectChunk)}${separatorColored}${theme.fg(thinkingToken, thinkingChunk)}${theme.fg("accent", contextChunk)}`;
 	const right = subagentMarker ? `${theme.fg(percentColor, percentPlain)} ${subagentMarker.styled}` : theme.fg(percentColor, percentPlain);
 	const minimumGap = 1;
 	const gapWidth = Math.max(minimumGap, width - visibleWidth(leftPlain) - visibleWidth(rightPlain) - 2);

package/extensions/qol.ts

@@ -668,6 +668,13 @@ export default function qol(pi: ExtensionAPI): void {
 	pi.on("thinking_level_select", (_event, ctx) => {
 		if (ctx.hasUI) requestRender();
 	});
+	// Renames land immediately instead of waiting for the SESSION_TITLE_SYNC_INTERVAL_MS
+	// poll installed by installSessionTitleSync; the poll stays as the backstop for
+	// name changes Pi does not announce.
+	pi.on("session_info_changed", (_event, ctx) => {
+		if (!ctx.hasUI) return;
+		syncSessionTitle(ctx);
+	});
 	pi.on("agent_start", (_event, ctx) => {
 		clearIdleCompactionTimer();
 		clearTmuxWindowMark();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vanillagreen/pi-qol",
-  "version": "1.7.2",
+  "version": "1.7.4",
   "description": "Pi quality-of-life extension: compact statusline/π prompt, reliable multiline input, styled pasted-image chips, session naming/search/context import, scheduled prompts, handoff, permission prompts, notifications, custom compaction, and a collapsed-thinking timer.",
   "license": "MIT",
   "keywords": [
@@ -79,7 +79,7 @@
         {
           "key": "showSessionNameTitle",
           "label": "Show session name title",
-          "description": "Show the named session above the prompt. In tmux, update the current pane title to \"π <session>\" and enable a pane border title padded with non-breaking spaces so tmux does not trim it.",
+          "description": "Show the named session above the prompt. In tmux, update the current pane title to \"π <session>\" and enable a pane border title padded with non-breaking spaces so tmux does not trim it. The title refreshes as soon as Pi reports a session metadata change.",
           "type": "boolean",
           "default": true,
           "category": "Statusline",
@@ -452,7 +452,7 @@
           "label": "Visible session rows",
           "description": "Maximum number of sessions shown at once in the overlay before scrolling.",
           "type": "number",
-          "default": 10,
+          "default": 8,
           "category": "Session Search",
           "apply": "live"
         },