fix: persistently storing new Gemini migrations

Author: HahaBill

webview-ui/src/components/settings/SettingsView.tsx

@@ -219,6 +219,38 @@ const SettingsView = forwardRef<SettingsViewRef, SettingsViewProps>(({ onDone, t
 		})
 	}, [])
 
+	const handleLegacyGeminiModelMapping = useCallback(
+		<K extends keyof ProviderSettings>(
+			field: K,
+			previousValue: ProviderSettings[K],
+			value: ProviderSettings[K],
+			provider: string | undefined,
+			updatedApiConfiguration: ProviderSettings,
+		): boolean => {
+			if (field !== "apiModelId" || typeof previousValue !== "string" || typeof value !== "string") {
+				return false
+			}
+
+			let isLegacyMapping = false
+			if (provider === "gemini") {
+				isLegacyMapping = mapLegacyGeminiModel(previousValue) === value
+			} else if (provider === "vertex") {
+				isLegacyMapping = mapLegacyVertexModel(previousValue) === value
+			}
+
+			if (isLegacyMapping && currentApiConfigName) {
+				vscode.postMessage({
+					type: "upsertApiConfiguration",
+					text: currentApiConfigName,
+					apiConfiguration: updatedApiConfiguration,
+				})
+			}
+
+			return isLegacyMapping
+		},
+		[currentApiConfigName],
+	)
+
 	const setApiConfigurationField = useCallback(
 		<K extends keyof ProviderSettings>(field: K, value: ProviderSettings[K]) => {
 			setCachedState((prevState) => {
@@ -232,23 +264,24 @@ const SettingsView = forwardRef<SettingsViewRef, SettingsViewProps>(({ onDone, t
 				// This prevents the dirty state when the component initializes and auto-syncs the model ID
 				const isInitialSync = previousValue === undefined && value !== undefined
 
-				let isLegacyGeminiMapping = false
-				if (field === "apiModelId" && typeof previousValue === "string" && typeof value === "string") {
-					const provider = prevState.apiConfiguration?.apiProvider
-					if (provider === "gemini") {
-						isLegacyGeminiMapping = mapLegacyGeminiModel(previousValue) === value
-					} else if (provider === "vertex") {
-						isLegacyGeminiMapping = mapLegacyVertexModel(previousValue) === value
-					}
-				}
+				const updatedApiConfiguration = { ...prevState.apiConfiguration, [field]: value }
+
+				const isLegacyMapping = handleLegacyGeminiModelMapping(
+					field,
+					previousValue,
+					value,
+					prevState.apiConfiguration?.apiProvider,
+					updatedApiConfiguration,
+				)
 
-				if (!isInitialSync && !isLegacyGeminiMapping) {
+				if (!isInitialSync && !isLegacyMapping) {
 					setChangeDetected(true)
 				}
-				return { ...prevState, apiConfiguration: { ...prevState.apiConfiguration, [field]: value } }
+
+				return { ...prevState, apiConfiguration: updatedApiConfiguration }
 			})
 		},
-		[],
+		[handleLegacyGeminiModelMapping],
 	)
 
 	const setExperimentEnabled: SetExperimentEnabled = useCallback((id: ExperimentId, enabled: boolean) => {