@vanillagreen/pi-qol 1.7.3 → 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. |
@@ -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,6 +1,6 @@
 import type { AgentMessage } from "@earendil-works/pi-agent-core";
 import type { Message } from "@earendil-works/pi-ai";
-import { complete } from "./pi-ai-compat.js";
+import { complete, retryAssistantCallCompat } from "./pi-ai-compat.js";
 import { convertToLlm, serializeConversation, type ExtensionContext } from "@earendil-works/pi-coding-agent";
 import {
 	DEFAULT_BUDGET_GUARD_PERCENT,
@@ -116,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");
@@ -164,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

@@ -50,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",
@@ -68,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

@@ -7,9 +7,31 @@ export interface PiAiCompatDeps {
 	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

@@ -21,6 +21,7 @@ type AutoRenameFallbackMode = "none" | "truncate" | "words";
 
 interface AutoRenameAuth {
 	apiKey?: string;
+	env?: Record<string, string>;
 	headers?: Record<string, string>;
 	label: string;
 	model: any;
@@ -263,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.
@@ -287,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");
@@ -331,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.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.3",
+  "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"
         },