From f668794bd41c02074a75341c5f92e8a4855473f0 Mon Sep 17 00:00:00 2001 From: Jacob Cable Date: Mon, 11 Nov 2024 16:38:37 +0000 Subject: [PATCH] fix(js/plugins/vertexai): fix subplugin names --- js/plugins/vertexai/src/evaluation/index.ts | 2 +- js/plugins/vertexai/src/modelgarden/index.ts | 2 +- js/plugins/vertexai/src/rerankers/index.ts | 2 +- js/plugins/vertexai/src/vectorsearch/index.ts | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/js/plugins/vertexai/src/evaluation/index.ts b/js/plugins/vertexai/src/evaluation/index.ts index 50675c2b5..799c77732 100644 --- a/js/plugins/vertexai/src/evaluation/index.ts +++ b/js/plugins/vertexai/src/evaluation/index.ts @@ -25,7 +25,7 @@ export { PluginOptions }; * Add Google Cloud Vertex AI Rerankers API to Genkit. */ export function vertexAIEvaluation(options: PluginOptions): GenkitPlugin { - return genkitPlugin('vertexai', async (ai: Genkit) => { + return genkitPlugin('vertexAIEvaluation', async (ai: Genkit) => { const { projectId, location, authClient } = await getDerivedParams(options); const metrics = diff --git a/js/plugins/vertexai/src/modelgarden/index.ts b/js/plugins/vertexai/src/modelgarden/index.ts index 1584de0ff..8acbe81d4 100644 --- a/js/plugins/vertexai/src/modelgarden/index.ts +++ b/js/plugins/vertexai/src/modelgarden/index.ts @@ -27,7 +27,7 @@ import { PluginOptions } from './types.js'; * Add Google Cloud Vertex AI Rerankers API to Genkit. */ export function vertexAIModelGarden(options: PluginOptions): GenkitPlugin { - return genkitPlugin('vertexai', async (ai: Genkit) => { + return genkitPlugin('vertexAIModelGarden', async (ai: Genkit) => { const { projectId, location, authClient } = await getDerivedParams(options); const mgModels = options?.modelGardenModels || options?.modelGarden?.models; diff --git a/js/plugins/vertexai/src/rerankers/index.ts b/js/plugins/vertexai/src/rerankers/index.ts index c5f143ecf..33cd8cf52 100644 --- a/js/plugins/vertexai/src/rerankers/index.ts +++ b/js/plugins/vertexai/src/rerankers/index.ts @@ -29,7 +29,7 @@ export interface PluginOptions extends CommonPluginOptions, RerankerOptions {} * Add Google Cloud Vertex AI Rerankers API to Genkit. */ export function vertexAIRerankers(options: PluginOptions): GenkitPlugin { - return genkitPlugin('vertexai', async (ai: Genkit) => { + return genkitPlugin('vertexAIRerankers', async (ai: Genkit) => { const { projectId, location, authClient } = await getDerivedParams(options); await vertexAiRerankers(ai, { diff --git a/js/plugins/vertexai/src/vectorsearch/index.ts b/js/plugins/vertexai/src/vectorsearch/index.ts index 048e18b72..e372660e5 100644 --- a/js/plugins/vertexai/src/vectorsearch/index.ts +++ b/js/plugins/vertexai/src/vectorsearch/index.ts @@ -38,7 +38,7 @@ export { * Add Google Cloud Vertex AI to Genkit. Includes Gemini and Imagen models and text embedder. */ export function vertexAIVectorSearch(options?: PluginOptions): GenkitPlugin { - return genkitPlugin('vertexai', async (ai: Genkit) => { + return genkitPlugin('vertexAIVectorSearch', async (ai: Genkit) => { const { projectId, location, vertexClientFactory, authClient } = await getDerivedParams(options);