openclaw/dist/image-generation-provider-Cx-aRUzz.mjs

Multi-channel AI gateway with extensible messaging integrations

import { l as normalizeOptionalString } from "./string-coerce-1VuiLaaR.mjs";
import "./string-coerce-runtime-50qKeWeL.mjs";
import { t as bufferToBlobPart } from "./blob-runtime-CBGGTQ96.mjs";
import { a as imageSourceUploadFileName } from "./image-assets-DT0442If.mjs";
import { t as createOpenAiCompatibleImageGenerationProvider } from "./image-generation-CltF7hkb.mjs";
import { t as LITELLM_BASE_URL } from "./onboard-BYfizjKa.mjs";
import { isIP } from "node:net";
//#region extensions/litellm/image-generation-provider.ts
const DEFAULT_SIZE = "1024x1024";
const DEFAULT_LITELLM_IMAGE_MODEL = "gpt-image-2";
const LITELLM_SUPPORTED_SIZES = [
	"256x256",
	"512x512",
	"1024x1024",
	"1024x1536",
	"1024x1792",
	"1536x1024",
	"1792x1024",
	"2048x2048",
	"2048x1152",
	"3840x2160",
	"2160x3840"
];
const LITELLM_MAX_INPUT_IMAGES = 5;
function shouldAutoAllowPrivateLitellmEndpoint(baseUrl) {
	const url = URL.parse(baseUrl);
	if (!url || url.protocol !== "http:" && url.protocol !== "https:") return false;
	const { hostname } = url;
	return hostname === "localhost" || hostname === "host.docker.internal" || hostname.endsWith(".localhost") || hostname === "[::1]" || isIP(hostname) === 4 && hostname.startsWith("127.");
}
function buildLitellmImageGenerationProvider() {
	return createOpenAiCompatibleImageGenerationProvider({
		id: "litellm",
		label: "LiteLLM",
		defaultModel: DEFAULT_LITELLM_IMAGE_MODEL,
		models: [DEFAULT_LITELLM_IMAGE_MODEL],
		capabilities: {
			generate: {
				maxCount: 4,
				supportsSize: true,
				supportsAspectRatio: false,
				supportsResolution: false
			},
			edit: {
				enabled: true,
				maxCount: 4,
				maxInputImages: LITELLM_MAX_INPUT_IMAGES,
				supportsSize: true,
				supportsAspectRatio: false,
				supportsResolution: false
			},
			geometry: { sizes: [...LITELLM_SUPPORTED_SIZES] }
		},
		defaultBaseUrl: LITELLM_BASE_URL,
		resolveAllowPrivateNetwork: ({ baseUrl }) => shouldAutoAllowPrivateLitellmEndpoint(baseUrl) ? true : void 0,
		useConfiguredRequest: true,
		buildGenerateRequest: ({ req, model, count }) => ({
			kind: "json",
			body: {
				model,
				prompt: req.prompt,
				n: count,
				size: req.size ?? DEFAULT_SIZE
			}
		}),
		buildEditRequest: ({ req, inputImages, model, count }) => {
			const form = new FormData();
			form.set("model", model);
			form.set("prompt", req.prompt);
			form.set("n", String(count));
			form.set("size", req.size ?? DEFAULT_SIZE);
			const partName = inputImages.length > 1 ? "image[]" : "image";
			for (const [index, image] of inputImages.entries()) {
				const mimeType = normalizeOptionalString(image.mimeType) ?? "image/png";
				form.append(partName, new Blob([bufferToBlobPart(image.buffer)], { type: mimeType }), imageSourceUploadFileName({
					image,
					index
				}));
			}
			return {
				kind: "multipart",
				form
			};
		},
		missingApiKeyError: "LiteLLM API key missing",
		failureLabels: {
			generate: "LiteLLM image generation failed",
			edit: "LiteLLM image edit failed"
		}
	});
}
//#endregion
export { buildLitellmImageGenerationProvider as t };