From d69623e472781e229084bd6805f32792db3a363b Mon Sep 17 00:00:00 2001
From: dhruvkej9 <dhruvkejri9@gmail.com>
Date: Tue, 27 Jan 2026 23:10:11 +0530
Subject: [PATCH] feat(antigravity): add gemini and claude models to default
 allowlist

When authenticating with Google Antigravity, the default model allowlist
now includes all commonly used models instead of just claude-opus-4-5-thinking:
- claude-opus-4-5-thinking
- claude-sonnet-4-5
- claude-sonnet-4-5-thinking
- gemini-3-flash
- gemini-3-pro
- gemini-2.5-flash
- gemini-2.5-pro

This ensures /models shows all available models after initial setup.
---
 extensions/google-antigravity-auth/index.ts | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/extensions/google-antigravity-auth/index.ts b/extensions/google-antigravity-auth/index.ts
index cd9e00b1a..b605b1f81 100644
--- a/extensions/google-antigravity-auth/index.ts
+++ b/extensions/google-antigravity-auth/index.ts
@@ -15,6 +15,16 @@ const TOKEN_URL = "https://oauth2.googleapis.com/token";
 const DEFAULT_PROJECT_ID = "rising-fact-p41fc";
 const DEFAULT_MODEL = "google-antigravity/claude-opus-4-5-thinking";
 
+const DEFAULT_MODELS = [
+  "google-antigravity/claude-opus-4-5-thinking",
+  "google-antigravity/claude-sonnet-4-5",
+  "google-antigravity/claude-sonnet-4-5-thinking",
+  "google-antigravity/gemini-3-flash",
+  "google-antigravity/gemini-3-pro",
+  "google-antigravity/gemini-2.5-flash",
+  "google-antigravity/gemini-2.5-pro",
+];
+
 const SCOPES = [
   "https://www.googleapis.com/auth/cloud-platform",
   "https://www.googleapis.com/auth/userinfo.email",
@@ -411,9 +421,9 @@ const antigravityPlugin = {
                 configPatch: {
                   agents: {
                     defaults: {
-                      models: {
-                        [DEFAULT_MODEL]: {},
-                      },
+                      models: Object.fromEntries(
+                        DEFAULT_MODELS.map((m) => [m, {}]),
+                      ),
                     },
                   },
                 },