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

Genkit AI framework plugin for OpenAI APIs.

import { genkitPlugin } from "genkit/plugin";
import { OpenAI } from "openai";
import {
  SpeechConfigSchema,
  TranscriptionConfigSchema,
  compatOaiSpeechModelRef,
  compatOaiTranscriptionModelRef,
  defineCompatOpenAISpeechModel,
  defineCompatOpenAITranscriptionModel
} from "./audio.js";
import { defineCompatOpenAIEmbedder } from "./embedder.js";
import {
  ImageGenerationCommonConfigSchema,
  compatOaiImageModelRef,
  defineCompatOpenAIImageModel
} from "./image.js";
import {
  ChatCompletionCommonConfigSchema,
  compatOaiModelRef,
  defineCompatOpenAIModel,
  openAIModelRunner
} from "./model.js";
const openAICompatible = (options) => {
  let listActionsCache;
  return genkitPlugin(
    options.name,
    async (ai) => {
      if (options.initializer) {
        const client = new OpenAI(options);
        await options.initializer(ai, client);
      }
    },
    async (ai, actionType, actionName) => {
      if (options.resolver) {
        const client = new OpenAI(options);
        await options.resolver(ai, client, actionType, actionName);
      }
    },
    options.listActions ? async () => {
      if (listActionsCache) return listActionsCache;
      const client = new OpenAI(options);
      listActionsCache = await options.listActions(client);
      return listActionsCache;
    } : void 0
  );
};
var index_default = openAICompatible;
export {
  ChatCompletionCommonConfigSchema,
  ImageGenerationCommonConfigSchema,
  SpeechConfigSchema,
  TranscriptionConfigSchema,
  compatOaiImageModelRef,
  compatOaiModelRef,
  compatOaiSpeechModelRef,
  compatOaiTranscriptionModelRef,
  index_default as default,
  defineCompatOpenAIEmbedder,
  defineCompatOpenAIImageModel,
  defineCompatOpenAIModel,
  defineCompatOpenAISpeechModel,
  defineCompatOpenAITranscriptionModel,
  openAICompatible,
  openAIModelRunner
};
//# sourceMappingURL=index.mjs.map