From b8fa278ab9257debf5315c0c80d1f293c9a8f2aa Mon Sep 17 00:00:00 2001
From: oskarmast <oskarmast9436@gmail.com>
Date: Thu, 28 Mar 2024 19:44:51 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix:=20fix=20gemini=201.5=20pro?=
 =?UTF-8?q?=20model=20id=20to=20support=20gemini=20new=20models=20(#1776)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/config/modelProviders/google.ts | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/src/config/modelProviders/google.ts b/src/config/modelProviders/google.ts
index 55f3d29e..b370066b 100644
--- 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',
 };