Use tags to find code search tool (#6643)

Part of #6632
microsoft · Feb 21, 2025 · b120817 · b120817
1 parent bc00326
commit b120817
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/lm/tools/suggestFixTool.ts b/src/lm/tools/suggestFixTool.ts
@@ -58,7 +58,12 @@ export class SuggestFixTool extends RepoToolBase<IssueToolParameters> {
 			messages.push(vscode.LanguageModelChatMessage.User(`Comment ${index}: ${comment.body}`));
 		});
 
-		const copilotCodebaseResult = await vscode.lm.invokeTool('copilot_searchCodebase', {
+		const codeSearchTool = vscode.lm.tools.find(value => value.tags.includes('vscode_codesearch'));
+		if (!codeSearchTool) {
+			throw new Error('Could not find the code search tool');
+		}
+
+		const copilotCodebaseResult = await vscode.lm.invokeTool(codeSearchTool.name, {
 			toolInvocationToken: undefined,
 			input: {
 				query: result.title