Merge pull request #599 from data5650/patch-1

feat: add Gemini 2.0 Flash Exp models
ItzCrazyKns · Feb 7, 2025 · 7ec201d · 7ec201d
2 parents 46541e6 + 3582695
commit 7ec201d
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/lib/providers/gemini.ts b/src/lib/providers/gemini.ts
@@ -36,6 +36,22 @@ export const loadGeminiChatModels = async () => {
           apiKey: geminiApiKey,
         }),
       },
+      'gemini-2.0-flash-exp': {
+        displayName: 'Gemini 2.0 Flash Exp',
+        model: new ChatGoogleGenerativeAI({
+          modelName: 'gemini-2.0-flash-exp',
+          temperature: 0.7,
+          apiKey: geminiApiKey,
+        }),
+      },
+      'gemini-2.0-flash-thinking-exp-01-21': {
+        displayName: 'Gemini 2.0 Flash Thinking Exp 01-21',
+        model: new ChatGoogleGenerativeAI({
+          modelName: 'gemini-2.0-flash-thinking-exp-01-21',
+          temperature: 0.7,
+          apiKey: geminiApiKey,
+        }),
+      },
     };
 
     return chatModels;