pi-banana 2.5.0 → 2.5.1

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,
@@ -250,6 +250,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",

package/extensions/wizard.ts

@@ -20,6 +20,53 @@ 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(/max|high|pro|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.1",
   "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",