chore(providers/google-generative-ai): update model list (#21)

moeru-ai · Jan 10, 2025 · fa030f8 · fa030f8
1 parent 8df5a86
commit fa030f8
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/packages/providers/src/providers/google-generative-ai.ts b/packages/providers/src/providers/google-generative-ai.ts
@@ -3,8 +3,17 @@ import type { ChatProvider, EmbedProvider, ModelProvider, ProviderOptions, Provi
 import { generateCRO } from '../utils/generate-cro'
 
 export const createGoogleGenerativeAI = (userOptions: ProviderOptions<true>):
-  /** @see {@link https://ai.google.dev/gemini-api/docs/models/gemini} */
-  ChatProvider<'gemini-1.5-flash' | 'gemini-1.5-flash-8b' | 'gemini-1.5-pro' | 'gemini-2.0-flash-exp'>
+  /** @see {@link https://aistudio.google.com/u/1/prompts/new_chat} */
+  ChatProvider<
+    // gemini 1.5
+    'gemini-1.5-flash' | 'gemini-1.5-flash-8b' | 'gemini-1.5-pro' |
+    // preview
+    'gemini-2.0-flash-exp' | 'gemini-2.0-flash-thinking-exp-1219' | 'gemini-exp-1206' |
+    // gemma
+    'gemma-2-2b-it' | 'gemma-2-9b-it' | 'gemma-2-27b-it' |
+    // preview
+    'lernlm-1.5-pro-experimental'
+  >
   & EmbedProvider<'text-embedding-004'>
   & ModelProvider => {
   const options: ProviderResult = {