@diegopetrucci/pi-extensions 0.1.52 → 0.1.54

package/extensions/git-footer/index.ts

@@ -31,6 +31,11 @@ type CommandRunner = (
 	options: { cwd: string; signal: AbortSignal },
 ) => Promise<CommandResult>;
 
+type RunCommandSafelyResult =
+	| { kind: "ok"; result: CommandResult }
+	| { kind: "transient" }
+	| { kind: "unavailable" };
+
 type TimerHandle = unknown;
 
 type Clock = {
@@ -286,6 +291,10 @@ function defaultClock(): Clock {
 	};
 }
 
+function isCommandUnavailableError(error: unknown): boolean {
+	return !!error && typeof error === "object" && (error as { code?: unknown }).code === "ENOENT";
+}
+
 class GitFooterCache {
 	private readonly cwd: () => string;
 	private readonly runner: CommandRunner;
@@ -345,7 +354,7 @@ class GitFooterCache {
 		const result = await this.fetchGitStatus();
 		if (this.disposed) return;
 		if (result.kind === "transient") return;
-		if (result.kind === "not-a-repo") {
+		if (result.kind === "not-a-repo" || result.kind === "unavailable") {
 			this.statusSnapshot = undefined;
 			this.pullRequestSnapshot = undefined;
 			this.lastSeenBranch = undefined;
@@ -371,8 +380,8 @@ class GitFooterCache {
 
 		const pr = await this.fetchPullRequest();
 		if (this.disposed) return;
-		if (pr !== undefined) this.pullRequestSnapshot = pr;
-		else if (branchChanged) this.pullRequestSnapshot = undefined;
+		if (pr.kind === "ok") this.pullRequestSnapshot = pr.pullRequest;
+		else if (pr.kind === "not-found" || pr.kind === "unavailable") this.pullRequestSnapshot = undefined;
 
 		this.emitChangeIfSnapshotsChanged(previousStatusSnapshot, previousPullRequestSnapshot);
 	}
@@ -399,32 +408,40 @@ class GitFooterCache {
 		| { kind: "ok"; status: GitStatusSnapshot }
 		| { kind: "not-a-repo" }
 		| { kind: "transient" }
+		| { kind: "unavailable" }
 	> {
 		const result = await this.runCommandSafely("git", GIT_STATUS_ARGS, this.gitTimeoutMs);
-		if (!result) return { kind: "transient" };
-		if (result.exitCode !== 0) return { kind: "not-a-repo" };
-		return { kind: "ok", status: parseGitStatusPorcelainV2(result.stdout) };
+		if (result.kind !== "ok") return result;
+		if (result.result.exitCode !== 0) return { kind: "not-a-repo" };
+		return { kind: "ok", status: parseGitStatusPorcelainV2(result.result.stdout) };
 	}
 
-	private async fetchPullRequest(): Promise<PullRequestSnapshot | undefined> {
+	private async fetchPullRequest(): Promise<
+		| { kind: "ok"; pullRequest: PullRequestSnapshot | undefined }
+		| { kind: "not-found" }
+		| { kind: "transient" }
+		| { kind: "unavailable" }
+	> {
 		const result = await this.runCommandSafely("gh", GH_PR_VIEW_ARGS, this.ghTimeoutMs);
-		if (!result || result.exitCode !== 0) return undefined;
-		return parsePullRequestJson(result.stdout);
+		if (result.kind !== "ok") return result;
+		if (result.result.exitCode !== 0) return { kind: "not-found" };
+		return { kind: "ok", pullRequest: parsePullRequestJson(result.result.stdout) };
 	}
 
 	private async runCommandSafely(
 		command: string,
 		args: readonly string[],
 		timeoutMs: number,
-	): Promise<CommandResult | undefined> {
-		if (this.disposed) return undefined;
+	): Promise<RunCommandSafelyResult> {
+		if (this.disposed) return { kind: "transient" };
 		const controller = new AbortController();
 		this.inflightControllers.add(controller);
 		const timeoutId = setTimeout(() => controller.abort(), timeoutMs);
 		try {
-			return await this.runner(command, args, { cwd: this.cwd(), signal: controller.signal });
-		} catch {
-			return undefined;
+			const result = await this.runner(command, args, { cwd: this.cwd(), signal: controller.signal });
+			return { kind: "ok", result };
+		} catch (error) {
+			return isCommandUnavailableError(error) ? { kind: "unavailable" } : { kind: "transient" };
 		} finally {
 			clearTimeout(timeoutId);
 			this.inflightControllers.delete(controller);
@@ -478,3 +495,14 @@ export default function (pi: ExtensionAPI) {
 		ctx.ui.setStatus(STATUS_KEY, undefined);
 	});
 }
+
+export const __testing = {
+	GitFooterCache,
+	GIT_STATUS_ARGS,
+	GH_PR_VIEW_ARGS,
+	parseGitStatusPorcelainV2,
+	parsePullRequestJson,
+	formatGitStatusFooterSegment,
+	formatPullRequestFooterSegment,
+	formatGitFooterStatus,
+};

package/extensions/git-footer/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@diegopetrucci/pi-git-footer",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "description": "A TLH-style git status add-on for pi's built-in footer.",
   "keywords": [
     "pi-package",

package/extensions/gnosis/.pi-fleet-tested-version

@@ -1 +1 @@
-0.79.10
+0.80.3

package/extensions/gnosis/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@diegopetrucci/pi-gnosis",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "description": "A pi extension that exposes the gnosis repo-local knowledge base CLI as an agent tool.",
   "keywords": [
     "pi-package",

package/extensions/illustrations-to-explain-things/.pi-fleet-tested-version

@@ -1 +1 @@
-0.79.10
+0.80.3

package/extensions/illustrations-to-explain-things/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@diegopetrucci/pi-illustrations-to-explain-things",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "description": "A pi skill for generating clean, absurd Xiaohei-style article illustrations, shot lists, and visual metaphors.",
   "keywords": [
     "pi-package",

package/extensions/inline-bash/.pi-fleet-tested-version

@@ -1 +1 @@
-0.79.10
+0.80.3

package/extensions/inline-bash/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@diegopetrucci/pi-inline-bash",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "description": "A pi extension that expands inline bash commands in user prompts.",
   "keywords": [
     "pi-package",

package/extensions/librarian/.pi-fleet-tested-version

@@ -1 +1 @@
-0.79.10
+0.80.3

package/extensions/librarian/index.ts

@@ -477,53 +477,104 @@ async function findAvailableModel(
 	return undefined;
 }
 
-async function selectLibrarianModel(
+// A model the catalog advertises but the active provider/subscription cannot
+// serve surfaces as a not-found/404-style error (legacy snapshots or
+// access-gated tiers). These are NOT transient: the same model keeps failing,
+// so we fall back to the next candidate instead of retrying it.
+const MODEL_AVAILABILITY_ERROR_PATTERN =
+	/\b(?:404|403|not[_ ]?found(?:[_ ]?error)?|model[_ ]?not[_ ]?found(?:[_ ]?error)?|no such model|unknown model|does not exist|is not available|not available|model[_ ]?not[_ ]?available|unsupported model|invalid model|forbidden|access[ _-]?denied|permission[ _-]?denied|not[ _-]?entitled|do(?:es)? not have access)\b/i;
+
+function isModelAvailabilityError(message: string | undefined): boolean {
+	if (!message) return false;
+	return MODEL_AVAILABILITY_ERROR_PATTERN.test(message);
+}
+
+type LibrarianCandidate = { model: any; details: LibrarianModelDetails };
+
+function withLibrarianFallbackReason(candidate: LibrarianCandidate, previous: LibrarianCandidate): LibrarianCandidate {
+	return {
+		model: candidate.model,
+		details: {
+			...candidate.details,
+			selectionReason: `${candidate.details.selectionReason} (Fell back from ${previous.details.modelRef} after a model-availability error.)`,
+		},
+	};
+}
+
+// Build an ordered list of model candidates to try. The catalog can advertise
+// models the active provider/subscription cannot actually serve, so callers run
+// these in order and skip ones that fail with a model-availability error,
+// ultimately falling back to the known-good current session model.
+async function buildLibrarianCandidates(
 	ctx: { model?: any; modelRegistry: { getAvailable(): any[] | Promise<any[]> } },
 	modelPreference: string | undefined,
 	thinkingLevel: ThinkingLevel,
-): Promise<{ model: any; details: LibrarianModelDetails }> {
+): Promise<LibrarianCandidate[]> {
+	const candidates: LibrarianCandidate[] = [];
+	const seen = new Set<string>();
+	const push = (model: any, details: LibrarianModelDetails) => {
+		if (!model) return;
+		const key = `${model.provider}/${model.id}`.toLowerCase();
+		if (seen.has(key)) return;
+		seen.add(key);
+		candidates.push({ model, details });
+	};
+
 	const normalized = modelPreference?.trim();
+	let configuredMatched: any | undefined;
 	if (normalized) {
-		const matched = await findAvailableModel(ctx, normalized);
-		if (matched) {
-			return {
-				model: matched,
-				details: {
-					modelRef: modelRef(matched),
-					provider: matched.provider,
-					modelId: matched.id,
-					thinkingLevel,
-					autoSelected: false,
-					selectionReason: "Using the configured librarian model.",
-				},
-			};
+		configuredMatched = await findAvailableModel(ctx, normalized);
+		if (configuredMatched) {
+			push(configuredMatched, {
+				modelRef: modelRef(configuredMatched),
+				provider: configuredMatched.provider,
+				modelId: configuredMatched.id,
+				thinkingLevel,
+				autoSelected: false,
+				selectionReason: "Using the configured librarian model.",
+			});
 		}
 	}
 
 	const available = await ctx.modelRegistry.getAvailable();
 	const preferred = available.filter(isPreferredFastLibrarianModel);
-	const winner = preferred.length > 0
-		? [...preferred].sort((a, b) => rankPreferredFastLibrarianModel(a) - rankPreferredFastLibrarianModel(b))[0]
-		: [...available].sort((a, b) => rankLibrarianModel(a) - rankLibrarianModel(b))[0];
-	if (!winner) {
-		throw new Error("No authenticated models are available for Librarian. Log in or configure an API key first.");
-	}
+	const ranked = preferred.length > 0
+		? [...preferred].sort((a, b) => rankPreferredFastLibrarianModel(a) - rankPreferredFastLibrarianModel(b))
+		: [...available].sort((a, b) => rankLibrarianModel(a) - rankLibrarianModel(b));
+	const fallbackText = normalized && !configuredMatched
+		? ` Configured model ${normalized} was unavailable, so Librarian fell back to auto-selection.`
+		: "";
+	ranked.forEach((model, index) => {
+		const topReason = preferred.length > 0
+			? `Selected a preferred fast Librarian model.${fallbackText}`
+			: `Selected the cheapest available model because no preferred fast Librarian model was available.${fallbackText}`;
+		push(model, {
+			modelRef: modelRef(model),
+			provider: model.provider,
+			modelId: model.id,
+			thinkingLevel,
+			autoSelected: true,
+			selectionReason: index === 0 ? topReason : `Auto-selected ${modelRef(model)} as a lower-ranked fallback.`,
+		});
+	});
 
-	const fallbackText = normalized ? ` Configured model ${normalized} was unavailable, so Librarian fell back to auto-selection.` : "";
-	const selectionReason = preferred.length > 0
-		? `Selected a preferred fast Librarian model.${fallbackText}`
-		: `Selected the cheapest available model because no preferred fast Librarian model was available.${fallbackText}`;
-	return {
-		model: winner,
-		details: {
-			modelRef: modelRef(winner),
-			provider: winner.provider,
-			modelId: winner.id,
+	// Final fallback: the active session model is known to be servable.
+	if (ctx.model) {
+		push(ctx.model, {
+			modelRef: modelRef(ctx.model),
+			provider: ctx.model.provider,
+			modelId: ctx.model.id,
 			thinkingLevel,
 			autoSelected: true,
-			selectionReason,
-		},
-	};
+			selectionReason: `Used the current session model ${modelRef(ctx.model)} as a final fallback.`,
+		});
+	}
+
+	if (candidates.length === 0) {
+		throw new Error("No authenticated models are available for Librarian. Log in or configure an API key first.");
+	}
+
+	return candidates;
 }
 
 function resolveToolPath(cwd: string, rawPath: string): string {
@@ -726,6 +777,13 @@ function isAbortLikeError(error: unknown): boolean {
 	return /aborted|cancelled|canceled/i.test(message);
 }
 
+export const __test__ = {
+	buildLibrarianCandidates,
+	findAvailableModel,
+	isModelAvailabilityError,
+	parseModelPreference,
+};
+
 export default function librarianExtension(pi: ExtensionAPI) {
 	let cachePreference: CacheMode = DEFAULT_CACHE_MODE;
 	let modelPreference: string | undefined;
@@ -944,7 +1002,8 @@ export default function librarianExtension(pi: ExtensionAPI) {
 				parseThinkingLevel((params as { thinkingLevel?: unknown }).thinkingLevel) ??
 				explicitModel.thinkingLevel ??
 				thinkingPreference;
-			const selectedModel = await selectLibrarianModel(ctx, explicitModel.model ?? modelPreference, thinkingLevel);
+			const candidates = await buildLibrarianCandidates(ctx, explicitModel.model ?? modelPreference, thinkingLevel);
+			const selectedModel = candidates[0];
 
 			const workspaceBase = path.join(os.tmpdir(), "pi-librarian");
 			await fs.mkdir(workspaceBase, { recursive: true });
@@ -1045,62 +1104,124 @@ export default function librarianExtension(pi: ExtensionAPI) {
 
 				await resourceLoader.reload();
 
-				const created = await createAgentSession({
-					cwd: workspace,
-					modelRegistry: ctx.modelRegistry,
-					resourceLoader,
-					sessionManager: SessionManager.inMemory(workspace),
-					model: selectedModel.model,
-					thinkingLevel,
-					tools: ["read", "bash"],
-				});
+				const runAttempt = async (
+					candidate: LibrarianCandidate,
+				): Promise<{ answer: string; lastAssistant: AssistantLikeMessage | undefined }> => {
+					details.model = candidate.details;
+					details.turns = 0;
+					details.toolCalls = [];
+					emit(true);
+
+					const created = await createAgentSession({
+						cwd: workspace,
+						modelRegistry: ctx.modelRegistry,
+						resourceLoader,
+						sessionManager: SessionManager.inMemory(workspace),
+						model: candidate.model,
+						thinkingLevel,
+						tools: ["read", "bash"],
+					});
 
-				session = created.session as typeof session;
-				unsubscribe = (created.session as any).subscribe((event: any) => {
-					switch (event.type) {
-						case "turn_end":
-							details.turns += 1;
-							emit();
-							break;
-						case "tool_execution_start":
-							details.toolCalls.push({
-								id: event.toolCallId,
-								name: event.toolName,
-								args: event.args,
-								startedAt: Date.now(),
-							});
-							if (details.toolCalls.length > MAX_TOOL_CALLS_TO_KEEP) {
-								details.toolCalls.splice(0, details.toolCalls.length - MAX_TOOL_CALLS_TO_KEEP);
-							}
-							emit(true);
-							break;
-						case "tool_execution_end": {
-							const call = details.toolCalls.find((item) => item.id === event.toolCallId);
-							if (call) {
-								call.endedAt = Date.now();
-								call.isError = event.isError;
+					session = created.session as typeof session;
+					unsubscribe = (created.session as any).subscribe((event: any) => {
+						switch (event.type) {
+							case "turn_end":
+								details.turns += 1;
+								emit();
+								break;
+							case "tool_execution_start":
+								details.toolCalls.push({
+									id: event.toolCallId,
+									name: event.toolName,
+									args: event.args,
+									startedAt: Date.now(),
+								});
+								if (details.toolCalls.length > MAX_TOOL_CALLS_TO_KEEP) {
+									details.toolCalls.splice(0, details.toolCalls.length - MAX_TOOL_CALLS_TO_KEEP);
+								}
+								emit(true);
+								break;
+							case "tool_execution_end": {
+								const call = details.toolCalls.find((item) => item.id === event.toolCallId);
+								if (call) {
+									call.endedAt = Date.now();
+									call.isError = event.isError;
+								}
+								emit(true);
+								break;
 							}
-							emit(true);
-							break;
 						}
+					});
+
+					try {
+						if (!aborted) {
+							const promptPromise = created.session.prompt(buildUserPrompt(query, repos, owners, maxSearchResults, details.cache), {
+								expandPromptTemplates: false,
+							});
+							const timeoutPromise = new Promise<never>((_resolve, reject) => {
+								runTimeout = setTimeout(() => {
+									abort();
+									reject(new Error(`Librarian timed out after ${Math.round(MAX_RUN_MS / 1000)} seconds.`));
+								}, MAX_RUN_MS);
+							});
+							await Promise.race([promptPromise, timeoutPromise]);
+						}
+
+						const lastAssistant = session ? getLastAssistantMessage(session.state.messages) : undefined;
+						const answer = session ? extractLastAssistantText(session.state.messages) : "";
+						return { answer, lastAssistant };
+					} finally {
+						if (runTimeout) {
+							clearTimeout(runTimeout);
+							runTimeout = undefined;
+						}
+						unsubscribe?.();
+						unsubscribe = undefined;
+						session?.dispose();
+						session = undefined;
 					}
-				});
+				};
 
-				if (!aborted) {
-					const promptPromise = created.session.prompt(buildUserPrompt(query, repos, owners, maxSearchResults, details.cache), {
-						expandPromptTemplates: false,
-					});
-					const timeoutPromise = new Promise<never>((_resolve, reject) => {
-						runTimeout = setTimeout(() => {
-							abort();
-							reject(new Error(`Librarian timed out after ${Math.round(MAX_RUN_MS / 1000)} seconds.`));
-						}, MAX_RUN_MS);
-					});
-					await Promise.race([promptPromise, timeoutPromise]);
+				let answer = "";
+				let lastAssistant: AssistantLikeMessage | undefined;
+				let attemptErrorReason: string | undefined;
+				for (let index = 0; index < candidates.length; index++) {
+					if (aborted) break;
+					const candidate =
+						index === 0 ? candidates[index] : withLibrarianFallbackReason(candidates[index], candidates[index - 1]);
+
+					let attempt: { answer: string; lastAssistant: AssistantLikeMessage | undefined };
+					try {
+						attempt = await runAttempt(candidate);
+					} catch (error) {
+						// Let aborts/timeouts propagate to the outer handler.
+						if (aborted || isAbortLikeError(error)) throw error;
+						// Some providers throw (rather than carry the error on the assistant
+						// message) when a model is unavailable; treat that like a
+						// message-carried availability error and fall back to the next model.
+						const message = error instanceof Error ? error.message : String(error);
+						if (index < candidates.length - 1 && isModelAvailabilityError(message)) {
+							answer = "";
+							lastAssistant = undefined;
+							attemptErrorReason = `the internal subagent stopped with an error: ${message}`;
+							continue;
+						}
+						throw error;
+					}
+
+					answer = attempt.answer;
+					lastAssistant = attempt.lastAssistant;
+					if (answer || aborted) break;
+
+					attemptErrorReason = describeNoAnswerReason(lastAssistant, details.turns);
+					const errorMessage =
+						lastAssistant && lastAssistant.stopReason === "error" && typeof lastAssistant.errorMessage === "string"
+							? lastAssistant.errorMessage
+							: undefined;
+					const canFallBack = index < candidates.length - 1 && isModelAvailabilityError(errorMessage);
+					if (!canFallBack) break;
 				}
 
-				const lastAssistant = session ? getLastAssistantMessage(session.state.messages) : undefined;
-				const answer = session ? extractLastAssistantText(session.state.messages) : "";
 				if (answer) {
 					lastContent = answer;
 					details.status = "done";
@@ -1109,7 +1230,7 @@ export default function librarianExtension(pi: ExtensionAPI) {
 					details.status = "aborted";
 				} else {
 					details.status = "error";
-					const reason = describeNoAnswerReason(lastAssistant, details.turns);
+					const reason = attemptErrorReason ?? describeNoAnswerReason(lastAssistant, details.turns);
 					lastContent = formatInternalFailure(details, reason);
 					details.error = reason;
 				}

package/extensions/librarian/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@diegopetrucci/pi-librarian",
-  "version": "0.1.7",
+  "version": "0.1.9",
   "description": "A pi GitHub research scout with a toggleable local repo checkout cache under the user's OS cache directory.",
   "keywords": [
     "pi-package",

package/extensions/minimal-footer/.pi-fleet-tested-version

@@ -1 +1 @@
-0.79.10
+0.80.3