feat: integrate WarpGrep as alternative codebase_search backend

packages/types/src/codebase-index.ts

@@ -50,6 +50,8 @@ export const codebaseIndexConfigSchema = z.object({
 	codebaseIndexBedrockProfile: z.string().optional(),
 	// OpenRouter specific fields
 	codebaseIndexOpenRouterSpecificProvider: z.string().optional(),
+	// WarpGrep specific fields
+	warpGrepEnabled: z.boolean().optional(),
 })
 
 export type CodebaseIndexConfig = z.infer<typeof codebaseIndexConfigSchema>
@@ -85,6 +87,7 @@ export const codebaseIndexProviderSchema = z.object({
 	codebaseIndexMistralApiKey: z.string().optional(),
 	codebaseIndexVercelAiGatewayApiKey: z.string().optional(),
 	codebaseIndexOpenRouterApiKey: z.string().optional(),
+	warpGrepApiKey: z.string().optional(),
 })
 
 export type CodebaseIndexProvider = z.infer<typeof codebaseIndexProviderSchema>

packages/types/src/global-settings.ts

@@ -275,6 +275,7 @@ export const SECRET_STATE_KEYS = [
 	"codebaseIndexMistralApiKey",
 	"codebaseIndexVercelAiGatewayApiKey",
 	"codebaseIndexOpenRouterApiKey",
+	"warpGrepApiKey",
 	"sambaNovaApiKey",
 	"zaiApiKey",
 	"fireworksApiKey",

packages/types/src/vscode-extension-host.ts

@@ -678,6 +678,10 @@ export interface WebviewMessage {
 		codebaseIndexMistralApiKey?: string
 		codebaseIndexVercelAiGatewayApiKey?: string
 		codebaseIndexOpenRouterApiKey?: string
+
+		// WarpGrep specific fields
+		warpGrepEnabled?: boolean
+		warpGrepApiKey?: string
 	}
 	updatedSettings?: RooCodeSettings
 	/** Task configuration applied via `createTask()` when starting a cloud task. */

src/core/prompts/tools/__tests__/filter-tools-for-mode.warpgrep.spec.ts

@@ -0,0 +1,43 @@
+import type OpenAI from "openai"
+
+import { filterNativeToolsForMode, isToolAllowedInMode } from "../filter-tools-for-mode"
+
+function makeTool(name: string): OpenAI.Chat.ChatCompletionTool {
+	return {
+		type: "function",
+		function: {
+			name,
+			description: `${name} tool`,
+			parameters: { type: "object", properties: {} },
+		},
+	} as OpenAI.Chat.ChatCompletionTool
+}
+
+describe("filterNativeToolsForMode - WarpGrep", () => {
+	it("keeps codebase_search available when WarpGrep is enabled", () => {
+		const result = filterNativeToolsForMode(
+			[makeTool("codebase_search"), makeTool("read_file")],
+			"code",
+			undefined,
+			undefined,
+			undefined,
+			{
+				codebaseIndexConfig: {
+					warpGrepEnabled: true,
+				},
+			},
+		)
+
+		expect(result.map((tool) => (tool as any).function.name)).toContain("codebase_search")
+	})
+
+	it("reports codebase_search as allowed when WarpGrep is enabled", () => {
+		expect(
+			isToolAllowedInMode("codebase_search", "code", undefined, undefined, undefined, {
+				codebaseIndexConfig: {
+					warpGrepEnabled: true,
+				},
+			}),
+		).toBe(true)
+	})
+})

src/core/prompts/tools/filter-tools-for-mode.ts

@@ -269,9 +269,16 @@ export function filterNativeToolsForMode(
 	allowedToolNames = customizedTools
 
 	// Conditionally exclude codebase_search if feature is disabled or not configured
+	// WarpGrep can serve as an alternative backend for codebase_search
+	const warpGrepEnabled = settings?.codebaseIndexConfig?.warpGrepEnabled === true
 	if (
-		!codeIndexManager ||
-		!(codeIndexManager.isFeatureEnabled && codeIndexManager.isFeatureConfigured && codeIndexManager.isInitialized)
+		!warpGrepEnabled &&
+		(!codeIndexManager ||
+			!(
+				codeIndexManager.isFeatureEnabled &&
+				codeIndexManager.isFeatureConfigured &&
+				codeIndexManager.isInitialized
+			))
 	) {
 		allowedToolNames.delete("codebase_search")
 	}
@@ -363,11 +370,15 @@ export function isToolAllowedInMode(
 	if (ALWAYS_AVAILABLE_TOOLS.includes(toolName)) {
 		// But still check for conditional exclusions
 		if (toolName === "codebase_search") {
-			return !!(
-				codeIndexManager &&
-				codeIndexManager.isFeatureEnabled &&
-				codeIndexManager.isFeatureConfigured &&
-				codeIndexManager.isInitialized
+			const warpGrepEnabled = settings?.codebaseIndexConfig?.warpGrepEnabled === true
+			return (
+				warpGrepEnabled ||
+				!!(
+					codeIndexManager &&
+					codeIndexManager.isFeatureEnabled &&
+					codeIndexManager.isFeatureConfigured &&
+					codeIndexManager.isInitialized
+				)
 			)
 		}
 		if (toolName === "update_todo_list") {

src/core/task/build-tools.ts

@@ -101,6 +101,7 @@ export async function buildNativeToolsArrayWithRestrictions(options: BuildToolsO
 	// Build settings object for tool filtering.
 	const filterSettings = {
 		todoListEnabled: apiConfiguration?.todoListEnabled ?? true,
+		codebaseIndexConfig: provider.contextProxy.getGlobalState("codebaseIndexConfig"),
 		disabledTools,
 		modelInfo,
 	}

src/core/tools/CodebaseSearchTool.ts

@@ -6,6 +6,7 @@ import { CodeIndexManager } from "../../services/code-index/manager"
 import { getWorkspacePath } from "../../utils/path"
 import { formatResponse } from "../prompts/responses"
 import { VectorStoreSearchResult } from "../../services/code-index/interfaces"
+import { executeWarpGrepSearch } from "../../services/warpgrep"
 import type { ToolUse } from "../../shared/tools"
 
 import { BaseTool, ToolCallbacks } from "./BaseTool"
@@ -52,7 +53,32 @@ export class CodebaseSearchTool extends BaseTool<"codebase_search"> {
 		task.consecutiveMistakeCount = 0
 
 		try {
-			const context = task.providerRef.deref()?.context
+			const provider = task.providerRef.deref()
+			const contextProxy = provider?.contextProxy
+
+			// Try WarpGrep first if enabled
+			if (contextProxy) {
+				const codebaseIndexConfig = contextProxy.getGlobalState("codebaseIndexConfig")
+				const warpGrepApiKey = contextProxy.getSecret("warpGrepApiKey")
+
+				if (codebaseIndexConfig?.warpGrepEnabled && warpGrepApiKey) {
+					const result = await executeWarpGrepSearch(
+						workspacePath,
+						query,
+						warpGrepApiKey,
+						task.rooIgnoreController,
+					)
+
+					if (result.success) {
+						pushToolResult(`Query: ${query}\n\n${result.content}`)
+						return
+					}
+
+					// Fall through to CodeIndexManager if available
+				}
+			}
+
+			const context = provider?.context
 			if (!context) {
 				throw new Error("Extension context is not available.")
 			}

src/core/webview/webviewMessageHandler.ts

@@ -2517,6 +2517,7 @@ export const webviewMessageHandler = async (
 					codebaseIndexSearchMaxResults: settings.codebaseIndexSearchMaxResults,
 					codebaseIndexSearchMinScore: settings.codebaseIndexSearchMinScore,
 					codebaseIndexOpenRouterSpecificProvider: settings.codebaseIndexOpenRouterSpecificProvider,
+					warpGrepEnabled: settings.warpGrepEnabled,
 				}
 
 				// Save global state first
@@ -2559,6 +2560,9 @@ export const webviewMessageHandler = async (
 						settings.codebaseIndexOpenRouterApiKey,
 					)
 				}
+				if (settings.warpGrepApiKey !== undefined) {
+					await provider.contextProxy.storeSecret("warpGrepApiKey", settings.warpGrepApiKey)
+				}
 
 				// Send success response first - settings are saved regardless of validation
 				await provider.postMessageToWebview({
@@ -2697,6 +2701,7 @@ export const webviewMessageHandler = async (
 				"codebaseIndexVercelAiGatewayApiKey",
 			))
 			const hasOpenRouterApiKey = !!(await provider.context.secrets.get("codebaseIndexOpenRouterApiKey"))
+			const hasWarpGrepApiKey = !!(await provider.context.secrets.get("warpGrepApiKey"))
 
 			provider.postMessageToWebview({
 				type: "codeIndexSecretStatus",
@@ -2708,6 +2713,7 @@ export const webviewMessageHandler = async (
 					hasMistralApiKey,
 					hasVercelAiGatewayApiKey,
 					hasOpenRouterApiKey,
+					hasWarpGrepApiKey,
 				},
 			})
 			break