From ed5eb70ddab5d462fa4b9aa18a99ea943fd79a89 Mon Sep 17 00:00:00 2001
From: can1357 <me@can.ac>
Date: Fri, 12 Jun 2026 11:50:21 +0200
Subject: [PATCH] fix(catalog): redirected Gemini 3.1 Pro high-effort traffic
 to gemini-pro-agent

- Updated the Gemini 3.1 Pro variant collapse so high effort now routes to `gemini-pro-agent` instead of `gemini-3.1-pro-high`.
- Kept `gemini-3.1-pro-high` in the collapse members list so the raw discovery ID remains handled.
- Lowered `contextWindow` from `163840` to `131072` tokens in `models.json`.
---
 packages/catalog/src/models.json         |  2 +-
 packages/catalog/src/variant-collapse.ts | 11 ++++++++---
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/packages/catalog/src/models.json b/packages/catalog/src/models.json
index 316718a24..b77bf47c2 100644
--- a/packages/catalog/src/models.json
+++ b/packages/catalog/src/models.json
@@ -54417,7 +54417,7 @@
 				"cacheRead": 0.135,
 				"cacheWrite": 0
 			},
-			"contextWindow": 163840,
+			"contextWindow": 131072,
 			"maxTokens": 16384,
 			"thinking": {
 				"mode": "effort",
diff --git a/packages/catalog/src/variant-collapse.ts b/packages/catalog/src/variant-collapse.ts
index f53f146d4..d9b6f7f29 100644
--- a/packages/catalog/src/variant-collapse.ts
+++ b/packages/catalog/src/variant-collapse.ts
@@ -123,12 +123,17 @@ export const ANTIGRAVITY_VARIANT_COLLAPSE_TABLE: VariantCollapseTable = {
 		{
 			id: "gemini-3.1-pro",
 			name: "Gemini 3.1 Pro",
-			// `gemini-pro-agent` is consumed as a member but unused in routing.
-			members: ["gemini-3.1-pro-low", "gemini-3.1-pro-high", "gemini-pro-agent"],
+			// High routes to `gemini-pro-agent` — the upstream `gemini-3.1-pro-high`
+			// deployment returns INVALID_ARGUMENT on every streamGenerateContent
+			// request (both CCA endpoints) while discovery still lists it;
+			// `gemini-pro-agent` is the same model ("Gemini 3.1 Pro (High)", same
+			// thinking budget/caps) and accepts the identical request body.
+			// `gemini-3.1-pro-high` stays a member so the dead raw id is consumed.
+			members: ["gemini-3.1-pro-low", "gemini-pro-agent", "gemini-3.1-pro-high"],
 			routing: {
 				off: "gemini-3.1-pro-low",
 				[Effort.Low]: "gemini-3.1-pro-low",
-				[Effort.High]: "gemini-3.1-pro-high",
+				[Effort.High]: "gemini-pro-agent",
 			},
 			thinking: { mode: "google-level", efforts: GEMINI_3_PRO_FAMILY_EFFORTS },
 			suppressWhenOff: true,