Resolve search repository exactly

Co-authored-by: alexr00 <38270282+alexr00@users.noreply.github.com>

Author: Copilot

src/lm/tools/searchTools.ts

@@ -86,7 +86,7 @@ export class SearchTool extends RepoToolBase<SearchToolParameters> {
 	}
 
 	async invoke(options: vscode.LanguageModelToolInvocationOptions<SearchToolParameters>, _token: vscode.CancellationToken): Promise<vscode.LanguageModelToolResult | undefined> {
-		const { owner, name, folderManager } = await this.getRepoInfo({ owner: options.input.repo?.owner, name: options.input.repo?.name });
+		const { owner, name, folderManager } = await this.getRepoInfo({ owner: options.input.repo?.owner, name: options.input.repo?.name, exact: true });
 
 		const parameterQuery = this.toRepositoryQuery(options.input.query, owner, name);
 		Logger.debug(`Searching with query \`${parameterQuery}\``, SearchTool.ID);

src/lm/tools/toolsUtils.ts

@@ -53,16 +53,19 @@ export abstract class RepoToolBase<T> extends ToolBase<T> {
 		super();
 	}
 
-	protected async getRepoInfo(options: { owner?: string, name?: string }): Promise<{ owner: string; name: string; folderManager: FolderRepositoryManager }> {
+	protected async getRepoInfo(options: { owner?: string, name?: string, exact?: boolean }): Promise<{ owner: string; name: string; folderManager: FolderRepositoryManager }> {
 		if (!this.credentialStore.isAnyAuthenticated()) {
 			throw new AuthenticationError();
 		}
+		if (options.exact && (!options.owner || !options.name)) {
+			throw new Error('Repository owner and name are required.');
+		}
 
 		let owner: string | undefined;
 		let name: string | undefined;
 		let folderManager: FolderRepositoryManager | undefined;
 		// The llm likes to make up an owner and name if it isn't provided one, and they tend to include 'owner' and 'name' respectively
-		if (options.owner && options.name && !options.owner.includes('owner') && !options.name.includes('name')) {
+		if (options.owner && options.name && (options.exact || (!options.owner.includes('owner') && !options.name.includes('name')))) {
 			owner = options.owner;
 			name = options.name;
 			folderManager = this.repositoriesManager.getManagerForRepository(options.owner, options.name);

src/test/lm/tools/searchTools.test.ts

@@ -52,7 +52,7 @@ describe('SearchTool', function () {
 			title: 'Known matching issue',
 			titleHTML: 'Known matching issue',
 			body: '',
-			url: 'https://github.com/microsoft/vscode/issues/332663',
+			url: 'https://github.com/repository-owner/repository-name/issues/332663',
 			number: 332663,
 			labels: [],
 			state: 'OPEN',
@@ -79,16 +79,17 @@ describe('SearchTool', function () {
 			repositoriesManager as unknown as RepositoriesManager,
 		).invoke({
 			input: {
-				repo: { owner: 'microsoft', name: 'vscode' },
+				repo: { owner: 'repository-owner', name: 'repository-name' },
 				query: 'is:issue is:open assignee:jruales',
 			},
 			toolInvocationToken: undefined,
 		}, tokenSource.token);
 		tokenSource.dispose();
 
+		assert.strictEqual(repositoriesManager.getManagerForRepository.calledOnceWith('repository-owner', 'repository-name'), true);
 		assert.strictEqual(githubRepository.getIssues.calledOnceWith(
 			undefined,
-			'(is:issue is:open assignee:jruales) AND repo:microsoft/vscode',
+			'(is:issue is:open assignee:jruales) AND repo:repository-owner/repository-name',
 		), true);
 		assert.strictEqual(folderManager.getIssues.notCalled, true);
 		const searchResult = JSON.parse((result!.content[0] as vscode.LanguageModelTextPart).value) as SearchToolResult;