@genkit-ai/compat-oai
Version:
Genkit AI framework plugin for OpenAI APIs.
48 lines • 1.44 kB
JavaScript
import { z } from "genkit";
import {
ImageGenerationCommonConfigSchema,
compatOaiImageModelRef as openAIImageModelRef
} from "../image";
const SUPPORTED_IMAGE_MODELS = {
"dall-e-3": openAIImageModelRef({ name: "openai/dall-e-3" }),
"gpt-image-1": openAIImageModelRef({
name: "openai/gpt-image-1",
configSchema: ImageGenerationCommonConfigSchema.omit({
response_format: true
}).extend({
size: z.enum(["1024x1024", "1536x1024", "1024x1536", "auto"]).optional(),
background: z.enum(["transparent", "opaque", "auto"]).optional(),
moderation: z.enum(["low", "auto"]).optional(),
output_compression: z.number().int().min(1).max(100).optional(),
output_format: z.enum(["png", "jpeg", "web"]).optional(),
quality: z.enum(["low", "medium", "high"]).optional()
})
})
};
const gptImage1RequestBuilder = (req, params) => {
const {
background,
moderation,
n,
output_compression,
output_format,
quality,
style,
user
} = req.config ?? {};
params.response_format = void 0;
params.background = background;
params.moderation = moderation;
params.n = n;
params.output_compression = output_compression;
params.output_format = output_format;
params.quality = quality;
params.style = style;
params.user = user;
};
export {
SUPPORTED_IMAGE_MODELS,
gptImage1RequestBuilder,
openAIImageModelRef
};
//# sourceMappingURL=dalle.mjs.map