From 96cf7425ca3acf4aeeab8dc3a26cf2df0cf5e44f Mon Sep 17 00:00:00 2001
From: audichuang <72485427+audichuang@users.noreply.github.com>
Date: Thu, 26 Feb 2026 07:18:03 +0800
Subject: [PATCH] feat(ai): add gemini-3.1-pro-preview to google-gemini-cli
 provider (#1599)

Add Gemini 3.1 Pro Preview model to the Cloud Code Assist (google-gemini-cli)
provider for parity with the google and google-vertex providers that already
include this model.

Tested and confirmed working via the Cloud Code Assist API endpoint.
---
 packages/ai/scripts/generate-models.ts | 12 ++++++++++++
 packages/ai/src/models.generated.ts    | 17 +++++++++++++++++
 2 files changed, 29 insertions(+)

diff --git a/packages/ai/scripts/generate-models.ts b/packages/ai/scripts/generate-models.ts
index 3deda493..b0b201f5 100644
--- a/packages/ai/scripts/generate-models.ts
+++ b/packages/ai/scripts/generate-models.ts
@@ -1012,6 +1012,18 @@ async function generateModels() {
 			contextWindow: 1048576,
 			maxTokens: 65535,
 		},
+		{
+			id: "gemini-3.1-pro-preview",
+			name: "Gemini 3.1 Pro Preview (Cloud Code Assist)",
+			api: "google-gemini-cli",
+			provider: "google-gemini-cli",
+			baseUrl: CLOUD_CODE_ASSIST_ENDPOINT,
+			reasoning: true,
+			input: ["text", "image"],
+			cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+			contextWindow: 1048576,
+			maxTokens: 65535,
+		},
 	];
 	allModels.push(...cloudCodeAssistModels);
 
diff --git a/packages/ai/src/models.generated.ts b/packages/ai/src/models.generated.ts
index afa6e144..791a7f3d 100644
--- a/packages/ai/src/models.generated.ts
+++ b/packages/ai/src/models.generated.ts
@@ -3498,6 +3498,23 @@ export const MODELS = {
 			contextWindow: 1048576,
 			maxTokens: 65535,
 		} satisfies Model<"google-gemini-cli">,
+		"gemini-3.1-pro-preview": {
+			id: "gemini-3.1-pro-preview",
+			name: "Gemini 3.1 Pro Preview (Cloud Code Assist)",
+			api: "google-gemini-cli",
+			provider: "google-gemini-cli",
+			baseUrl: "https://cloudcode-pa.googleapis.com",
+			reasoning: true,
+			input: ["text", "image"],
+			cost: {
+				input: 0,
+				output: 0,
+				cacheRead: 0,
+				cacheWrite: 0,
+			},
+			contextWindow: 1048576,
+			maxTokens: 65535,
+		} satisfies Model<"google-gemini-cli">,
 	},
 	"google-vertex": {
 		"gemini-1.5-flash": {