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, };