From d3be1fe5eb4c656c8fd2258e6b572030e5b4e030 Mon Sep 17 00:00:00 2001 From: Michael Doyle Date: Tue, 7 May 2024 14:35:15 -0400 Subject: [PATCH] Use the same names from modelRefs as keys in SUPPORTED_GEMINI_MODELS for consistency --- js/plugins/vertexai/src/gemini.ts | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/js/plugins/vertexai/src/gemini.ts b/js/plugins/vertexai/src/gemini.ts index 35fd72109a..a755f7a948 100644 --- a/js/plugins/vertexai/src/gemini.ts +++ b/js/plugins/vertexai/src/gemini.ts @@ -104,8 +104,8 @@ export const gemini15ProPreview = modelRef({ }); export const SUPPORTED_V1_MODELS = { - 'gemini-pro': geminiPro, - 'gemini-pro-vision': geminiProVision, + 'gemini-1.0-pro': geminiPro, + 'gemini-1.0-pro-vision': geminiProVision, // 'gemini-ultra': geminiUltra, };