Merge pull request #426 from miurla/feat/update-gemini-2.0

feat(models): update Gemini models
miurla · Feb 5, 2025 · 54958b2 · 54958b2
2 parents 3b9ab6e + fcafd07
commit 54958b2
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -175,8 +175,9 @@ This will allow you to use Morphic as your default search engine in the browser.
   - gpt-4-turbo
   - gpt-3.5-turbo
 - Google
-  - Gemini 1.5 Pro (Unstable)
-  - Gemini 2.0 Flash (Experimental)
+  - Gemini 2.0 Pro (Experimental)
+  - Gemini 2.0 Flash Thinking (Experimental)
+  - Gemini 2.0 Flash
 - Anthropic
   - Claude 3.5 Sonnet
   - Claude 3.5 Hike

diff --git a/lib/types/models.ts b/lib/types/models.ts
@@ -43,14 +43,20 @@ export const models: Model[] = [
     providerId: 'deepseek'
   },
   {
-    id: 'gemini-1.5-pro-002',
-    name: 'Gemini 1.5 Pro',
+    id: 'gemini-2.0-pro-exp-02-05',
+    name: 'Gemini 2.0 Pro (Exp)',
     provider: 'Google Generative AI',
     providerId: 'google'
   },
   {
-    id: 'gemini-2.0-flash-exp',
-    name: 'Gemini 2.0 Flash (Experimental)',
+    id: 'gemini-2.0-flash-thinking-exp-01-21',
+    name: 'Gemini 2.0 Flash Thinking (Exp)',
+    provider: 'Google Generative AI',
+    providerId: 'google'
+  },
+  {
+    id: 'gemini-2.0-flash',
+    name: 'Gemini 2.0 Flash',
     provider: 'Google Generative AI',
     providerId: 'google'
   },

diff --git a/lib/utils/registry.ts b/lib/utils/registry.ts
@@ -129,6 +129,8 @@ export function getToolCallModel(model?: string) {
       const ollamaModel =
         process.env.NEXT_PUBLIC_OLLAMA_TOOL_CALL_MODEL || modelName
       return getModel(`ollama:${ollamaModel}`)
+    case 'google':
+      return getModel('google:gemini-2.0-flash')
     default:
       return getModel('openai:gpt-4o-mini')
   }
@@ -141,6 +143,11 @@ export function isToolCallSupported(model?: string) {
   if (provider === 'ollama') {
     return false
   }
+
+  if (provider === 'google') {
+    return false
+  }
+
   // Deepseek R1 is not supported
   // Deepseek v3's tool call is unstable, so we include it in the list
   return !modelName?.includes('deepseek')