feat: Tuning the setting specifically for Gemini

Author: HahaBill

packages/types/src/provider-settings.ts

@@ -57,7 +57,6 @@ const baseProviderSettingsSchema = z.object({
 	diffEnabled: z.boolean().optional(),
 	fuzzyMatchThreshold: z.number().optional(),
 	modelTemperature: z.number().nullish(),
-	modelMaxContextWindow: z.number().nullish(),
 	rateLimitSeconds: z.number().optional(),
 
 	// Model reasoning.
@@ -153,6 +152,7 @@ const lmStudioSchema = baseProviderSettingsSchema.extend({
 const geminiSchema = apiModelIdProviderModelSchema.extend({
 	geminiApiKey: z.string().optional(),
 	googleGeminiBaseUrl: z.string().optional(),
+	modelMaxContextWindow: z.number().nullish(),
 })
 
 const openAiNativeSchema = apiModelIdProviderModelSchema.extend({
@@ -319,6 +319,7 @@ export const PROVIDER_SETTINGS_KEYS = keysOf<ProviderSettings>()([
 	// Gemini
 	"geminiApiKey",
 	"googleGeminiBaseUrl",
+	"modelMaxContextWindow",
 	// OpenAI Native
 	"openAiNativeApiKey",
 	"openAiNativeBaseUrl",
@@ -347,7 +348,6 @@ export const PROVIDER_SETTINGS_KEYS = keysOf<ProviderSettings>()([
 	"diffEnabled",
 	"fuzzyMatchThreshold",
 	"modelTemperature",
-	"modelMaxContextWindow",
 	"rateLimitSeconds",
 	// Fake AI
 	"fakeAi",

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

@@ -485,11 +485,15 @@ const ApiOptions = ({
 						onChange={handleInputChange("modelTemperature", noTransform)}
 						maxValue={2}
 					/>
-					<MaxContextWindowControl
-						value={apiConfiguration.modelMaxContextWindow}
-						onChange={handleInputChange("modelMaxContextWindow", noTransform)}
-						maxValue={1000000}
-					/>
+
+					{selectedProvider === "gemini" && (
+						<MaxContextWindowControl
+							value={apiConfiguration.modelMaxContextWindow}
+							onChange={handleInputChange("modelMaxContextWindow", noTransform)}
+							maxValue={1000000}
+						/>
+					)}
+
 					<RateLimitSecondsControl
 						value={apiConfiguration.rateLimitSeconds || 0}
 						onChange={(value) => setApiConfigurationField("rateLimitSeconds", value)}

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

@@ -13,26 +13,26 @@ interface MaxContextWindowControlProps {
 
 export const MaxContextWindowControl = ({ value, onChange, maxValue = 1000000 }: MaxContextWindowControlProps) => {
 	const { t } = useAppTranslation()
-	const [isCustomTemperature, setIsCustomTemperature] = useState(value !== undefined)
+	const [isCustomMaxContextWindow, setIsCustomMaxContextWindow] = useState(value !== undefined)
 	const [inputValue, setInputValue] = useState(value)
 
 	useDebounce(() => onChange(inputValue), 50, [onChange, inputValue])
 
 	// Sync internal state with prop changes when switching profiles.
 	useEffect(() => {
 		const hasCustomTemperature = value !== undefined && value !== null
-		setIsCustomTemperature(hasCustomTemperature)
+		setIsCustomMaxContextWindow(hasCustomTemperature)
 		setInputValue(value)
 	}, [value])
 
 	return (
 		<>
 			<div>
 				<VSCodeCheckbox
-					checked={isCustomTemperature}
+					checked={isCustomMaxContextWindow}
 					onChange={(e: any) => {
 						const isChecked = e.target.checked
-						setIsCustomTemperature(isChecked)
+						setIsCustomMaxContextWindow(isChecked)
 
 						if (!isChecked) {
 							setInputValue(null) // Unset the temperature, note that undefined is unserializable.
@@ -47,7 +47,7 @@ export const MaxContextWindowControl = ({ value, onChange, maxValue = 1000000 }:
 				</div>
 			</div>
 
-			{isCustomTemperature && (
+			{isCustomMaxContextWindow && (
 				<div className="flex flex-col gap-3 pl-3 border-l-2 border-vscode-button-background">
 					<div>
 						<div className="flex items-center gap-2">