@genkit-ai/compat-oai/lib/image.mjs

Genkit AI framework plugin for OpenAI APIs.

import { Message, modelRef, z } from "genkit";
const IMAGE_GENERATION_MODEL_INFO = {
  supports: {
    media: false,
    output: ["media"],
    multiturn: false,
    systemRole: false,
    tools: false
  }
};
const ImageGenerationCommonConfigSchema = z.object({
  size: z.enum(["1024x1024", "1792x1024", "1024x1792"]).optional(),
  style: z.enum(["vivid", "natural"]).optional(),
  user: z.string().optional(),
  n: z.number().int().min(1).max(10).default(1),
  quality: z.enum(["standard", "hd"]).optional(),
  response_format: z.enum(["b64_json", "url"]).default("b64_json").optional()
});
function toImageGenerateParams(modelName, request, requestBuilder) {
  const {
    temperature,
    version: modelVersion,
    maxOutputTokens,
    stopSequences,
    topK,
    topP,
    response_format,
    ...restOfConfig
  } = request.config ?? {};
  let options = {
    model: modelVersion ?? modelName,
    prompt: new Message(request.messages[0]).text,
    response_format: response_format || "b64_json"
  };
  if (requestBuilder) {
    requestBuilder(request, options);
  } else {
    options = { ...options, ...restOfConfig };
  }
  for (const k in options) {
    if (options[k] === void 0) {
      delete options[k];
    }
  }
  return options;
}
function toGenerateResponse(result) {
  const images = result.data;
  if (!images) {
    return { finishReason: "stop" };
  } else {
    const content = (result.data ?? []).map((image) => ({
      media: {
        contentType: "image/png",
        url: image.url || `data:image/png;base64,${image.b64_json}`
      }
    }));
    return { message: { role: "model", content }, raw: result };
  }
}
function defineCompatOpenAIImageModel(params) {
  const { ai, name, client, modelRef: modelRef2, requestBuilder } = params;
  const modelName = name.substring(name.indexOf("/") + 1);
  return ai.defineModel(
    {
      name,
      ...modelRef2?.info,
      apiVersion: "v2",
      configSchema: modelRef2?.configSchema
    },
    async (request, { abortSignal }) => {
      const result = await client.images.generate(
        toImageGenerateParams(modelName, request, requestBuilder),
        { signal: abortSignal }
      );
      return toGenerateResponse(result);
    }
  );
}
function compatOaiImageModelRef(params) {
  const {
    name,
    info = IMAGE_GENERATION_MODEL_INFO,
    configSchema,
    config = void 0
  } = params;
  return modelRef({
    name,
    configSchema: configSchema || ImageGenerationCommonConfigSchema,
    info,
    config
  });
}
export {
  IMAGE_GENERATION_MODEL_INFO,
  ImageGenerationCommonConfigSchema,
  compatOaiImageModelRef,
  defineCompatOpenAIImageModel
};
//# sourceMappingURL=image.mjs.map