UNPKG

@genkit-ai/vertexai

Version:

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

36 lines (33 loc) 1.51 kB
import * as _genkit_ai_ai_retriever from '@genkit-ai/ai/retriever'; import { z, Genkit } from 'genkit'; import { IndexerAction } from 'genkit/retriever'; import { f as VertexVectorSearchOptions } from '../../types-Ddt5ljfS.js'; import '../../types-DgDIietx.js'; import 'google-auth-library'; import '@google-cloud/vertexai'; import 'genkit/model'; import '@google-cloud/aiplatform'; import 'genkit/embedder'; /** * Creates a reference to a Vertex AI indexer. * * @param {Object} params - The parameters for the indexer reference. * @param {string} params.indexId - The ID of the Vertex AI index. * @param {string} [params.displayName] - An optional display name for the indexer. * @returns {Object} - The indexer reference object. */ declare const vertexAiIndexerRef: (params: { indexId: string; displayName?: string; }) => _genkit_ai_ai_retriever.IndexerReference<z.ZodOptional<z.ZodAny>>; /** * Creates Vertex AI indexers. * * This function returns a list of indexer actions for Vertex AI based on the provided * vector search options and embedder configurations. * * @param {VertexVectorSearchOptions<EmbedderCustomOptions>} params - The parameters for creating the indexers. * @returns {IndexerAction<z.ZodTypeAny>[]} - An array of indexer actions. */ declare function vertexAiIndexers<EmbedderCustomOptions extends z.ZodTypeAny>(ai: Genkit, params: VertexVectorSearchOptions<EmbedderCustomOptions>): IndexerAction<z.ZodTypeAny>[]; export { vertexAiIndexerRef, vertexAiIndexers };