💄 style: Add Gemini 1.5 stable version model & 2M context window (lob…

…ehub#2744) * Update google.ts * Update google.ts * Update google.ts * Update google.ts * Update google.ts * 去掉 Latest * Update google.ts
bentwnghk · Jul 1, 2024 · a35143b · a35143b
1 parent 4bd05fc
commit a35143b
Showing 1 changed file with 44 additions and 61 deletions.
diff --git a/src/config/modelProviders/google.ts b/src/config/modelProviders/google.ts
@@ -5,99 +5,82 @@ import { ModelProviderCard } from '@/types/llm';
 const Google: ModelProviderCard = {
   chatModels: [
     {
-      description: 'A legacy text-only model optimized for chat conversations',
-      displayName: 'PaLM 2 Chat (Legacy)',
-      id: 'chat-bison-001',
-      legacy: true,
-      maxOutput: 1024,
-      // tokens: 4096 + 1024, // none tokens test
+      description: 'Fast and versatile multimodal model for scaling across diverse tasks',
+      displayName: 'Gemini 1.5 Flash',
+      enabled: true,
+      functionCall: true,
+      id: 'gemini-1.5-flash-latest',
+      maxOutput: 8192,
+      tokens: 1_048_576 + 8192,
+      vision: true,
     },
     {
-      description: 'A legacy model that understands text and generates text as an output',
-      displayName: 'PaLM 2 (Legacy)',
-      id: 'text-bison-001',
-      legacy: true,
-      maxOutput: 1024,
-      tokens: 8196 + 1024,
+      description: 'Fast and versatile multimodal model for scaling across diverse tasks',
+      displayName: 'Gemini 1.5 Flash 001',
+      functionCall: true,
+      id: 'gemini-1.5-flash-001',
+      maxOutput: 8192,
+      tokens: 1_048_576 + 8192,
+      vision: true,
     },
     {
-      description: 'The best model for scaling across a wide range of tasks',
-      displayName: 'Gemini 1.0 Pro',
+      description: 'Mid-size multimodal model that supports up to 1 million tokens',
+      displayName: 'Gemini 1.5 Pro',
+      enabled: true,
       functionCall: true,
-      id: 'gemini-pro',
-      maxOutput: 2048,
-      tokens: 30_720 + 2048,
+      id: 'gemini-1.5-pro-latest',
+      maxOutput: 8192,
+      tokens: 2_097_152 + 8192,
+      vision: true,
     },
     {
-      description: 'The best image understanding model to handle a broad range of applications',
-      displayName: 'Gemini 1.0 Pro Vision',
-      id: 'gemini-1.0-pro-vision-latest',
-      maxOutput: 4096,
-      tokens: 12_288 + 4096,
+      description: 'Mid-size multimodal model that supports up to 1 million tokens',
+      displayName: 'Gemini 1.5 Pro 001',
+      functionCall: true,
+      id: 'gemini-1.5-pro-001',
+      maxOutput: 8192,
+      tokens: 2_097_152 + 8192,
       vision: true,
     },
     {
-      description: 'The best image understanding model to handle a broad range of applications',
-      displayName: 'Gemini 1.0 Pro Vision',
-      id: 'gemini-pro-vision',
-      maxOutput: 4096,
-      tokens: 12_288 + 4096,
-      vision: true,
+      description: 'The best model for scaling across a wide range of tasks. This is the latest model.',
+      displayName: 'Gemini 1.0 Pro',
+      id: 'gemini-1.0-pro-latest',
+      maxOutput: 2048,
+      tokens: 30_720 + 2048,
     },
     {
-      description:
-        'The best model for scaling across a wide range of tasks. This is a stable model that supports tuning.',
+      description: 'The best model for scaling across a wide range of tasks. This is a stable model that supports tuning.',
       displayName: 'Gemini 1.0 Pro 001 (Tuning)',
       functionCall: true,
       id: 'gemini-1.0-pro-001',
       maxOutput: 2048,
       tokens: 30_720 + 2048,
     },
     {
-      description:
-        'The best model for scaling across a wide range of tasks. Released April 9, 2024.',
+      description: 'The best model for scaling across a wide range of tasks. Released April 9, 2024.',
       displayName: 'Gemini 1.0 Pro 002 (Tuning)',
       id: 'gemini-1.0-pro-002',
       maxOutput: 2048,
       tokens: 30_720 + 2048,
     },
-    {
-      description:
-        'The best model for scaling across a wide range of tasks. This is the latest model.',
-      displayName: 'Gemini 1.0 Pro Latest',
-      id: 'gemini-1.0-pro-latest',
-      maxOutput: 2048,
-      tokens: 30_720 + 2048,
-    },
-    {
-      description: 'Fast and versatile multimodal model for scaling across diverse tasks',
-      displayName: 'Gemini 1.5 Flash',
-      enabled: true,
-      functionCall: true,
-      id: 'gemini-1.5-flash-latest',
-      maxOutput: 8192,
-      tokens: 1_048_576 + 8192,
-      vision: true,
-    },
-    {
-      description: 'Mid-size multimodal model that supports up to 1 million tokens',
-      displayName: 'Gemini 1.5 Pro',
-      enabled: true,
-      functionCall: true,
-      id: 'gemini-1.5-pro-latest',
-      maxOutput: 8192,
-      tokens: 1_048_576 + 8192,
-      vision: true,
-    },
     {
       description: 'The most capable model for highly complex tasks',
       displayName: 'Gemini 1.0 Ultra',
       id: 'gemini-ultra',
       maxOutput: 2048,
       tokens: 32_768,
     },
+    {
+      description: 'A legacy text-only model optimized for chat conversations',
+      displayName: 'PaLM 2 Chat (Legacy)',
+      id: 'chat-bison-001',
+      legacy: true,
+      maxOutput: 1024,
+      // tokens: 4096 + 1024, // none tokens test
+    }
   ],
-  checkModel: 'gemini-pro',
+  checkModel: 'gemini-1.5-flash',
   id: 'google',
   name: 'Google',
   proxyUrl: {