🐛 fix: fix gemini 1.5 pro model id to support gemini new models (#1776)

miroshar-success · Mar 28, 2024 · b8fa278 · b8fa278
1 parent e1800db
commit b8fa278
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/src/config/modelProviders/google.ts b/src/config/modelProviders/google.ts
@@ -2,11 +2,6 @@ import { ModelProviderCard } from '@/types/llm';
 
 const Google: ModelProviderCard = {
   chatModels: [
-    {
-      displayName: 'Gemini 1.5 Pro',
-      id: 'gemini-1.5-pro',
-      tokens: 1_048_576,
-    },
     {
       displayName: 'Gemini Pro',
       id: 'gemini-pro',
@@ -18,6 +13,16 @@ const Google: ModelProviderCard = {
       tokens: 12_288,
       vision: true,
     },
+    {
+      displayName: 'Gemini 1.5 Pro',
+      id: 'gemini-1.5-pro-latest',
+      tokens: 1_048_576,
+    },
+    {
+      displayName: 'Gemini Ultra',
+      id: 'gemini-ultra-latest',
+      tokens: 30_720,
+    },
   ],
   id: 'google',
 };