From f2025cc776ab4fa7b8c997da79f14a7f8d4902bc Mon Sep 17 00:00:00 2001
From: decolua <decoluadt@example.com>
Date: Fri, 20 Feb 2026 21:05:02 +0700
Subject: [PATCH] feat: add Gemini 3.1 Pro models to provider

---
 open-sse/config/providerModels.js | 3 ++-
 open-sse/services/usage.js        | 2 ++
 src/shared/constants/cliTools.js  | 2 ++
 3 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/open-sse/config/providerModels.js b/open-sse/config/providerModels.js
index 787685b..20ebb4c 100644
--- a/open-sse/config/providerModels.js
+++ b/open-sse/config/providerModels.js
@@ -62,10 +62,11 @@ export const PROVIDER_MODELS = {
   ag: [  // Antigravity - special case: models call different backends
     { id: "claude-opus-4-6-thinking", name: "Claude Opus 4.6 Thinking" },
     { id: "claude-sonnet-4-6", name: "Claude Sonnet 4.6" },
+    { id: "gemini-3-1-pro-high", name: "Gemini 3.1 Pro High" },
+    { id: "gemini-3-1-pro-low", name: "Gemini 3.1 Pro Low" },
     { id: "gemini-3-pro-high", name: "Gemini 3 Pro High" },
     { id: "gemini-3-pro-low", name: "Gemini 3 Pro Low" },
     { id: "gemini-3-flash", name: "Gemini 3 Flash" },
-    { id: "gemini-2.5-flash", name: "Gemini 2.5 Flash" },
   ],
   gh: [  // GitHub Copilot - OpenAI models
     { id: "gpt-3.5-turbo", name: "GPT-3.5 Turbo" },
diff --git a/open-sse/services/usage.js b/open-sse/services/usage.js
index e6a44dd..016ae96 100644
--- a/open-sse/services/usage.js
+++ b/open-sse/services/usage.js
@@ -246,6 +246,8 @@ async function getAntigravityUsage(accessToken, providerSpecificData) {
       const importantModels = [
         'claude-opus-4-6-thinking',
         'claude-sonnet-4-6',
+        'gemini-3-1-pro-high',
+        'gemini-3-1-pro-low',
         'gemini-3-pro-high',
         'gemini-3-pro-low',
         'gemini-3-flash',
diff --git a/src/shared/constants/cliTools.js b/src/shared/constants/cliTools.js
index 5c923b3..c9ba947 100644
--- a/src/shared/constants/cliTools.js
+++ b/src/shared/constants/cliTools.js
@@ -132,6 +132,8 @@ export const CLI_TOOLS = {
     defaultModels: [
       { id: "claude-opus-4-6-thinking", name: "Claude Opus 4.6 Thinking", alias: "claude-opus-4-6-thinking" },
       { id: "claude-sonnet-4-6", name: "Claude Sonnet 4.6", alias: "claude-sonnet-4-6" },
+      { id: "gemini-3-1-pro-high", name: "Gemini 3.1 Pro High", alias: "gemini-3-1-pro-high" },
+      { id: "gemini-3-1-pro-low", name: "Gemini 3.1 Pro Low", alias: "gemini-3-1-pro-low" },
       { id: "gemini-3-pro-high", name: "Gemini 3 Pro High", alias: "gemini-3-pro-high" },
       { id: "gemini-3-flash", name: "Gemini 3 Flash", alias: "gemini-3-flash" },
     ],