Use the same names from modelRefs as keys in SUPPORTED_GEMINI_MODELS …

…for consistency
firebase · May 7, 2024 · d3be1fe · d3be1fe
1 parent 5d5e651
commit d3be1fe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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,
 };