feat: rewrite VertexEmbedder to use @google/genai library with multiple auth options

- Replace OpenAI-compatible approach with native @google/genai SDK
- Add support for multiple authentication methods:
  - API key (uses regular Gemini API endpoint)
  - JSON credentials (service account)
  - Key file path
  - Application default credentials
- Add projectId and location fields for Vertex AI configuration
- Update UI to show all authentication options for Vertex
- Update tests to reflect new implementation
- Update all related type definitions and interfaces

Author: roomote

packages/types/src/codebase-index.ts

@@ -36,6 +36,9 @@ export const codebaseIndexConfigSchema = z.object({
 	// OpenAI Compatible specific fields
 	codebaseIndexOpenAiCompatibleBaseUrl: z.string().optional(),
 	codebaseIndexOpenAiCompatibleModelDimension: z.number().optional(),
+	// Vertex AI specific fields
+	codebaseIndexVertexProjectId: z.string().optional(),
+	codebaseIndexVertexLocation: z.string().optional(),
 })
 
 export type CodebaseIndexConfig = z.infer<typeof codebaseIndexConfigSchema>
@@ -68,6 +71,8 @@ export const codebaseIndexProviderSchema = z.object({
 	codebaseIndexGeminiApiKey: z.string().optional(),
 	codebaseIndexMistralApiKey: z.string().optional(),
 	codebaseIndexVertexApiKey: z.string().optional(),
+	codebaseIndexVertexJsonCredentials: z.string().optional(),
+	codebaseIndexVertexKeyFile: z.string().optional(),
 })
 
 export type CodebaseIndexProvider = z.infer<typeof codebaseIndexProviderSchema>

packages/types/src/global-settings.ts

@@ -185,6 +185,8 @@ export const SECRET_STATE_KEYS = [
 	"codebaseIndexGeminiApiKey",
 	"codebaseIndexMistralApiKey",
 	"codebaseIndexVertexApiKey",
+	"codebaseIndexVertexJsonCredentials",
+	"codebaseIndexVertexKeyFile",
 	"huggingFaceApiKey",
 ] as const satisfies readonly (keyof ProviderSettings)[]
 export type SecretState = Pick<ProviderSettings, (typeof SECRET_STATE_KEYS)[number]>

src/core/webview/ClineProvider.ts

@@ -1554,6 +1554,8 @@ export class ClineProvider
 				codebaseIndexOpenAiCompatibleBaseUrl: codebaseIndexConfig?.codebaseIndexOpenAiCompatibleBaseUrl,
 				codebaseIndexSearchMaxResults: codebaseIndexConfig?.codebaseIndexSearchMaxResults,
 				codebaseIndexSearchMinScore: codebaseIndexConfig?.codebaseIndexSearchMinScore,
+				codebaseIndexVertexProjectId: codebaseIndexConfig?.codebaseIndexVertexProjectId,
+				codebaseIndexVertexLocation: codebaseIndexConfig?.codebaseIndexVertexLocation,
 			},
 			mdmCompliant: this.checkMdmCompliance(),
 			profileThresholds: profileThresholds ?? {},
@@ -1726,6 +1728,8 @@ export class ClineProvider
 					stateValues.codebaseIndexConfig?.codebaseIndexOpenAiCompatibleBaseUrl,
 				codebaseIndexSearchMaxResults: stateValues.codebaseIndexConfig?.codebaseIndexSearchMaxResults,
 				codebaseIndexSearchMinScore: stateValues.codebaseIndexConfig?.codebaseIndexSearchMinScore,
+				codebaseIndexVertexProjectId: stateValues.codebaseIndexConfig?.codebaseIndexVertexProjectId,
+				codebaseIndexVertexLocation: stateValues.codebaseIndexConfig?.codebaseIndexVertexLocation,
 			},
 			profileThresholds: stateValues.profileThresholds ?? {},
 			// Add diagnostic message settings

src/core/webview/webviewMessageHandler.ts

@@ -1998,6 +1998,8 @@ export const webviewMessageHandler = async (
 					codebaseIndexOpenAiCompatibleBaseUrl: settings.codebaseIndexOpenAiCompatibleBaseUrl,
 					codebaseIndexSearchMaxResults: settings.codebaseIndexSearchMaxResults,
 					codebaseIndexSearchMinScore: settings.codebaseIndexSearchMinScore,
+					codebaseIndexVertexProjectId: settings.codebaseIndexVertexProjectId,
+					codebaseIndexVertexLocation: settings.codebaseIndexVertexLocation,
 				}
 
 				// Save global state first
@@ -2028,6 +2030,24 @@ export const webviewMessageHandler = async (
 						settings.codebaseIndexMistralApiKey,
 					)
 				}
+				if (settings.codebaseIndexVertexApiKey !== undefined) {
+					await provider.contextProxy.storeSecret(
+						"codebaseIndexVertexApiKey",
+						settings.codebaseIndexVertexApiKey,
+					)
+				}
+				if (settings.codebaseIndexVertexJsonCredentials !== undefined) {
+					await provider.contextProxy.storeSecret(
+						"codebaseIndexVertexJsonCredentials",
+						settings.codebaseIndexVertexJsonCredentials,
+					)
+				}
+				if (settings.codebaseIndexVertexKeyFile !== undefined) {
+					await provider.contextProxy.storeSecret(
+						"codebaseIndexVertexKeyFile",
+						settings.codebaseIndexVertexKeyFile,
+					)
+				}
 
 				// Send success response first - settings are saved regardless of validation
 				await provider.postMessageToWebview({
@@ -2149,6 +2169,11 @@ export const webviewMessageHandler = async (
 			))
 			const hasGeminiApiKey = !!(await provider.context.secrets.get("codebaseIndexGeminiApiKey"))
 			const hasMistralApiKey = !!(await provider.context.secrets.get("codebaseIndexMistralApiKey"))
+			const hasVertexApiKey = !!(await provider.context.secrets.get("codebaseIndexVertexApiKey"))
+			const hasVertexJsonCredentials = !!(await provider.context.secrets.get(
+				"codebaseIndexVertexJsonCredentials",
+			))
+			const hasVertexKeyFile = !!(await provider.context.secrets.get("codebaseIndexVertexKeyFile"))
 
 			provider.postMessageToWebview({
 				type: "codeIndexSecretStatus",
@@ -2158,6 +2183,9 @@ export const webviewMessageHandler = async (
 					hasOpenAiCompatibleApiKey,
 					hasGeminiApiKey,
 					hasMistralApiKey,
+					hasVertexApiKey,
+					hasVertexJsonCredentials,
+					hasVertexKeyFile,
 				},
 			})
 			break