From 49dc155cdf315a78b5d8f25242b882d7f3c04dc9 Mon Sep 17 00:00:00 2001
From: Kevin Lin <klin5061@gmail.com>
Date: Tue, 10 Mar 2026 10:14:54 -0700
Subject: [PATCH] fix: use Gemini 3.1 Pro preview handles (#1326)

Co-authored-by: Letta Code <noreply@letta.com>
---
 src/index.ts    |  2 +-
 src/models.json | 19 ++++---------------
 2 files changed, 5 insertions(+), 16 deletions(-)

diff --git a/src/index.ts b/src/index.ts
index e7376a3..8be5633 100755
--- a/src/index.ts
+++ b/src/index.ts
@@ -250,7 +250,7 @@ function getModelForToolLoading(
     return "openai/gpt-4";
   }
   if (specifiedToolset === "gemini") {
-    return "google/gemini-3-pro";
+    return "google_ai/gemini-3.1-pro-preview";
   }
   if (specifiedToolset === "default") {
     return "anthropic/claude-sonnet-4";
diff --git a/src/models.json b/src/models.json
index 06ac913..1f2cba7 100644
--- a/src/models.json
+++ b/src/models.json
@@ -1282,17 +1282,6 @@
       "parallel_tool_calls": true
     }
   },
-  {
-    "id": "gemini-3",
-    "handle": "google_ai/gemini-3-pro-preview",
-    "label": "Gemini 3 Pro",
-    "description": "Google's smartest model",
-    "updateArgs": {
-      "context_window": 180000,
-      "temperature": 1.0,
-      "parallel_tool_calls": true
-    }
-  },
   {
     "id": "gemini-3-flash",
     "handle": "google_ai/gemini-3-flash-preview",
@@ -1365,10 +1354,10 @@
     }
   },
   {
-    "id": "gemini-3-vertex",
-    "handle": "google_vertex/gemini-3-pro-preview",
-    "label": "Gemini 3 Pro",
-    "description": "Google's smartest Gemini 3 Pro model (via Vertex AI)",
+    "id": "gemini-3.1-vertex",
+    "handle": "google_vertex/gemini-3.1-pro-preview",
+    "label": "Gemini 3.1 Pro",
+    "description": "Google's latest Gemini 3.1 Pro model (via Vertex AI)",
     "updateArgs": {
       "context_window": 180000,
       "temperature": 1.0,