pi-banana 2.5.0 → 2.5.2

package/extensions/index.ts

@@ -20,7 +20,7 @@
 
 import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
 import { resolveProvider, openaiGenerateImage, openaiEditImage, openaiVision, sizeFor } from "./providers.ts";
-import { runBananaSetup } from "./wizard.ts";
+import { runBananaSetup, configureBananaProvider } from "./wizard.ts";
 import { StringEnum } from "@earendil-works/pi-ai";
 import {
 	Box,
@@ -53,7 +53,7 @@ const ASPECT_RATIOS = [
 	"21:9",
 ] as const;
 const IMAGE_SIZES = ["1K", "2K", "4K"] as const;
-const IMAGE_QUALITY = ["lite", "fast", "high"] as const;
+const IMAGE_QUALITY = ["lite", "fast", "high", "max"] as const;
 const VISION_QUALITY = ["fast", "high"] as const;
 type ImageQuality = (typeof IMAGE_QUALITY)[number];
 
@@ -67,6 +67,7 @@ const MODEL_FOR_QUALITY: Record<ImageQuality, string> = {
 	lite: "gemini-3.1-flash-lite-image", // Nano Banana 2 Lite (ultra-low latency, cheapest)
 	fast: "gemini-3.1-flash-image", // Nano Banana 2
 	high: "gemini-3-pro-image", // Nano Banana Pro
+	max: "gemini-3-pro-image", // Google fallback; configured provider maps max to GPT Image 2
 };
 
 const DEFAULT_OUTPUT_DIR =
@@ -250,6 +251,29 @@ export default function (pi: ExtensionAPI) {
 		},
 	});
 
+	pi.registerTool({
+		name: "banana_configure",
+		label: "Banana Configure",
+		description:
+			"Configure banana to generate images through an OpenAI-compatible provider (mantice) instead of Google. " +
+			"Probes the endpoint, auto-selects image models per quality tier, and saves the configuration. " +
+			"Call this when the user asks to set up or switch the image provider.",
+		promptSnippet: "Configure the banana image provider (mantice) non-interactively.",
+		parameters: Type.Object({
+			baseUrl: Type.String({ description: "OpenAI-compatible base URL, e.g. https://llm.fornace.net/v1" }),
+			apiKey: Type.Optional(Type.String({
+				description: "API key. Omit to reuse the mantice/google key already stored in pi credentials.",
+			})),
+			visionModel: Type.Optional(Type.String({
+				description: "Model for image analysis. Omit to auto-detect (e.g. fornace-vision).",
+			})),
+		}),
+		async execute(_toolCallId, params, _signal, _onUpdate, ctx) {
+			const report = await configureBananaProvider(ctx, params);
+			return { content: [{ type: "text", text: report }], details: {} };
+		},
+	});
+
 	pi.registerTool({
 		name: "banana_image",
 		label: "Banana Image",
@@ -288,7 +312,7 @@ export default function (pi: ExtensionAPI) {
 			quality: Type.Optional(
 				StringEnum(IMAGE_QUALITY, {
 					description:
-						"'lite' = Nano Banana 2 Lite (ultra-low latency, cheapest). 'fast' = Nano Banana 2 (default, ~3s, cheap). 'high' = Nano Banana Pro (slower, top quality).",
+						"'lite' = Nano Banana 2 Lite (ultra-low latency, cheapest). 'fast' = Nano Banana 2 (default, ~3s, cheap). 'high' = Nano Banana Pro (strong editing). 'max' = premium benchmark tier, GPT Image 2 when the configured provider supports it.",
 					default: DEFAULT_QUALITY,
 				}),
 			),
@@ -339,7 +363,7 @@ export default function (pi: ExtensionAPI) {
 			const provider = await resolveProvider(ctx);
 			if (provider.kind === "openai-compat") {
 				const { config } = provider;
-				const oaiModel = config.models[quality] ?? config.models.fast;
+				const oaiModel = config.models[quality] ?? config.models.high ?? config.models.fast;
 				if (!oaiModel) {
 					throw new Error(`banana: no model mapped for quality "${quality}". Run /banana-setup.`);
 				}
@@ -392,7 +416,7 @@ export default function (pi: ExtensionAPI) {
 				content: [
 					{
 						type: "text",
-						text: `🎨 ${verbLabel} ${aspectRatio} ${imageSize} image with ${quality === "lite" ? "Nano Banana 2 Lite" : quality === "fast" ? "Nano Banana 2" : "Nano Banana Pro"}…`,
+						text: `🎨 ${verbLabel} ${aspectRatio} ${imageSize} image with ${quality === "lite" ? "Nano Banana 2 Lite" : quality === "fast" ? "Nano Banana 2" : quality === "high" ? "Nano Banana Pro" : "GPT Image 2"}…`,
 					},
 				],
 				details: { model, aspectRatio, imageSize, quality, editing },

package/extensions/providers.ts

@@ -18,7 +18,7 @@ import { resolve } from "path";
 export interface OpenAiCompatConfig {
 	baseUrl: string;
 	apiKey: string;
-	models: { lite?: string; fast?: string; high?: string };
+	models: { lite?: string; fast?: string; high?: string; max?: string };
 	visionModel?: string;
 	/** Optional provider id whose key lives in pi's auth.json (e.g. "mantice"). */
 	providerName?: string;

package/extensions/wizard.ts

@@ -20,6 +20,54 @@ function pickDefault(candidates: string[], hint: RegExp): string | undefined {
 	return candidates.find((m) => hint.test(m)) ?? candidates[0];
 }
 
+export interface ConfigureOptions {
+	baseUrl: string;
+	apiKey?: string;
+	visionModel?: string;
+}
+
+/** Non-interactive provider configuration: resolves the key, probes the
+ * endpoint, auto-assigns quality tiers, and persists to settings.json.
+ * Returns a human-readable report. Used by banana_configure and the wizard. */
+export async function configureBananaProvider(ctx: any, opts: ConfigureOptions): Promise<string> {
+	const baseUrl = opts.baseUrl.replace(/\/+$/, "");
+	let apiKey = opts.apiKey;
+	if (!apiKey && ctx?.modelRegistry) {
+		for (const p of ["mantice", "google"]) {
+			try {
+				apiKey = await ctx.modelRegistry.getApiKeyForProvider(p);
+				if (apiKey) break;
+			} catch { /* next */ }
+		}
+	}
+	if (!apiKey) apiKey = process.env.MANTICE_API_KEY;
+	if (!apiKey) throw new Error("No API key found: pass apiKey, log in via /login, or set MANTICE_API_KEY.");
+
+	const probe = await probeCapabilities(baseUrl, apiKey);
+	if (!probe.supportsImageGen || probe.imageModels.length === 0) {
+		throw new Error("Endpoint has no image generation models. Nothing was saved.");
+	}
+	const pick = (hint: RegExp, fallback?: string) =>
+		probe.imageModels.find((m) => hint.test(m)) ?? fallback ?? probe.imageModels[0];
+	const models = {
+		lite: pick(/lite|fast|turbo|mini/i),
+		fast: pick(/^(?!.*(lite|high|max|pro|ultra)).*$/i),
+		high: pick(/high|pro/i),
+		max: pick(/max|ultra/i),
+	};
+	const visionModel = opts.visionModel ?? probe.visionModels[0];
+	persistBananaConfig({ baseUrl, apiKey, models, ...(visionModel ? { visionModel } : {}) });
+	return [
+		`Configured banana on ${baseUrl}:`,
+		`  lite  = ${models.lite}`,
+		`  fast  = ${models.fast}`,
+		`  high  = ${models.high}`,
+		visionModel ? `  vision = ${visionModel}` : "  vision = (google default kept)",
+		`Image edit endpoint: ${probe.supportsImageEdit ? "available" : "not available"}.`,
+		"banana_image and banana_vision now use this provider.",
+	].join("\n");
+}
+
 export async function runBananaSetup(ctx: any): Promise<void> {
 	const ui = ctx.ui;
 	ui.notify("Banana setup: configure an OpenAI-compatible image provider (mantice works).", "info");

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-banana",
-  "version": "2.5.0",
+  "version": "2.5.2",
   "description": "Generate, edit, and analyze images in pi using Google Nano Banana (image gen) and Gemini Vision (analysis). Inline terminal preview, reference-image editing, auto-save.",
   "author": "Francesco Frapporti <effedue@gmail.com>",
   "license": "MIT",