genkitx-openai/lib/index.mjs

Firebase Genkit AI framework plugin for OpenAI APIs.

import {
  __async
} from "./chunk-MLCSNVBT.mjs";
import { genkitPlugin } from "genkit/plugin";
import { OpenAI } from "openai";
import { dallE3, dallE3Model } from "./dalle.js";
import {
  openaiEmbedder,
  SUPPORTED_EMBEDDING_MODELS,
  textEmbedding3Large,
  textEmbedding3Small,
  textEmbeddingAda002
} from "./embedder.js";
import {
  gpt35Turbo,
  gpt4,
  gpt41,
  gpt41Mini,
  gpt41Nano,
  gpt45,
  gpt4o,
  gpt4oMini,
  gpt4Turbo,
  gpt4Vision,
  gptModel,
  o1,
  o1Mini,
  o1Preview,
  o3,
  o3Mini,
  o4Mini,
  SUPPORTED_GPT_MODELS
} from "./gpt.js";
import {
  SUPPORTED_TTS_MODELS,
  tts1,
  tts1Hd,
  gpt4oMiniTts,
  ttsModel
} from "./tts.js";
import {
  whisper1,
  gpt4oTranscribe,
  SUPPORTED_STT_MODELS,
  sttModel
} from "./whisper.js";
const openAI = (options) => genkitPlugin("openai", (ai) => __async(void 0, null, function* () {
  var _a;
  const client = new OpenAI(options);
  for (const name of Object.keys(SUPPORTED_GPT_MODELS)) {
    gptModel(ai, name, client);
  }
  (_a = options == null ? void 0 : options.models) == null ? void 0 : _a.map((model) => {
    if (!model.name || !model.info || !model.configSchema) {
      throw new Error(`Model ${model.name} is missing required fields`);
    }
    gptModel(ai, model.name, client, model.info, model.configSchema);
  });
  dallE3Model(ai, client);
  for (const name of Object.keys(SUPPORTED_STT_MODELS)) {
    sttModel(ai, name, client);
  }
  for (const name of Object.keys(SUPPORTED_TTS_MODELS)) {
    ttsModel(ai, name, client);
  }
  for (const name of Object.keys(SUPPORTED_EMBEDDING_MODELS)) {
    openaiEmbedder(ai, name, options);
  }
}));
var src_default = openAI;
export {
  dallE3,
  src_default as default,
  gpt35Turbo,
  gpt4,
  gpt41,
  gpt41Mini,
  gpt41Nano,
  gpt45,
  gpt4Turbo,
  gpt4Vision,
  gpt4o,
  gpt4oMini,
  gpt4oMiniTts,
  gpt4oTranscribe,
  o1,
  o1Mini,
  o1Preview,
  o3,
  o3Mini,
  o4Mini,
  openAI,
  textEmbedding3Large,
  textEmbedding3Small,
  textEmbeddingAda002,
  tts1,
  tts1Hd,
  whisper1
};
//# sourceMappingURL=index.mjs.map