lootforge 0.3.0

package/dist/providers/nano.js

@@ -0,0 +1,402 @@
+import { mkdir, readFile, writeFile } from "node:fs/promises";
+import path from "node:path";
+import { DEFAULT_PROVIDER_REQUEST_TIMEOUT_MS, fetchWithTimeout, MAX_DECODED_IMAGE_BYTES, normalizeNonNegativeInteger, normalizePositiveInteger, ProviderRequestTimeoutError, toOptionalNonNegativeInteger, withCandidateSuffix, } from "./runtime.js";
+import { createProviderJob, nowIso, ProviderError, PROVIDER_CAPABILITIES, } from "./types.js";
+import { resolvePathWithinRoot } from "../shared/paths.js";
+const GEMINI_API_BASE = "https://generativelanguage.googleapis.com/v1beta/models";
+export const DEFAULT_NANO_MODEL = "gemini-2.5-flash-image";
+const DEFAULT_EDIT_IMAGE_MIME_TYPE = "image/png";
+export class NanoProvider {
+    name = "nano";
+    capabilities;
+    model;
+    apiBase;
+    timeoutMs;
+    maxRetries;
+    constructor(options = {}) {
+        const defaults = PROVIDER_CAPABILITIES.nano;
+        const model = options.model ?? DEFAULT_NANO_MODEL;
+        const supportsEdits = typeof options.supportsEdits === "boolean"
+            ? options.supportsEdits
+            : supportsNanoImageEdits(model);
+        this.capabilities = {
+            ...defaults,
+            supportsEdits,
+            defaultConcurrency: normalizePositiveInteger(options.defaultConcurrency, defaults.defaultConcurrency),
+            minDelayMs: normalizeNonNegativeInteger(options.minDelayMs, defaults.minDelayMs),
+        };
+        this.model = model;
+        this.apiBase = options.apiBase ?? GEMINI_API_BASE;
+        this.timeoutMs = normalizePositiveInteger(options.timeoutMs, DEFAULT_PROVIDER_REQUEST_TIMEOUT_MS);
+        this.maxRetries = toOptionalNonNegativeInteger(options.maxRetries);
+    }
+    prepareJobs(targets, ctx) {
+        return targets.map((target) => createProviderJob({
+            provider: this.name,
+            target,
+            model: target.model ?? this.model,
+            imagesDir: ctx.imagesDir,
+            defaults: {
+                maxRetries: this.maxRetries,
+            },
+        }));
+    }
+    supports(feature) {
+        if (feature === "image-generation")
+            return true;
+        if (feature === "transparent-background")
+            return false;
+        if (feature === "image-edits")
+            return this.capabilities.supportsEdits;
+        if (feature === "multi-candidate")
+            return true;
+        // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition
+        if (feature === "controlnet")
+            return false;
+        return false;
+    }
+    normalizeError(error) {
+        if (error instanceof ProviderError && error.provider === this.name) {
+            return error;
+        }
+        if (error instanceof Error) {
+            return new ProviderError({
+                provider: this.name,
+                code: "nano_request_failed",
+                message: error.message,
+                cause: error,
+                actionable: "Check GEMINI_API_KEY, selected model, and request payload compatibility.",
+            });
+        }
+        return new ProviderError({
+            provider: this.name,
+            code: "nano_request_failed",
+            message: "Nano provider failed with a non-error throwable.",
+            cause: error,
+            actionable: "Check GEMINI_API_KEY, selected model, and request payload compatibility.",
+        });
+    }
+    async runJob(job, ctx) {
+        const startedAt = nowIso(ctx.now);
+        const fetchImpl = ctx.fetchImpl ?? globalThis.fetch;
+        const apiKey = process.env.GEMINI_API_KEY?.trim();
+        const useEdits = this.shouldUseEdits(job);
+        if (!apiKey) {
+            throw new ProviderError({
+                provider: this.name,
+                code: "missing_api_key",
+                message: "GEMINI_API_KEY is required for Nano provider generation.",
+                actionable: "Set GEMINI_API_KEY in the environment and rerun generate.",
+            });
+        }
+        // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition
+        if (!fetchImpl) {
+            throw new ProviderError({
+                provider: this.name,
+                code: "missing_fetch",
+                message: "Global fetch is unavailable for Nano provider.",
+                actionable: "Use Node.js 18+ or pass a fetch implementation in the run context.",
+            });
+        }
+        if (useEdits && !this.capabilities.supportsEdits) {
+            throw new ProviderError({
+                provider: this.name,
+                code: "nano_edit_unsupported_model",
+                message: `Nano provider model "${job.model}" does not advertise edit-first image support.`,
+                actionable: "Use an image-edit-capable Gemini image model (for example gemini-2.5-flash-image) or switch providers for edit-first targets.",
+            });
+        }
+        const endpoint = `${this.apiBase}/${encodeURIComponent(job.model)}:generateContent` +
+            `?key=${encodeURIComponent(apiKey)}`;
+        try {
+            const requestPayload = useEdits
+                ? await this.buildEditRequestPayload(job, ctx)
+                : await this.buildTextRequestPayload(job, ctx);
+            const actualPrompt = this.extractPromptText(requestPayload);
+            const candidateOutputs = [];
+            const count = Math.max(job.candidateCount, 1);
+            for (let index = 0; index < count; index += 1) {
+                const response = await this.requestWithTimeout(fetchImpl, endpoint, {
+                    method: "POST",
+                    headers: {
+                        "Content-Type": "application/json",
+                    },
+                    body: JSON.stringify(requestPayload),
+                });
+                if (!response.ok) {
+                    const bodyText = await response.text();
+                    throw new ProviderError({
+                        provider: this.name,
+                        code: "nano_http_error",
+                        status: response.status,
+                        message: `Nano provider request failed with status ${response.status}.`,
+                        actionable: "Confirm GEMINI_API_KEY access and that the chosen model supports image output.",
+                        cause: bodyText,
+                    });
+                }
+                const payload = (await response.json());
+                const inlineData = this.findInlineImage(payload);
+                if (!inlineData?.data) {
+                    throw new ProviderError({
+                        provider: this.name,
+                        code: "nano_missing_image",
+                        message: "Nano provider response did not include image data. The selected Gemini model may not support image generation for this request.",
+                        actionable: "Use an image-capable Gemini model and ensure generationConfig.responseModalities includes IMAGE.",
+                    });
+                }
+                const imageBytes = Buffer.from(inlineData.data, "base64");
+                if (imageBytes.byteLength === 0) {
+                    throw new ProviderError({
+                        provider: this.name,
+                        code: "nano_empty_image",
+                        message: "Nano provider returned empty image bytes.",
+                        actionable: "Retry with a simpler prompt and verify the model returns inlineData image payloads.",
+                    });
+                }
+                if (imageBytes.byteLength > MAX_DECODED_IMAGE_BYTES) {
+                    throw new ProviderError({
+                        provider: this.name,
+                        code: "nano_image_too_large",
+                        message: `Decoded image exceeds ${MAX_DECODED_IMAGE_BYTES} byte safety limit.`,
+                        actionable: "Request smaller images or adjust model settings.",
+                    });
+                }
+                const outputPath = index === 0 ? job.outPath : withCandidateSuffix(job.outPath, index + 1);
+                await mkdir(path.dirname(outputPath), { recursive: true });
+                await writeFile(outputPath, imageBytes);
+                candidateOutputs.push({
+                    outputPath,
+                    bytesWritten: imageBytes.byteLength,
+                });
+            }
+            return {
+                jobId: job.id,
+                provider: this.name,
+                model: job.model,
+                targetId: job.targetId,
+                prompt: actualPrompt,
+                outputPath: candidateOutputs[0].outputPath,
+                bytesWritten: candidateOutputs[0].bytesWritten,
+                inputHash: job.inputHash,
+                startedAt,
+                finishedAt: nowIso(ctx.now),
+                candidateOutputs,
+            };
+        }
+        catch (error) {
+            throw this.normalizeError(error);
+        }
+    }
+    shouldUseEdits(job) {
+        if (job.target.generationMode !== "edit-first") {
+            return false;
+        }
+        return (job.target.edit?.inputs?.length ?? 0) > 0;
+    }
+    async buildTextRequestPayload(job, ctx) {
+        const parts = [
+            {
+                text: [
+                    job.prompt,
+                    "",
+                    "Output requirements:",
+                    `- image size: ${job.size}`,
+                    `- image format: ${job.outputFormat}`,
+                    `- background: ${job.background}`,
+                ].join("\n"),
+            },
+        ];
+        const styleRefs = job.target.styleReferenceImages ?? [];
+        if (styleRefs.length > 0) {
+            parts.push({ text: "\nStyle references to mimic:\n" });
+            for (const [index, styleRef] of styleRefs.entries()) {
+                const resolvedPath = this.resolveEditInputPath(styleRef, ctx.outDir, job.targetId);
+                const imageBytes = await this.readEditInputBytes(resolvedPath, job.targetId);
+                parts.push({ text: `Style Reference Image ${index + 1}:` });
+                parts.push({
+                    inlineData: {
+                        mimeType: this.mimeTypeForPath(resolvedPath),
+                        data: imageBytes.toString("base64"),
+                    },
+                });
+            }
+        }
+        return {
+            contents: [
+                {
+                    role: "user",
+                    parts,
+                },
+            ],
+            generationConfig: {
+                responseModalities: ["TEXT", "IMAGE"],
+            },
+        };
+    }
+    async buildEditRequestPayload(job, ctx) {
+        const editInputs = job.target.edit?.inputs ?? [];
+        const baseAndReferenceInputs = editInputs.filter((input) => input.role !== "mask");
+        if (baseAndReferenceInputs.length === 0) {
+            throw new ProviderError({
+                provider: this.name,
+                code: "nano_edit_missing_base_image",
+                message: `Target "${job.targetId}" requested edit-first mode but no base/reference inputs were provided.`,
+                actionable: "Add at least one edit input with role base/reference for generationMode=edit-first.",
+            });
+        }
+        const parts = [
+            { text: this.buildEditPrompt(job, editInputs) },
+        ];
+        for (const [index, input] of editInputs.entries()) {
+            const resolvedPath = this.resolveEditInputPath(input.path, ctx.outDir, job.targetId);
+            const imageBytes = await this.readEditInputBytes(resolvedPath, job.targetId);
+            parts.push({
+                text: this.describeEditInput(index + 1, input),
+            });
+            parts.push({
+                inlineData: {
+                    mimeType: this.mimeTypeForPath(resolvedPath),
+                    data: imageBytes.toString("base64"),
+                },
+            });
+        }
+        return {
+            contents: [
+                {
+                    role: "user",
+                    parts,
+                },
+            ],
+            generationConfig: {
+                responseModalities: ["TEXT", "IMAGE"],
+            },
+        };
+    }
+    buildEditPrompt(job, inputs) {
+        const instruction = job.target.edit?.instruction?.trim();
+        const preserveCompositionHint = job.target.edit?.preserveComposition
+            ? "Preserve composition and camera framing unless the instruction explicitly changes them."
+            : "";
+        const hasMask = inputs.some((input) => input.role === "mask");
+        const maskHint = hasMask
+            ? "If a mask input is provided, treat white/opaque regions as editable and preserve dark/transparent regions."
+            : "";
+        if (!instruction) {
+            return [
+                "Use the supplied edit input images to modify the asset.",
+                preserveCompositionHint,
+                maskHint,
+                "Target style/output requirements:",
+                job.prompt,
+            ]
+                .filter(Boolean)
+                .join("\n\n");
+        }
+        return [
+            instruction,
+            preserveCompositionHint,
+            maskHint,
+            "Target style/output requirements:",
+            job.prompt,
+        ]
+            .filter(Boolean)
+            .join("\n\n");
+    }
+    extractPromptText(payload) {
+        for (const content of payload.contents) {
+            for (const part of content.parts) {
+                if ("text" in part && typeof part.text === "string") {
+                    return part.text;
+                }
+            }
+        }
+        return "";
+    }
+    describeEditInput(index, input) {
+        const role = input.role ?? "reference";
+        const fidelity = input.fidelity ?? "medium";
+        if (role === "mask") {
+            return `Input ${index}: role=mask, fidelity=${fidelity}. Use this as the edit mask guide.`;
+        }
+        if (role === "base") {
+            return `Input ${index}: role=base, fidelity=${fidelity}. Preserve identity and composition unless instructed otherwise.`;
+        }
+        return `Input ${index}: role=reference, fidelity=${fidelity}. Use this for style and detail guidance.`;
+    }
+    resolveEditInputPath(inputPath, outDir, targetId) {
+        try {
+            return resolvePathWithinRoot(outDir, inputPath, `edit input path for target "${targetId}"`);
+        }
+        catch (error) {
+            throw new ProviderError({
+                provider: this.name,
+                code: "nano_edit_input_unsafe_path",
+                message: `Unsafe edit input path "${inputPath}" for target "${targetId}".`,
+                actionable: `Edit input paths must stay within the output root (${path.resolve(outDir)}).`,
+                cause: error,
+            });
+        }
+    }
+    async readEditInputBytes(inputPath, targetId) {
+        try {
+            return await readFile(inputPath);
+        }
+        catch (error) {
+            throw new ProviderError({
+                provider: this.name,
+                code: "nano_edit_input_unreadable",
+                message: `Failed to read edit input "${inputPath}" for target "${targetId}".`,
+                actionable: "Ensure edit input files exist and paths are correct relative to --out (or absolute).",
+                cause: error,
+            });
+        }
+    }
+    mimeTypeForPath(filePath) {
+        const ext = path.extname(filePath).toLowerCase();
+        if (ext === ".jpg" || ext === ".jpeg")
+            return "image/jpeg";
+        if (ext === ".webp")
+            return "image/webp";
+        if (ext === ".gif")
+            return "image/gif";
+        if (ext === ".bmp")
+            return "image/bmp";
+        if (ext === ".svg")
+            return "image/svg+xml";
+        return DEFAULT_EDIT_IMAGE_MIME_TYPE;
+    }
+    findInlineImage(payload) {
+        for (const candidate of payload.candidates ?? []) {
+            for (const part of candidate.content?.parts ?? []) {
+                if (part.inlineData?.data) {
+                    return part.inlineData;
+                }
+            }
+        }
+        return undefined;
+    }
+    async requestWithTimeout(fetchImpl, input, init) {
+        try {
+            return await fetchWithTimeout(fetchImpl, input, init, this.timeoutMs);
+        }
+        catch (error) {
+            if (error instanceof ProviderRequestTimeoutError) {
+                throw new ProviderError({
+                    provider: this.name,
+                    code: "nano_request_timeout",
+                    message: `Nano provider request timed out after ${error.timeoutMs}ms.`,
+                    actionable: "Increase providers.nano.timeoutMs or LOOTFORGE_NANO_TIMEOUT_MS and retry.",
+                    cause: error,
+                });
+            }
+            throw error;
+        }
+    }
+}
+export function createNanoProvider(options = {}) {
+    return new NanoProvider(options);
+}
+function supportsNanoImageEdits(model) {
+    return model.toLowerCase().includes("image");
+}
+//# sourceMappingURL=nano.js.map

package/dist/providers/nano.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"nano.js","sourceRoot":"","sources":["../../src/providers/nano.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC9D,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,OAAO,EACL,mCAAmC,EACnC,gBAAgB,EAChB,uBAAuB,EACvB,2BAA2B,EAC3B,wBAAwB,EACxB,2BAA2B,EAC3B,4BAA4B,EAC5B,mBAAmB,GACpB,MAAM,cAAc,CAAC;AACtB,OAAO,EACL,iBAAiB,EAEjB,MAAM,EAIN,aAAa,EAMb,qBAAqB,GAEtB,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAE3D,MAAM,eAAe,GAAG,yDAAyD,CAAC;AAClF,MAAM,CAAC,MAAM,kBAAkB,GAAG,wBAAwB,CAAC;AAC3D,MAAM,4BAA4B,GAAG,WAAW,CAAC;AAiBjD,MAAM,OAAO,YAAY;IACd,IAAI,GAAG,MAAe,CAAC;IACvB,YAAY,CAAuB;IAE3B,KAAK,CAAS;IACd,OAAO,CAAS;IAChB,SAAS,CAAS;IAClB,UAAU,CAAU;IAErC,YAAY,UAA+B,EAAE;QAC3C,MAAM,QAAQ,GAAG,qBAAqB,CAAC,IAAI,CAAC;QAC5C,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,kBAAkB,CAAC;QAClD,MAAM,aAAa,GACjB,OAAO,OAAO,CAAC,aAAa,KAAK,SAAS;YACxC,CAAC,CAAC,OAAO,CAAC,aAAa;YACvB,CAAC,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC;QACpC,IAAI,CAAC,YAAY,GAAG;YAClB,GAAG,QAAQ;YACX,aAAa;YACb,kBAAkB,EAAE,wBAAwB,CAC1C,OAAO,CAAC,kBAAkB,EAC1B,QAAQ,CAAC,kBAAkB,CAC5B;YACD,UAAU,EAAE,2BAA2B,CAAC,OAAO,CAAC,UAAU,EAAE,QAAQ,CAAC,UAAU,CAAC;SACjF,CAAC;QACF,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,eAAe,CAAC;QAClD,IAAI,CAAC,SAAS,GAAG,wBAAwB,CACvC,OAAO,CAAC,SAAS,EACjB,mCAAmC,CACpC,CAAC;QACF,IAAI,CAAC,UAAU,GAAG,4BAA4B,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IACrE,CAAC;IAED,WAAW,CAAC,OAAwB,EAAE,GAA2B;QAC/D,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAC5B,iBAAiB,CAAC;YAChB,QAAQ,EAAE,IAAI,CAAC,IAAI;YACnB,MAAM;YACN,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK;YACjC,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,QAAQ,EAAE;gBACR,UAAU,EAAE,IAAI,CAAC,UAAU;aAC5B;SACF,CAAC,CACH,CAAC;IACJ,CAAC;IAED,QAAQ,CAAC,OAAwB;QAC/B,IAAI,OAAO,KAAK,kBAAkB;YAAE,OAAO,IAAI,CAAC;QAChD,IAAI,OAAO,KAAK,wBAAwB;YAAE,OAAO,KAAK,CAAC;QACvD,IAAI,OAAO,KAAK,aAAa;YAAE,OAAO,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC;QACtE,IAAI,OAAO,KAAK,iBAAiB;YAAE,OAAO,IAAI,CAAC;QAC/C,uEAAuE;QACvE,IAAI,OAAO,KAAK,YAAY;YAAE,OAAO,KAAK,CAAC;QAC3C,OAAO,KAAK,CAAC;IACf,CAAC;IAED,cAAc,CAAC,KAAc;QAC3B,IAAI,KAAK,YAAY,aAAa,IAAI,KAAK,CAAC,QAAQ,KAAK,IAAI,CAAC,IAAI,EAAE,CAAC;YACnE,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;YAC3B,OAAO,IAAI,aAAa,CAAC;gBACvB,QAAQ,EAAE,IAAI,CAAC,IAAI;gBACnB,IAAI,EAAE,qBAAqB;gBAC3B,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,KAAK,EAAE,KAAK;gBACZ,UAAU,EAAE,0EAA0E;aACvF,CAAC,CAAC;QACL,CAAC;QAED,OAAO,IAAI,aAAa,CAAC;YACvB,QAAQ,EAAE,IAAI,CAAC,IAAI;YACnB,IAAI,EAAE,qBAAqB;YAC3B,OAAO,EAAE,kDAAkD;YAC3D,KAAK,EAAE,KAAK;YACZ,UAAU,EAAE,0EAA0E;SACvF,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,GAAgB,EAAE,GAAuB;QACpD,MAAM,SAAS,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAClC,MAAM,SAAS,GAAG,GAAG,CAAC,SAAS,IAAI,UAAU,CAAC,KAAK,CAAC;QACpD,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,IAAI,EAAE,CAAC;QAClD,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;QAE1C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,aAAa,CAAC;gBACtB,QAAQ,EAAE,IAAI,CAAC,IAAI;gBACnB,IAAI,EAAE,iBAAiB;gBACvB,OAAO,EAAE,0DAA0D;gBACnE,UAAU,EAAE,2DAA2D;aACxE,CAAC,CAAC;QACL,CAAC;QAED,uEAAuE;QACvE,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,aAAa,CAAC;gBACtB,QAAQ,EAAE,IAAI,CAAC,IAAI;gBACnB,IAAI,EAAE,eAAe;gBACrB,OAAO,EAAE,gDAAgD;gBACzD,UAAU,EAAE,oEAAoE;aACjF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,QAAQ,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,aAAa,EAAE,CAAC;YACjD,MAAM,IAAI,aAAa,CAAC;gBACtB,QAAQ,EAAE,IAAI,CAAC,IAAI;gBACnB,IAAI,EAAE,6BAA6B;gBACnC,OAAO,EAAE,wBAAwB,GAAG,CAAC,KAAK,gDAAgD;gBAC1F,UAAU,EACR,+HAA+H;aAClI,CAAC,CAAC;QACL,CAAC;QAED,MAAM,QAAQ,GACZ,GAAG,IAAI,CAAC,OAAO,IAAI,kBAAkB,CAAC,GAAG,CAAC,KAAK,CAAC,kBAAkB;YAClE,QAAQ,kBAAkB,CAAC,MAAM,CAAC,EAAE,CAAC;QAEvC,IAAI,CAAC;YACH,MAAM,cAAc,GAAG,QAAQ;gBAC7B,CAAC,CAAC,MAAM,IAAI,CAAC,uBAAuB,CAAC,GAAG,EAAE,GAAG,CAAC;gBAC9C,CAAC,CAAC,MAAM,IAAI,CAAC,uBAAuB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YACjD,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;YAC5D,MAAM,gBAAgB,GAA8B,EAAE,CAAC;YACvD,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;YAC9C,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,KAAK,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;gBAC9C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,SAAS,EAAE,QAAQ,EAAE;oBAClE,MAAM,EAAE,MAAM;oBACd,OAAO,EAAE;wBACP,cAAc,EAAE,kBAAkB;qBACnC;oBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC;iBACrC,CAAC,CAAC;gBAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBACjB,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;oBACvC,MAAM,IAAI,aAAa,CAAC;wBACtB,QAAQ,EAAE,IAAI,CAAC,IAAI;wBACnB,IAAI,EAAE,iBAAiB;wBACvB,MAAM,EAAE,QAAQ,CAAC,MAAM;wBACvB,OAAO,EAAE,4CAA4C,QAAQ,CAAC,MAAM,GAAG;wBACvE,UAAU,EACR,gFAAgF;wBAClF,KAAK,EAAE,QAAQ;qBAChB,CAAC,CAAC;gBACL,CAAC;gBAED,MAAM,OAAO,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAQrC,CAAC;gBAEF,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;gBACjD,IAAI,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC;oBACtB,MAAM,IAAI,aAAa,CAAC;wBACtB,QAAQ,EAAE,IAAI,CAAC,IAAI;wBACnB,IAAI,EAAE,oBAAoB;wBAC1B,OAAO,EACL,iIAAiI;wBACnI,UAAU,EACR,kGAAkG;qBACrG,CAAC,CAAC;gBACL,CAAC;gBAED,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;gBAC1D,IAAI,UAAU,CAAC,UAAU,KAAK,CAAC,EAAE,CAAC;oBAChC,MAAM,IAAI,aAAa,CAAC;wBACtB,QAAQ,EAAE,IAAI,CAAC,IAAI;wBACnB,IAAI,EAAE,kBAAkB;wBACxB,OAAO,EAAE,2CAA2C;wBACpD,UAAU,EACR,qFAAqF;qBACxF,CAAC,CAAC;gBACL,CAAC;gBAED,IAAI,UAAU,CAAC,UAAU,GAAG,uBAAuB,EAAE,CAAC;oBACpD,MAAM,IAAI,aAAa,CAAC;wBACtB,QAAQ,EAAE,IAAI,CAAC,IAAI;wBACnB,IAAI,EAAE,sBAAsB;wBAC5B,OAAO,EAAE,yBAAyB,uBAAuB,qBAAqB;wBAC9E,UAAU,EAAE,kDAAkD;qBAC/D,CAAC,CAAC;gBACL,CAAC;gBAED,MAAM,UAAU,GAAG,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,mBAAmB,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;gBAC3F,MAAM,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC3D,MAAM,SAAS,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;gBACxC,gBAAgB,CAAC,IAAI,CAAC;oBACpB,UAAU;oBACV,YAAY,EAAE,UAAU,CAAC,UAAU;iBACpC,CAAC,CAAC;YACL,CAAC;YAED,OAAO;gBACL,KAAK,EAAE,GAAG,CAAC,EAAE;gBACb,QAAQ,EAAE,IAAI,CAAC,IAAI;gBACnB,KAAK,EAAE,GAAG,CAAC,KAAK;gBAChB,QAAQ,EAAE,GAAG,CAAC,QAAQ;gBACtB,MAAM,EAAE,YAAY;gBACpB,UAAU,EAAE,gBAAgB,CAAC,CAAC,CAAC,CAAC,UAAU;gBAC1C,YAAY,EAAE,gBAAgB,CAAC,CAAC,CAAC,CAAC,YAAY;gBAC9C,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,SAAS;gBACT,UAAU,EAAE,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC;gBAC3B,gBAAgB;aACjB,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;IAEO,cAAc,CAAC,GAAgB;QACrC,IAAI,GAAG,CAAC,MAAM,CAAC,cAAc,KAAK,YAAY,EAAE,CAAC;YAC/C,OAAO,KAAK,CAAC;QACf,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;IACpD,CAAC;IAEO,KAAK,CAAC,uBAAuB,CACnC,GAAgB,EAChB,GAAuB;QAQvB,MAAM,KAAK,GAA8E;YACvF;gBACE,IAAI,EAAE;oBACJ,GAAG,CAAC,MAAM;oBACV,EAAE;oBACF,sBAAsB;oBACtB,iBAAiB,GAAG,CAAC,IAAI,EAAE;oBAC3B,mBAAmB,GAAG,CAAC,YAAY,EAAE;oBACrC,iBAAiB,GAAG,CAAC,UAAU,EAAE;iBAClC,CAAC,IAAI,CAAC,IAAI,CAAC;aACb;SACF,CAAC;QAEF,MAAM,SAAS,GAAG,GAAG,CAAC,MAAM,CAAC,oBAAoB,IAAI,EAAE,CAAC;QACxD,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gCAAgC,EAAE,CAAC,CAAC;YACvD,KAAK,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,SAAS,CAAC,OAAO,EAAE,EAAE,CAAC;gBACpD,MAAM,YAAY,GAAG,IAAI,CAAC,oBAAoB,CAAC,QAAQ,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC;gBACnF,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,YAAY,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC;gBAC7E,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,yBAAyB,KAAK,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC;gBAC5D,KAAK,CAAC,IAAI,CAAC;oBACT,UAAU,EAAE;wBACV,QAAQ,EAAE,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC;wBAC5C,IAAI,EAAE,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC;qBACpC;iBACF,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO;YACL,QAAQ,EAAE;gBACR;oBACE,IAAI,EAAE,MAAM;oBACZ,KAAK;iBACN;aACF;YACD,gBAAgB,EAAE;gBAChB,kBAAkB,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC;aACtC;SACF,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,uBAAuB,CACnC,GAAgB,EAChB,GAAuB;QAQvB,MAAM,UAAU,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,IAAI,EAAE,CAAC;QACjD,MAAM,sBAAsB,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;QAEnF,IAAI,sBAAsB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxC,MAAM,IAAI,aAAa,CAAC;gBACtB,QAAQ,EAAE,IAAI,CAAC,IAAI;gBACnB,IAAI,EAAE,8BAA8B;gBACpC,OAAO,EAAE,WAAW,GAAG,CAAC,QAAQ,yEAAyE;gBACzG,UAAU,EACR,qFAAqF;aACxF,CAAC,CAAC;QACL,CAAC;QAED,MAAM,KAAK,GAA8E;YACvF,EAAE,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,UAAU,CAAC,EAAE;SAChD,CAAC;QAEF,KAAK,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC;YAClD,MAAM,YAAY,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,IAAI,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC;YACrF,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,YAAY,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC7E,KAAK,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,IAAI,CAAC,iBAAiB,CAAC,KAAK,GAAG,CAAC,EAAE,KAAK,CAAC;aAC/C,CAAC,CAAC;YACH,KAAK,CAAC,IAAI,CAAC;gBACT,UAAU,EAAE;oBACV,QAAQ,EAAE,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC;oBAC5C,IAAI,EAAE,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC;iBACpC;aACF,CAAC,CAAC;QACL,CAAC;QAED,OAAO;YACL,QAAQ,EAAE;gBACR;oBACE,IAAI,EAAE,MAAM;oBACZ,KAAK;iBACN;aACF;YACD,gBAAgB,EAAE;gBAChB,kBAAkB,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC;aACtC;SACF,CAAC;IACJ,CAAC;IAEO,eAAe,CAAC,GAAgB,EAAE,MAAyB;QACjE,MAAM,WAAW,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;QACzD,MAAM,uBAAuB,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,mBAAmB;YAClE,CAAC,CAAC,yFAAyF;YAC3F,CAAC,CAAC,EAAE,CAAC;QACP,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;QAC9D,MAAM,QAAQ,GAAG,OAAO;YACtB,CAAC,CAAC,4GAA4G;YAC9G,CAAC,CAAC,EAAE,CAAC;QAEP,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,OAAO;gBACL,yDAAyD;gBACzD,uBAAuB;gBACvB,QAAQ;gBACR,mCAAmC;gBACnC,GAAG,CAAC,MAAM;aACX;iBACE,MAAM,CAAC,OAAO,CAAC;iBACf,IAAI,CAAC,MAAM,CAAC,CAAC;QAClB,CAAC;QAED,OAAO;YACL,WAAW;YACX,uBAAuB;YACvB,QAAQ;YACR,mCAAmC;YACnC,GAAG,CAAC,MAAM;SACX;aACE,MAAM,CAAC,OAAO,CAAC;aACf,IAAI,CAAC,MAAM,CAAC,CAAC;IAClB,CAAC;IAEO,iBAAiB,CAAC,OAEzB;QACC,KAAK,MAAM,OAAO,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACvC,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;gBACjC,IAAI,MAAM,IAAI,IAAI,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;oBACpD,OAAO,IAAI,CAAC,IAAI,CAAC;gBACnB,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC;IAEO,iBAAiB,CAAC,KAAa,EAAE,KAAsB;QAC7D,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,IAAI,WAAW,CAAC;QACvC,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,IAAI,QAAQ,CAAC;QAC5C,IAAI,IAAI,KAAK,MAAM,EAAE,CAAC;YACpB,OAAO,SAAS,KAAK,yBAAyB,QAAQ,oCAAoC,CAAC;QAC7F,CAAC;QACD,IAAI,IAAI,KAAK,MAAM,EAAE,CAAC;YACpB,OAAO,SAAS,KAAK,yBAAyB,QAAQ,kEAAkE,CAAC;QAC3H,CAAC;QACD,OAAO,SAAS,KAAK,8BAA8B,QAAQ,2CAA2C,CAAC;IACzG,CAAC;IAEO,oBAAoB,CAAC,SAAiB,EAAE,MAAc,EAAE,QAAgB;QAC9E,IAAI,CAAC;YACH,OAAO,qBAAqB,CAAC,MAAM,EAAE,SAAS,EAAE,+BAA+B,QAAQ,GAAG,CAAC,CAAC;QAC9F,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,aAAa,CAAC;gBACtB,QAAQ,EAAE,IAAI,CAAC,IAAI;gBACnB,IAAI,EAAE,6BAA6B;gBACnC,OAAO,EAAE,2BAA2B,SAAS,iBAAiB,QAAQ,IAAI;gBAC1E,UAAU,EAAE,sDAAsD,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI;gBAC1F,KAAK,EAAE,KAAK;aACb,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAAC,SAAiB,EAAE,QAAgB;QAClE,IAAI,CAAC;YACH,OAAO,MAAM,QAAQ,CAAC,SAAS,CAAC,CAAC;QACnC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,aAAa,CAAC;gBACtB,QAAQ,EAAE,IAAI,CAAC,IAAI;gBACnB,IAAI,EAAE,4BAA4B;gBAClC,OAAO,EAAE,8BAA8B,SAAS,iBAAiB,QAAQ,IAAI;gBAC7E,UAAU,EACR,sFAAsF;gBACxF,KAAK,EAAE,KAAK;aACb,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,eAAe,CAAC,QAAgB;QACtC,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;QACjD,IAAI,GAAG,KAAK,MAAM,IAAI,GAAG,KAAK,OAAO;YAAE,OAAO,YAAY,CAAC;QAC3D,IAAI,GAAG,KAAK,OAAO;YAAE,OAAO,YAAY,CAAC;QACzC,IAAI,GAAG,KAAK,MAAM;YAAE,OAAO,WAAW,CAAC;QACvC,IAAI,GAAG,KAAK,MAAM;YAAE,OAAO,WAAW,CAAC;QACvC,IAAI,GAAG,KAAK,MAAM;YAAE,OAAO,eAAe,CAAC;QAC3C,OAAO,4BAA4B,CAAC;IACtC,CAAC;IAEO,eAAe,CAAC,OAIvB;QACC,KAAK,MAAM,SAAS,IAAI,OAAO,CAAC,UAAU,IAAI,EAAE,EAAE,CAAC;YACjD,KAAK,MAAM,IAAI,IAAI,SAAS,CAAC,OAAO,EAAE,KAAK,IAAI,EAAE,EAAE,CAAC;gBAClD,IAAI,IAAI,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC;oBAC1B,OAAO,IAAI,CAAC,UAAU,CAAC;gBACzB,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAC9B,SAAuB,EACvB,KAAkC,EAClC,IAAkB;QAElB,IAAI,CAAC;YACH,OAAO,MAAM,gBAAgB,CAAC,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QACxE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,2BAA2B,EAAE,CAAC;gBACjD,MAAM,IAAI,aAAa,CAAC;oBACtB,QAAQ,EAAE,IAAI,CAAC,IAAI;oBACnB,IAAI,EAAE,sBAAsB;oBAC5B,OAAO,EAAE,yCAAyC,KAAK,CAAC,SAAS,KAAK;oBACtE,UAAU,EAAE,2EAA2E;oBACvF,KAAK,EAAE,KAAK;iBACb,CAAC,CAAC;YACL,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF;AAED,MAAM,UAAU,kBAAkB,CAAC,UAA+B,EAAE;IAClE,OAAO,IAAI,YAAY,CAAC,OAAO,CAAC,CAAC;AACnC,CAAC;AAED,SAAS,sBAAsB,CAAC,KAAa;IAC3C,OAAO,KAAK,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;AAC/C,CAAC"}

package/dist/providers/openai.d.ts

@@ -0,0 +1,37 @@
+import { type GenerationProvider, type PlannedTarget, type ProviderCapabilities, ProviderError, type ProviderFeature, type ProviderJob, type ProviderPrepareContext, type ProviderRunContext, type ProviderRunResult } from "./types.js";
+export declare const DEFAULT_OPENAI_MODEL = "gpt-image-1";
+export interface OpenAIProviderOptions {
+    model?: string;
+    endpoint?: string;
+    editsEndpoint?: string;
+    timeoutMs?: number;
+    maxRetries?: number;
+    minDelayMs?: number;
+    defaultConcurrency?: number;
+}
+export declare class OpenAIProvider implements GenerationProvider {
+    readonly name: "openai";
+    readonly capabilities: ProviderCapabilities;
+    private readonly model;
+    private readonly endpoint;
+    private readonly editsEndpoint;
+    private readonly timeoutMs;
+    private readonly maxRetries?;
+    constructor(options?: OpenAIProviderOptions);
+    prepareJobs(targets: PlannedTarget[], ctx: ProviderPrepareContext): ProviderJob[];
+    supports(feature: ProviderFeature): boolean;
+    normalizeError(error: unknown): ProviderError;
+    runJob(job: ProviderJob, ctx: ProviderRunContext): Promise<ProviderRunResult>;
+    private shouldUseEdits;
+    private runEditRequest;
+    private resolveEditInputPath;
+    private readEditInputBytes;
+    private buildEditPrompt;
+    private mimeTypeForPath;
+    private resolveImageBytes;
+    private buildHttpError;
+    private parseErrorPayload;
+    private httpErrorActionable;
+    private requestWithTimeout;
+}
+export declare function createOpenAIProvider(options?: OpenAIProviderOptions): OpenAIProvider;