fix: adding mapping on the frontend-side

Author: HahaBill

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

@@ -69,6 +69,22 @@ function mapLegacyGeminiModel(modelId: string): string {
 		if (modelId === "gemini-exp-1206") {
 			return geminiDefaultModelId
 		}
+
+		if (modelId === "gemini-2.0-pro-exp-02-05") {
+			return "gemini-2.5-pro"
+		}
+
+		if (
+			modelId === "gemini-2.0-flash-thinking-exp-1219" ||
+			modelId === "gemini-2.0-flash-thinking-exp-01-21" ||
+			modelId === "gemini-2.5-flash-preview-04-17"
+		) {
+			return "gemini-2.5-flash-preview-05-20"
+		}
+
+		if (modelId === "gemini-2.0-flash-exp") {
+			return geminiDefaultModelId
+		}
 	}
 
 	return geminiDefaultModelId
@@ -98,6 +114,18 @@ function mapLegacyVertexModel(modelId: string): string {
 		if (modelId.startsWith("gemini-2.5-pro-exp-")) {
 			return "gemini-2.5-pro"
 		}
+
+		if (modelId === "gemini-2.0-pro-exp-02-05") {
+			return "gemini-2.5-pro"
+		}
+
+		if (
+			modelId === "gemini-2.0-flash-thinking-exp-1219" ||
+			modelId === "gemini-2.0-flash-thinking-exp-01-21" ||
+			modelId === "gemini-2.5-flash-preview-04-17"
+		) {
+			return "gemini-2.5-flash-preview-05-20"
+		}
 	}
 
 	return vertexDefaultModelId