From 27e47de3aa7a5382230b88758fa10f44ff4b5667 Mon Sep 17 00:00:00 2001
From: Daniel <57051444+daniel-lxs@users.noreply.github.com>
Date: Tue, 20 May 2025 20:13:21 -0500
Subject: [PATCH] Add Gemini Flash 2.5 05-20 variants for the Vertex provider
 (#3758)

* feat(api): add gemini-2.5-flash-preview-05-20 model configuration

* feat(tests): update apiModelId to gemini-2.5-flash-preview-05-20 in ProviderSettingsManager tests in case the old version is deprecated

* chore: add changeset

* feat(api): update vertexModels to add gemini-2.5-flash-preview-05-20 variants

* chore: update changeset
---
 .changeset/clean-taxis-feel.md                |  6 ++++++
 .../__tests__/ProviderSettingsManager.test.ts |  2 +-
 src/shared/api.ts                             | 19 +++++++++++++++++++
 3 files changed, 26 insertions(+), 1 deletion(-)
 create mode 100644 .changeset/clean-taxis-feel.md

diff --git a/.changeset/clean-taxis-feel.md b/.changeset/clean-taxis-feel.md
new file mode 100644
index 0000000000..4c8d60f3ef
--- /dev/null
+++ b/.changeset/clean-taxis-feel.md
@@ -0,0 +1,6 @@
+---
+"roo-cline": patch
+---
+
+Add support for `gemini-2.5-flash-preview-05-20` on the Vertex provider
+Add support for `gemini-2.5-flash-preview-05-20:thinking` on the Vertex provider
diff --git a/src/core/config/__tests__/ProviderSettingsManager.test.ts b/src/core/config/__tests__/ProviderSettingsManager.test.ts
index 7e50544095..3eb436a079 100644
--- a/src/core/config/__tests__/ProviderSettingsManager.test.ts
+++ b/src/core/config/__tests__/ProviderSettingsManager.test.ts
@@ -222,7 +222,7 @@ describe("ProviderSettingsManager", () => {
 
 			const newConfig: ProviderSettings = {
 				apiProvider: "vertex",
-				apiModelId: "gemini-2.5-flash-preview-04-17",
+				apiModelId: "gemini-2.5-flash-preview-05-20",
 				vertexKeyFile: "test-key-file",
 			}
 
diff --git a/src/shared/api.ts b/src/shared/api.ts
index 05d6c3f58d..7350d96f6c 100644
--- a/src/shared/api.ts
+++ b/src/shared/api.ts
@@ -472,6 +472,25 @@ export const openRouterDefaultModelInfo: ModelInfo = {
 export type VertexModelId = keyof typeof vertexModels
 export const vertexDefaultModelId: VertexModelId = "claude-3-7-sonnet@20250219"
 export const vertexModels = {
+	"gemini-2.5-flash-preview-05-20:thinking": {
+		maxTokens: 65_535,
+		contextWindow: 1_048_576,
+		supportsImages: true,
+		supportsPromptCache: false,
+		inputPrice: 0.15,
+		outputPrice: 3.5,
+		thinking: true,
+		maxThinkingTokens: 24_576,
+	},
+	"gemini-2.5-flash-preview-05-20": {
+		maxTokens: 65_535,
+		contextWindow: 1_048_576,
+		supportsImages: true,
+		supportsPromptCache: false,
+		inputPrice: 0.15,
+		outputPrice: 0.6,
+		thinking: false,
+	},
 	"gemini-2.5-flash-preview-04-17:thinking": {
 		maxTokens: 65_535,
 		contextWindow: 1_048_576,