From 18cf33f96d7aa518e50b12fd66380d38a0f3cc23 Mon Sep 17 00:00:00 2001
From: ssweens <1149151+ssweens@users.noreply.github.com>
Date: Sat, 6 Sep 2025 19:18:55 -0700
Subject: [PATCH] Feature/update vertex ai models and regions (#7727)

---
 packages/types/src/providers/vertex.ts        | 55 +++++++++++++++++++
 .../providers/__tests__/Vertex.spec.tsx       |  1 +
 2 files changed, 56 insertions(+)

diff --git a/packages/types/src/providers/vertex.ts b/packages/types/src/providers/vertex.ts
index ee8a56ae2c..8010fccf8e 100644
--- a/packages/types/src/providers/vertex.ts
+++ b/packages/types/src/providers/vertex.ts
@@ -294,6 +294,60 @@ export const vertexModels = {
 		outputPrice: 1.15,
 		description: "Meta Llama 4 Maverick 17B Instruct model, 128K context.",
 	},
+	"deepseek-r1-0528-maas": {
+		maxTokens: 32_768,
+		contextWindow: 163_840,
+		supportsImages: false,
+		supportsPromptCache: false,
+		inputPrice: 1.35,
+		outputPrice: 5.4,
+		description: "DeepSeek R1 (0528). Available in us-central1",
+	},
+	"deepseek-v3.1-maas": {
+		maxTokens: 32_768,
+		contextWindow: 163_840,
+		supportsImages: false,
+		supportsPromptCache: false,
+		inputPrice: 0.6,
+		outputPrice: 1.7,
+		description: "DeepSeek V3.1. Available in us-west2",
+	},
+	"gpt-oss-120b-maas": {
+		maxTokens: 32_768,
+		contextWindow: 131_072,
+		supportsImages: false,
+		supportsPromptCache: false,
+		inputPrice: 0.15,
+		outputPrice: 0.6,
+		description: "OpenAI gpt-oss 120B. Available in us-central1",
+	},
+	"gpt-oss-20b-maas": {
+		maxTokens: 32_768,
+		contextWindow: 131_072,
+		supportsImages: false,
+		supportsPromptCache: false,
+		inputPrice: 0.075,
+		outputPrice: 0.3,
+		description: "OpenAI gpt-oss 20B. Available in us-central1",
+	},
+	"qwen3-coder-480b-a35b-instruct-maas": {
+		maxTokens: 32_768,
+		contextWindow: 262_144,
+		supportsImages: false,
+		supportsPromptCache: false,
+		inputPrice: 1.0,
+		outputPrice: 4.0,
+		description: "Qwen3 Coder 480B A35B Instruct. Available in us-south1",
+	},
+	"qwen3-235b-a22b-instruct-2507-maas": {
+		maxTokens: 16_384,
+		contextWindow: 262_144,
+		supportsImages: false,
+		supportsPromptCache: false,
+		inputPrice: 0.25,
+		outputPrice: 1.0,
+		description: "Qwen3 235B A22B Instruct. Available in us-south1",
+	},
 } as const satisfies Record<string, ModelInfo>
 
 export const VERTEX_REGIONS = [
@@ -302,6 +356,7 @@ export const VERTEX_REGIONS = [
 	{ value: "us-east1", label: "us-east1" },
 	{ value: "us-east4", label: "us-east4" },
 	{ value: "us-east5", label: "us-east5" },
+	{ value: "us-south1", label: "us-south1" },
 	{ value: "us-west1", label: "us-west1" },
 	{ value: "us-west2", label: "us-west2" },
 	{ value: "us-west3", label: "us-west3" },
diff --git a/webview-ui/src/components/settings/providers/__tests__/Vertex.spec.tsx b/webview-ui/src/components/settings/providers/__tests__/Vertex.spec.tsx
index 4abe1e8847..c9765429cd 100644
--- a/webview-ui/src/components/settings/providers/__tests__/Vertex.spec.tsx
+++ b/webview-ui/src/components/settings/providers/__tests__/Vertex.spec.tsx
@@ -74,6 +74,7 @@ describe("Vertex", () => {
 				{ value: "us-east1", label: "us-east1" },
 				{ value: "us-east4", label: "us-east4" },
 				{ value: "us-east5", label: "us-east5" },
+				{ value: "us-south1", label: "us-south1" },
 				{ value: "us-west1", label: "us-west1" },
 				{ value: "us-west2", label: "us-west2" },
 				{ value: "us-west3", label: "us-west3" },