feat: Add configurable maximum search results setting for codebase indexing

- Add searchMaxResults property to CodeIndexConfig with default value of 50
- Update VectorStore interface to accept configurable maxResults parameter
- Modify QdrantClient to use configurable limit instead of hardcoded MAX_SEARCH_RESULTS
- Add slider control in CodeIndexSettings UI (range: 1-500)
- Add localization keys for the new setting
- Update all tests to include new configuration field

Fixes #5149

Author: cte

packages/types/src/codebase-index.ts

@@ -10,6 +10,7 @@ export const codebaseIndexConfigSchema = z.object({
 	codebaseIndexEmbedderProvider: z.enum(["openai", "ollama", "openai-compatible"]).optional(),
 	codebaseIndexEmbedderBaseUrl: z.string().optional(),
 	codebaseIndexEmbedderModelId: z.string().optional(),
+	codebaseIndexSearchMaxResults: z.number().int().min(1).max(500).optional(),
 })
 
 export type CodebaseIndexConfig = z.infer<typeof codebaseIndexConfigSchema>

src/services/code-index/__tests__/config-manager.spec.ts

@@ -36,9 +36,11 @@ describe("CodeIndexConfigManager", () => {
 				modelId: undefined,
 				openAiOptions: { openAiNativeApiKey: "" },
 				ollamaOptions: { ollamaBaseUrl: "" },
+				openAiCompatibleOptions: undefined,
 				qdrantUrl: "http://localhost:6333",
 				qdrantApiKey: "",
 				searchMinScore: 0.4,
+				searchMaxResults: 50,
 			})
 			expect(result.requiresRestart).toBe(false)
 		})
@@ -67,9 +69,11 @@ describe("CodeIndexConfigManager", () => {
 				modelId: "text-embedding-3-large",
 				openAiOptions: { openAiNativeApiKey: "test-openai-key" },
 				ollamaOptions: { ollamaBaseUrl: "" },
+				openAiCompatibleOptions: undefined,
 				qdrantUrl: "http://qdrant.local",
 				qdrantApiKey: "test-qdrant-key",
 				searchMinScore: 0.4,
+				searchMaxResults: 50,
 			})
 		})
 
@@ -104,10 +108,12 @@ describe("CodeIndexConfigManager", () => {
 				openAiCompatibleOptions: {
 					baseUrl: "https://api.example.com/v1",
 					apiKey: "test-openai-compatible-key",
+					modelDimension: undefined,
 				},
 				qdrantUrl: "http://qdrant.local",
 				qdrantApiKey: "test-qdrant-key",
 				searchMinScore: 0.4,
+				searchMaxResults: 50,
 			})
 		})
 
@@ -148,6 +154,7 @@ describe("CodeIndexConfigManager", () => {
 				qdrantUrl: "http://qdrant.local",
 				qdrantApiKey: "test-qdrant-key",
 				searchMinScore: 0.4,
+				searchMaxResults: 50,
 			})
 		})
 
@@ -183,10 +190,12 @@ describe("CodeIndexConfigManager", () => {
 				openAiCompatibleOptions: {
 					baseUrl: "https://api.example.com/v1",
 					apiKey: "test-openai-compatible-key",
+					modelDimension: undefined,
 				},
 				qdrantUrl: "http://qdrant.local",
 				qdrantApiKey: "test-qdrant-key",
 				searchMinScore: 0.4,
+				searchMaxResults: 50,
 			})
 		})
 
@@ -227,6 +236,7 @@ describe("CodeIndexConfigManager", () => {
 				qdrantUrl: "http://qdrant.local",
 				qdrantApiKey: "test-qdrant-key",
 				searchMinScore: 0.4,
+				searchMaxResults: 50,
 			})
 		})
 
@@ -939,9 +949,11 @@ describe("CodeIndexConfigManager", () => {
 				modelId: "text-embedding-3-large",
 				openAiOptions: { openAiNativeApiKey: "test-openai-key" },
 				ollamaOptions: { ollamaBaseUrl: undefined },
+				openAiCompatibleOptions: undefined,
 				qdrantUrl: "http://qdrant.local",
 				qdrantApiKey: "test-qdrant-key",
 				searchMinScore: 0.4,
+				searchMaxResults: 50,
 			})
 		})
 

src/services/code-index/config-manager.ts

@@ -19,6 +19,7 @@ export class CodeIndexConfigManager {
 	private qdrantUrl?: string = "http://localhost:6333"
 	private qdrantApiKey?: string
 	private searchMinScore?: number
+	private searchMaxResults?: number
 
 	constructor(private readonly contextProxy: ContextProxy) {
 		// Initialize with current configuration to avoid false restart triggers
@@ -35,6 +36,7 @@ export class CodeIndexConfigManager {
 			codebaseIndexEnabled: false,
 			codebaseIndexQdrantUrl: "http://localhost:6333",
 			codebaseIndexSearchMinScore: 0.4,
+			codebaseIndexSearchMaxResults: 50,
 			codebaseIndexEmbedderProvider: "openai",
 			codebaseIndexEmbedderBaseUrl: "",
 			codebaseIndexEmbedderModelId: "",
@@ -46,6 +48,7 @@ export class CodeIndexConfigManager {
 			codebaseIndexEmbedderProvider,
 			codebaseIndexEmbedderBaseUrl,
 			codebaseIndexEmbedderModelId,
+			codebaseIndexSearchMaxResults,
 		} = codebaseIndexConfig
 
 		const openAiKey = this.contextProxy?.getSecret("codeIndexOpenAiKey") ?? ""
@@ -62,6 +65,7 @@ export class CodeIndexConfigManager {
 		this.qdrantApiKey = qdrantApiKey ?? ""
 		this.openAiOptions = { openAiNativeApiKey: openAiKey }
 		this.searchMinScore = SEARCH_MIN_SCORE
+		this.searchMaxResults = codebaseIndexSearchMaxResults ?? 50
 
 		// Set embedder provider with support for openai-compatible
 		if (codebaseIndexEmbedderProvider === "ollama") {
@@ -104,6 +108,7 @@ export class CodeIndexConfigManager {
 			qdrantUrl?: string
 			qdrantApiKey?: string
 			searchMinScore?: number
+			searchMaxResults?: number
 		}
 		requiresRestart: boolean
 	}> {
@@ -140,6 +145,7 @@ export class CodeIndexConfigManager {
 				qdrantUrl: this.qdrantUrl,
 				qdrantApiKey: this.qdrantApiKey,
 				searchMinScore: this.searchMinScore,
+				searchMaxResults: this.searchMaxResults,
 			},
 			requiresRestart,
 		}
@@ -295,6 +301,7 @@ export class CodeIndexConfigManager {
 			qdrantUrl: this.qdrantUrl,
 			qdrantApiKey: this.qdrantApiKey,
 			searchMinScore: this.searchMinScore,
+			searchMaxResults: this.searchMaxResults,
 		}
 	}
 
@@ -342,4 +349,11 @@ export class CodeIndexConfigManager {
 	public get currentSearchMinScore(): number | undefined {
 		return this.searchMinScore
 	}
+
+	/**
+	 * Gets the configured maximum search results.
+	 */
+	public get currentSearchMaxResults(): number | undefined {
+		return this.searchMaxResults
+	}
 }

src/services/code-index/interfaces/config.ts

@@ -15,6 +15,7 @@ export interface CodeIndexConfig {
 	qdrantUrl?: string
 	qdrantApiKey?: string
 	searchMinScore?: number
+	searchMaxResults?: number
 }
 
 /**

src/services/code-index/interfaces/vector-store.ts

@@ -23,10 +23,17 @@ export interface IVectorStore {
 	/**
 	 * Searches for similar vectors
 	 * @param queryVector Vector to search for
-	 * @param limit Maximum number of results to return
+	 * @param directoryPrefix Optional directory path to filter results by
+	 * @param minScore Optional minimum score threshold
+	 * @param maxResults Maximum number of results to return
 	 * @returns Promise resolving to search results
 	 */
-	search(queryVector: number[], directoryPrefix?: string, minScore?: number): Promise<VectorStoreSearchResult[]>
+	search(
+		queryVector: number[],
+		directoryPrefix?: string,
+		minScore?: number,
+		maxResults?: number,
+	): Promise<VectorStoreSearchResult[]>
 
 	/**
 	 * Deletes points by file path

src/services/code-index/search-service.ts

@@ -30,6 +30,7 @@ export class CodeIndexSearchService {
 		}
 
 		const minScore = this.configManager.currentSearchMinScore
+		const maxResults = this.configManager.currentSearchMaxResults ?? 50
 
 		const currentState = this.stateManager.getCurrentStatus().systemStatus
 		if (currentState !== "Indexed" && currentState !== "Indexing") {
@@ -52,7 +53,7 @@ export class CodeIndexSearchService {
 			}
 
 			// Perform search
-			const results = await this.vectorStore.search(vector, normalizedPrefix, minScore)
+			const results = await this.vectorStore.search(vector, normalizedPrefix, minScore, maxResults)
 			return results
 		} catch (error) {
 			console.error("[CodeIndexSearchService] Error during search:", error)

src/services/code-index/vector-store/qdrant-client.ts

@@ -278,6 +278,7 @@ export class QdrantVectorStore implements IVectorStore {
 		queryVector: number[],
 		directoryPrefix?: string,
 		minScore?: number,
+		maxResults?: number,
 	): Promise<VectorStoreSearchResult[]> {
 		try {
 			let filter = undefined
@@ -297,7 +298,7 @@ export class QdrantVectorStore implements IVectorStore {
 				query: queryVector,
 				filter,
 				score_threshold: SEARCH_MIN_SCORE,
-				limit: MAX_SEARCH_RESULTS,
+				limit: maxResults ?? MAX_SEARCH_RESULTS,
 				params: {
 					hnsw_ef: 128,
 					exact: false,

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

@@ -27,6 +27,7 @@ import {
 	AlertDialogHeader,
 	AlertDialogTitle,
 	AlertDialogTrigger,
+	Slider,
 } from "@src/components/ui"
 
 import { SetCachedStateField } from "./types"
@@ -449,6 +450,34 @@ export const CodeIndexSettings: React.FC<CodeIndexSettingsProps> = ({
 						</div>
 					</div>
 
+					<div className="flex flex-col gap-3">
+						<div className="flex items-center gap-4 font-bold">
+							<div>{t("settings:codeIndex.searchMaxResultsLabel")}</div>
+						</div>
+						<div className="flex items-center gap-4">
+							<Slider
+								value={[codebaseIndexConfig.codebaseIndexSearchMaxResults || 50]}
+								onValueChange={([value]) =>
+									setCachedStateField("codebaseIndexConfig", {
+										...codebaseIndexConfig,
+										codebaseIndexSearchMaxResults: value,
+									})
+								}
+								min={1}
+								max={500}
+								step={1}
+								className="flex-1"
+								data-testid="search-max-results-slider"
+							/>
+							<span className="text-sm text-vscode-descriptionForeground min-w-[3rem] text-right">
+								{codebaseIndexConfig.codebaseIndexSearchMaxResults || 50}
+							</span>
+						</div>
+						<p className="text-vscode-descriptionForeground text-sm mt-0">
+							{t("settings:codeIndex.searchMaxResultsDescription")}
+						</p>
+					</div>
+
 					{(!areSettingsCommitted || !validateIndexingConfig(codebaseIndexConfig, apiConfiguration)) && (
 						<p className="text-sm text-vscode-descriptionForeground mb-2">
 							{t("settings:codeIndex.unsavedSettingsMessage")}

webview-ui/src/i18n/locales/en/settings.json

@@ -56,6 +56,8 @@
 		"ollamaUrlLabel": "Ollama URL:",
 		"qdrantUrlLabel": "Qdrant URL",
 		"qdrantKeyLabel": "Qdrant Key:",
+		"searchMaxResultsLabel": "Maximum Search Results",
+		"searchMaxResultsDescription": "Maximum number of search results returned by codebase search. Higher values provide more comprehensive results but may increase processing time and token usage.",
 		"startIndexingButton": "Start Indexing",
 		"clearIndexDataButton": "Clear Index Data",
 		"unsavedSettingsMessage": "Please save your settings before starting the indexing process.",