feat: enable grounding features for Vertex AI (#6777)

* feat: enable grounding features for Vertex AI

- Add enableUrlContext and enableGrounding options to vertexSchema in provider-settings.ts
- Update Vertex.tsx UI component to include URL context and grounding checkboxes
- Add comprehensive tests for the new grounding features in Vertex.spec.tsx
- Reuse existing localization strings from Gemini provider

Fixes #6776

* fix: address review feedback - hide grounding features in welcome view and only show for Gemini models

---------

Co-authored-by: Roo Code <[email protected]>
Co-authored-by: daniel-lxs <[email protected]>

Author: 3 people

packages/types/src/provider-settings.ts

@@ -163,6 +163,8 @@ const vertexSchema = apiModelIdProviderModelSchema.extend({
 	vertexJsonCredentials: z.string().optional(),
 	vertexProjectId: z.string().optional(),
 	vertexRegion: z.string().optional(),
+	enableUrlContext: z.boolean().optional(),
+	enableGrounding: z.boolean().optional(),
 })
 
 const openAiSchema = baseProviderSettingsSchema.extend({

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

@@ -479,7 +479,11 @@ const ApiOptions = ({
 			)}
 
 			{selectedProvider === "vertex" && (
-				<Vertex apiConfiguration={apiConfiguration} setApiConfigurationField={setApiConfigurationField} />
+				<Vertex
+					apiConfiguration={apiConfiguration}
+					setApiConfigurationField={setApiConfigurationField}
+					fromWelcomeView={fromWelcomeView}
+				/>
 			)}
 
 			{selectedProvider === "gemini" && (

webview-ui/src/components/settings/providers/Vertex.tsx

@@ -1,4 +1,5 @@
 import { useCallback } from "react"
+import { Checkbox } from "vscrui"
 import { VSCodeLink, VSCodeTextField } from "@vscode/webview-ui-toolkit/react"
 
 import { type ProviderSettings, VERTEX_REGIONS } from "@roo-code/types"
@@ -11,9 +12,10 @@ import { inputEventTransform } from "../transforms"
 type VertexProps = {
 	apiConfiguration: ProviderSettings
 	setApiConfigurationField: (field: keyof ProviderSettings, value: ProviderSettings[keyof ProviderSettings]) => void
+	fromWelcomeView?: boolean
 }
 
-export const Vertex = ({ apiConfiguration, setApiConfigurationField }: VertexProps) => {
+export const Vertex = ({ apiConfiguration, setApiConfigurationField, fromWelcomeView }: VertexProps) => {
 	const { t } = useAppTranslation()
 
 	const handleInputChange = useCallback(
@@ -91,6 +93,30 @@ export const Vertex = ({ apiConfiguration, setApiConfigurationField }: VertexPro
 					</SelectContent>
 				</Select>
 			</div>
+
+			{!fromWelcomeView && apiConfiguration.apiModelId?.startsWith("gemini") && (
+				<div className="mt-6">
+					<Checkbox
+						data-testid="checkbox-url-context"
+						checked={!!apiConfiguration.enableUrlContext}
+						onChange={(checked: boolean) => setApiConfigurationField("enableUrlContext", checked)}>
+						{t("settings:providers.geminiParameters.urlContext.title")}
+					</Checkbox>
+					<div className="text-sm text-vscode-descriptionForeground mb-3 mt-1.5">
+						{t("settings:providers.geminiParameters.urlContext.description")}
+					</div>
+
+					<Checkbox
+						data-testid="checkbox-grounding-search"
+						checked={!!apiConfiguration.enableGrounding}
+						onChange={(checked: boolean) => setApiConfigurationField("enableGrounding", checked)}>
+						{t("settings:providers.geminiParameters.groundingSearch.title")}
+					</Checkbox>
+					<div className="text-sm text-vscode-descriptionForeground mb-3 mt-1.5">
+						{t("settings:providers.geminiParameters.groundingSearch.description")}
+					</div>
+				</div>
+			)}
 		</>
 	)
 }