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

Genkit AI framework plugin for OpenAI APIs.

import { embedder } from "genkit/plugin";
import { maybeCreateRequestScopedOpenAIClient, toModelName } from "./utils.mjs";
function defineCompatOpenAIEmbedder(params) {
  const { name, client: defaultClient, pluginOptions, embedderRef } = params;
  const modelName = toModelName(name, pluginOptions?.name);
  const actionName = embedderRef?.name ?? `${pluginOptions?.name ?? "compat-oai"}/${modelName}`;
  return embedder(
    {
      name: actionName,
      configSchema: embedderRef?.configSchema,
      ...embedderRef?.info
    },
    async (req) => {
      const { encodingFormat: encoding_format, ...restOfConfig } = req.options;
      const client = maybeCreateRequestScopedOpenAIClient(
        pluginOptions,
        req,
        defaultClient
      );
      const embeddings = await client.embeddings.create({
        model: modelName,
        input: req.input.map((d) => d.text),
        encoding_format,
        ...restOfConfig
      });
      return {
        embeddings: embeddings.data.map((d) => ({ embedding: d.embedding }))
      };
    }
  );
}
export {
  defineCompatOpenAIEmbedder
};
//# sourceMappingURL=embedder.mjs.map