UNPKG

@genkit-ai/vertexai

Version:

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

46 lines (43 loc) 1.56 kB
import * as genkit from 'genkit'; import { Genkit } from 'genkit'; import * as zod from 'zod'; import { VertexRerankOptions } from './types.js'; import 'google-auth-library'; import '../../types-B3i-Lt7D.js'; import '@google-cloud/vertexai'; import 'genkit/model'; /** * Creates Vertex AI rerankers. * * This function creates and registers rerankers for the specified models. * * @param {VertexRerankOptions<EmbedderCustomOptions>} options - The parameters for creating the rerankers. * @returns {Promise<void>} * @deprecated please use vertexRerankers instead */ declare function vertexAiRerankers(ai: Genkit, options: VertexRerankOptions): Promise<void>; /** * Creates a reference to a Vertex AI reranker. * * @param {Object} params - The parameters for the reranker reference. * @param {string} [params.displayName] - An optional display name for the reranker. * @returns {Object} - The reranker reference object. * @deprecated please user vertexRerankerRef instead */ declare const vertexAiRerankerRef: (params: { rerankerName: string; displayName?: string; }) => genkit.RerankerReference<zod.ZodOptional<zod.ZodObject<{ k: zod.ZodOptional<zod.ZodNumber>; model: zod.ZodOptional<zod.ZodString>; location: zod.ZodOptional<zod.ZodString>; }, "strip", zod.ZodTypeAny, { location?: string | undefined; model?: string | undefined; k?: number | undefined; }, { location?: string | undefined; model?: string | undefined; k?: number | undefined; }>>>; export { vertexAiRerankerRef, vertexAiRerankers };