feat: add Vertex AI as embedder provider for codebase indexing

packages/types/src/codebase-index.ts

@@ -21,7 +21,9 @@ export const CODEBASE_INDEX_DEFAULTS = {
 export const codebaseIndexConfigSchema = z.object({
 	codebaseIndexEnabled: z.boolean().optional(),
 	codebaseIndexQdrantUrl: z.string().optional(),
-	codebaseIndexEmbedderProvider: z.enum(["openai", "ollama", "openai-compatible", "gemini", "mistral"]).optional(),
+	codebaseIndexEmbedderProvider: z
+		.enum(["openai", "ollama", "openai-compatible", "gemini", "mistral", "vertex"])
+		.optional(),
 	codebaseIndexEmbedderBaseUrl: z.string().optional(),
 	codebaseIndexEmbedderModelId: z.string().optional(),
 	codebaseIndexEmbedderModelDimension: z.number().optional(),
@@ -34,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>
@@ -48,6 +53,7 @@ export const codebaseIndexModelsSchema = z.object({
 	"openai-compatible": z.record(z.string(), z.object({ dimension: z.number() })).optional(),
 	gemini: z.record(z.string(), z.object({ dimension: z.number() })).optional(),
 	mistral: z.record(z.string(), z.object({ dimension: z.number() })).optional(),
+	vertex: z.record(z.string(), z.object({ dimension: z.number() })).optional(),
 })
 
 export type CodebaseIndexModels = z.infer<typeof codebaseIndexModelsSchema>
@@ -64,6 +70,9 @@ export const codebaseIndexProviderSchema = z.object({
 	codebaseIndexOpenAiCompatibleModelDimension: z.number().optional(),
 	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

@@ -184,6 +184,9 @@ export const SECRET_STATE_KEYS = [
 	"codebaseIndexOpenAiCompatibleApiKey",
 	"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

src/i18n/locales/en/embeddings.json

@@ -47,6 +47,7 @@
 		"openAiCompatibleConfigMissing": "OpenAI Compatible configuration missing for embedder creation",
 		"geminiConfigMissing": "Gemini configuration missing for embedder creation",
 		"mistralConfigMissing": "Mistral configuration missing for embedder creation",
+		"vertexConfigMissing": "Vertex AI configuration missing for embedder creation",
 		"invalidEmbedderType": "Invalid embedder type configured: {{embedderProvider}}",
 		"vectorDimensionNotDeterminedOpenAiCompatible": "Could not determine vector dimension for model '{{modelId}}' with provider '{{provider}}'. Please ensure the 'Embedding Dimension' is correctly set in the OpenAI-Compatible provider settings.",
 		"vectorDimensionNotDetermined": "Could not determine vector dimension for model '{{modelId}}' with provider '{{provider}}'. Check model profiles or configuration.",

src/services/code-index/constants/index.ts

@@ -29,3 +29,6 @@ export const BATCH_PROCESSING_CONCURRENCY = 10
 
 /**Gemini Embedder */
 export const GEMINI_MAX_ITEM_TOKENS = 2048
+
+/**Vertex AI Embedder */
+export const VERTEX_MAX_ITEM_TOKENS = 2048