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

Genkit AI framework plugin for OpenAI APIs.

function defineCompatOpenAIEmbedder(params) {
  const { ai, name, client, embedderRef } = params;
  const modelName = name.substring(name.indexOf("/") + 1);
  return ai.defineEmbedder(
    {
      name,
      configSchema: embedderRef?.configSchema,
      ...embedderRef?.info
    },
    async (input, options) => {
      const { encodingFormat: encoding_format, ...restOfConfig } = options;
      const embeddings = await client.embeddings.create({
        model: modelName,
        input: input.map((d) => d.text),
        encoding_format,
        ...restOfConfig
      });
      return {
        embeddings: embeddings.data.map((d) => ({ embedding: d.embedding }))
      };
    }
  );
}
export {
  defineCompatOpenAIEmbedder
};
//# sourceMappingURL=embedder.mjs.map