genkitx-azure-openai/lib/dalle.js

Genkit AI framework plugin for Azure OpenAI APIs.

"use strict";
var __defProp = Object.defineProperty;
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
var __getOwnPropNames = Object.getOwnPropertyNames;
var __hasOwnProp = Object.prototype.hasOwnProperty;
var __export = (target, all) => {
  for (var name in all)
    __defProp(target, name, { get: all[name], enumerable: true });
};
var __copyProps = (to, from, except, desc) => {
  if (from && typeof from === "object" || typeof from === "function") {
    for (let key of __getOwnPropNames(from))
      if (!__hasOwnProp.call(to, key) && key !== except)
        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
  }
  return to;
};
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
var dalle_exports = {};
__export(dalle_exports, {
  DallE3ConfigSchema: () => DallE3ConfigSchema,
  dallE3: () => dallE3,
  dallE3Model: () => dallE3Model
});
module.exports = __toCommonJS(dalle_exports);
var import_genkit = require("genkit");
var import_model = require("genkit/model");
const DallE3ConfigSchema = import_genkit.GenerationCommonConfigSchema.extend({
  size: import_genkit.z.enum(["1024x1024", "1792x1024", "1024x1792"]).optional(),
  style: import_genkit.z.enum(["vivid", "natural"]).optional(),
  user: import_genkit.z.string().optional(),
  quality: import_genkit.z.enum(["standard", "hd"]).optional(),
  response_format: import_genkit.z.enum(["b64_json", "url"]).optional()
});
const dallE3 = (0, import_model.modelRef)({
  name: "azure-openai/dall-e-3",
  info: {
    label: "OpenAI - DALL-E 3",
    supports: {
      media: false,
      output: ["media"],
      multiturn: false,
      tools: false,
      systemRole: false
    }
  },
  configSchema: DallE3ConfigSchema
});
function toDallE3Request(request) {
  const options = {
    model: "dall-e-3",
    prompt: new import_genkit.Message(request.messages[0]).text,
    n: request.candidates || 1,
    size: request.config?.custom?.size,
    style: request.config?.custom?.style,
    user: request.config?.custom?.user,
    quality: request.config?.custom?.quality,
    response_format: request.config?.custom?.response_format || "b64_json"
  };
  for (const k in options) {
    if (options[k] === void 0) {
      delete options[k];
    }
  }
  return options;
}
function toGenerateResponse(result) {
  const candidates = (result.data || []).map((image, index) => ({
    index,
    finishReason: "stop",
    custom: { revisedPrompt: image.revised_prompt },
    message: {
      role: "model",
      content: [
        {
          media: {
            contentType: "image/png",
            url: image.url || `data:image/png;base64,${image.b64_json}`
          }
        }
      ]
    }
  }));
  return { candidates };
}
function dallE3Model(ai, client) {
  return ai.defineModel(
    {
      name: dallE3.name,
      ...dallE3.info,
      configSchema: dallE3.configSchema
    },
    async (request) => {
      const result = await client.images.generate(toDallE3Request(request));
      return toGenerateResponse(result);
    }
  );
}
// Annotate the CommonJS export names for ESM import in node:
0 && (module.exports = {
  DallE3ConfigSchema,
  dallE3,
  dallE3Model
});
//# sourceMappingURL=dalle.js.map