refactor(vertex): centralize Vertex pricing and 1M context in types helper and consume in webview

Implements feedback to move config logic into packages/types for reuse across providers.

Author: roomote

packages/types/src/providers/vertex.ts

@@ -430,3 +430,85 @@ export const VERTEX_REGIONS = [
 	{ value: "me-central1", label: "me-central1" },
 	{ value: "africa-south1", label: "africa-south1" },
 ]
+
+// Regional pricing constants for Vertex Claude Sonnet 4.5
+export const VERTEX_SONNET_45_REGIONAL_PRICING_REGIONS = ["us-east5", "europe-west1", "asia-southeast1"] as const
+
+export type VertexRegionalPricingRegion = (typeof VERTEX_SONNET_45_REGIONAL_PRICING_REGIONS)[number] | "global"
+
+/**
+ * getVertexAdjustedModelInfo
+ *
+ * Centralizes Vertex Claude Sonnet pricing and 1M context adjustments.
+ * - Applies region-aware pricing for claude-sonnet-4-5@20250929
+ * - Applies global pricing for claude-sonnet-4@20250514
+ * - Enables 1M context window when either:
+ *   • Model id ends with "[1m]" OR
+ *   • largeInputTierEnabled is true
+ */
+export function getVertexAdjustedModelInfo(
+	id: string,
+	base: ModelInfo | undefined,
+	opts?: { region?: string; largeInputTierEnabled?: boolean },
+): ModelInfo | undefined {
+	if (!base) return undefined
+
+	const isSonnet45 = id.startsWith("claude-sonnet-4-5@20250929")
+	const isSonnet4 = id.startsWith("claude-sonnet-4@20250514")
+
+	// If not a Sonnet 4/4.5 model, return base info as-is.
+	if (!isSonnet45 && !isSonnet4) return base
+
+	const region = opts?.region ?? "global"
+	const is1m = id.endsWith("[1m]") || opts?.largeInputTierEnabled === true
+
+	const regionalSet = new Set<string>(VERTEX_SONNET_45_REGIONAL_PRICING_REGIONS)
+	const useRegionalPricing = regionalSet.has(region)
+
+	if (isSonnet45) {
+		if (is1m) {
+			// Over 200k (1M tier) with regional pricing
+			return {
+				...base,
+				contextWindow: 1_000_000,
+				inputPrice: useRegionalPricing ? 6.6 : 6.0,
+				outputPrice: useRegionalPricing ? 24.75 : 22.5,
+				cacheWritesPrice: useRegionalPricing ? 8.25 : 7.5,
+				cacheReadsPrice: useRegionalPricing ? 0.66 : 0.6,
+			}
+		}
+
+		// Under 200k with regional pricing
+		return {
+			...base,
+			contextWindow: 200_000,
+			inputPrice: useRegionalPricing ? 3.3 : 3.0,
+			outputPrice: useRegionalPricing ? 16.5 : 15.0,
+			cacheWritesPrice: useRegionalPricing ? 4.13 : 3.75,
+			cacheReadsPrice: useRegionalPricing ? 0.33 : 0.3,
+		}
+	}
+
+	// Sonnet 4 (global pricing only)
+	if (is1m) {
+		// Over 200k (1M tier)
+		return {
+			...base,
+			contextWindow: 1_000_000,
+			inputPrice: 6.0,
+			outputPrice: 22.5,
+			cacheWritesPrice: 7.5,
+			cacheReadsPrice: 0.6,
+		}
+	}
+
+	// Under 200k
+	return {
+		...base,
+		contextWindow: 200_000,
+		inputPrice: 3.0,
+		outputPrice: 15.0,
+		cacheWritesPrice: 3.75,
+		cacheReadsPrice: 0.3,
+	}
+}

webview-ui/src/components/ui/hooks/useSelectedModel.ts

@@ -57,6 +57,7 @@ import {
 	vercelAiGatewayDefaultModelId,
 	BEDROCK_1M_CONTEXT_MODEL_IDS,
 	deepInfraDefaultModelId,
+	getVertexAdjustedModelInfo,
 } from "@roo-code/types"
 
 import type { ModelRecord, RouterModels } from "@roo/api"
@@ -225,86 +226,13 @@ function getSelectedModel({
 				return { id, info: undefined }
 			}
 
-			// Region-aware, tiered pricing for Vertex Claude Sonnet models
-			// Source: https://cloud.google.com/vertex-ai/generative-ai/pricing
-			//
-			// Rules implemented:
-			// - Sonnet 4 pricing is the same globally in all regions
-			//   • Under 200k Input Tokens:
-			//     Input: $3, Output: $15, Cache Write: $3.75, Cache Hit: $0.30
-			//   • Over 200k Input Tokens ([1m] variants or largeInputTierEnabled):
-			//     Input: $6, Output: $22.50, Cache Write: $7.50, Cache Hit: $0.60
-			//
-			// - Sonnet 4.5 has different pricing per region and per input context size
-			//   • Global region (all regions except us-east5, europe-west1, asia-southeast1)
-			//     - Under 200k Input Tokens:
-			//       Input: $3.00, Output: $15.00, Cache Write: $3.75, Cache Hit: $0.30
-			//     - Over 200k Input Tokens ([1m] variants or largeInputTierEnabled):
-			//       Input: $6.00, Output: $22.50, Cache Write: $7.50, Cache Hit: $0.60
-			//   • Regional (us-east5, europe-west1, asia-southeast1)
-			//     - Under 200k Input Tokens:
-			//       Input: $3.30, Output: $16.50, Cache Write: $4.13, Cache Hit: $0.33
-			//     - Over 200k Input Tokens ([1m] variants or largeInputTierEnabled):
-			//       Input: $6.60, Output: $24.75, Cache Write: $8.25, Cache Hit: $0.66
-			//
-			// We derive "over 200k" from Roo's explicit [1m] model variants, the selected Vertex region,
-			// or the generic largeInputTierEnabled setting.
-			const region = apiConfiguration.vertexRegion ?? "global"
-			const is1m = id.endsWith("[1m]") || apiConfiguration.largeInputTierEnabled === true
-			const isSonnet45 = id.startsWith("claude-sonnet-4-5@20250929")
-			const isSonnet4 = id.startsWith("claude-sonnet-4@20250514")
-			const regionalPricingRegions = new Set(["us-east5", "europe-west1", "asia-southeast1"])
-			const useRegionalPricing = regionalPricingRegions.has(region)
+			// Delegate pricing and 1M context adjustments to shared types helper
+			const info = getVertexAdjustedModelInfo(id, baseInfo, {
+				region: apiConfiguration.vertexRegion ?? "global",
+				largeInputTierEnabled: apiConfiguration.largeInputTierEnabled === true,
+			})
 
-			let adjustedInfo: ModelInfo = baseInfo as ModelInfo
-
-			if (isSonnet45) {
-				if (is1m) {
-					// Over 200k (1M beta)
-					adjustedInfo = {
-						...baseInfo,
-						contextWindow: 1_000_000,
-						inputPrice: useRegionalPricing ? (6.6 as number) : (6.0 as number),
-						outputPrice: useRegionalPricing ? (24.75 as number) : (22.5 as number),
-						cacheWritesPrice: useRegionalPricing ? (8.25 as number) : (7.5 as number),
-						cacheReadsPrice: useRegionalPricing ? (0.66 as number) : (0.6 as number),
-					} as ModelInfo
-				} else {
-					// Under 200k
-					adjustedInfo = {
-						...baseInfo,
-						contextWindow: 200_000,
-						inputPrice: useRegionalPricing ? (3.3 as number) : (3.0 as number),
-						outputPrice: useRegionalPricing ? (16.5 as number) : (15.0 as number),
-						cacheWritesPrice: useRegionalPricing ? (4.13 as number) : (3.75 as number),
-						cacheReadsPrice: useRegionalPricing ? (0.33 as number) : (0.3 as number),
-					} as ModelInfo
-				}
-			} else if (isSonnet4) {
-				if (is1m) {
-					// Over 200k (1M beta) - global pricing
-					adjustedInfo = {
-						...baseInfo,
-						contextWindow: 1_000_000,
-						inputPrice: 6.0 as number,
-						outputPrice: 22.5 as number,
-						cacheWritesPrice: 7.5 as number,
-						cacheReadsPrice: 0.6 as number,
-					} as ModelInfo
-				} else {
-					// Under 200k - global pricing
-					adjustedInfo = {
-						...baseInfo,
-						contextWindow: 200_000,
-						inputPrice: 3.0 as number,
-						outputPrice: 15.0 as number,
-						cacheWritesPrice: 3.75 as number,
-						cacheReadsPrice: 0.3 as number,
-					} as ModelInfo
-				}
-			}
-
-			return { id, info: adjustedInfo }
+			return { id, info }
 		}
 		case "gemini": {
 			const id = apiConfiguration.apiModelId ?? geminiDefaultModelId