@genkit-ai/vertexai/lib/rerankers/index.mjs

Genkit AI framework plugin for Google Cloud Vertex AI APIs including Gemini APIs, Imagen, and more.

import { genkitPlugin } from "genkit/plugin";
import { getDerivedParams } from "../common/index.js";
import { vertexAiRerankers } from "./reranker.js";
function vertexAIRerankers(options) {
  return genkitPlugin("vertexAIRerankers", async (ai) => {
    const { projectId, location, authClient } = await getDerivedParams(options);
    await vertexAiRerankers(ai, {
      projectId,
      location,
      authClient,
      rerankOptions: options.rerankers.map(
        (o) => typeof o === "string" ? { model: o } : o
      )
    });
  });
}
export {
  vertexAIRerankers
};
//# sourceMappingURL=index.mjs.map