Attach GHPR comments as chat context to fix comments with Agent (#6818)

Fixes #6782

Author: alexr00

package.json

@@ -3728,6 +3728,20 @@
 					]
 				},
 				"when": "config.githubPullRequests.experimental.chat"
+			},
+			{
+				"name": "github-pull-request_activePullRequest",
+				"tags": [
+					"github",
+					"pull request"
+				],
+				"toolReferenceName": "activePullRequest",
+				"displayName": "%languageModelTools.github-pull-request_activePullRequest.displayName%",
+				"modelDescription": "Get information about the active GitHub pull request. This information includes: comments, files changed, pull request title + description, pull request state, and pull request status checks/CI. When asked about the active pull request, do this first!",
+				"icon": "$(pull-request)",
+				"canBeReferencedInPrompt": true,
+				"userDescription": "%languageModelTools.github-pull-request_activePullRequest.description%",
+				"when": "config.githubPullRequests.experimental.chat"
 			}
 		]
 	},

package.nls.json

@@ -359,6 +359,8 @@
 	"languageModelTools.github-pull-request_suggest-fix.displayName": "Suggest a Fix for a GitHub Issue",
 	"languageModelTools.github-pull-request_formSearchQuery.displayName": "Convert natural language to a GitHub search query",
 	"languageModelTools.github-pull-request_doSearch.displayName": "Execute a GitHub search",
-	"languageModelTools.github-pull-request_renderIssues.displayName": "Render issue items in a markdown table"
+	"languageModelTools.github-pull-request_renderIssues.displayName": "Render issue items in a markdown table",
+	"languageModelTools.github-pull-request_activePullRequest.displayName": "Active Pull Request",
+	"languageModelTools.github-pull-request_activePullRequest.description": "Get information about the active GitHub pull request. This information includes: comments, files changed, pull request title + description, pull request state, and pull request status checks/CI."
 
 }

src/github/interface.ts

@@ -18,9 +18,9 @@ export enum ReviewEvent {
 }
 
 export enum GithubItemStateEnum {
-	Open,
-	Merged,
-	Closed,
+	Open = 'OPEN',
+	Merged = 'MERGED',
+	Closed = 'CLOSED',
 }
 
 export enum PullRequestMergeability {

src/github/issueOverview.ts

@@ -599,4 +599,8 @@ export class IssueOverviewPanel<TItem extends IssueModel = IssueModel> extends W
 	public getCurrentTitle(): string {
 		return this._panel.title;
 	}
+
+	public getCurrentItem(): TItem | undefined {
+		return this._item;
+	}
 }

src/lm/tools/activePullRequestTool.ts

@@ -0,0 +1,71 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+'use strict';
+
+import * as vscode from 'vscode';
+import { PullRequestModel } from '../../github/pullRequestModel';
+import { PullRequestOverviewPanel } from '../../github/pullRequestOverview';
+import { RepositoriesManager } from '../../github/repositoriesManager';
+import { FetchIssueResult } from './fetchIssueTool';
+
+export class ActivePullRequestTool implements vscode.LanguageModelTool<FetchIssueResult> {
+	public static readonly toolId = 'github-pull-request_activePullRequest';
+	constructor(private readonly folderManagers: RepositoriesManager) { }
+
+	private _findActivePullRequest(): PullRequestModel | undefined {
+		const folderManager = this.folderManagers.folderManagers.find((manager) => manager.activePullRequest);
+		return folderManager?.activePullRequest ?? PullRequestOverviewPanel.currentPanel?.getCurrentItem();
+	}
+
+	async prepareInvocation(): Promise<vscode.PreparedToolInvocation> {
+		const pullRequest = this._findActivePullRequest();
+		return {
+			invocationMessage: pullRequest ? vscode.l10n.t('Pull request "{0}"', pullRequest.title) : vscode.l10n.t('Active pull request'),
+		};
+	}
+
+	async invoke(_options: vscode.LanguageModelToolInvocationOptions<any>, _token: vscode.CancellationToken): Promise<vscode.LanguageModelToolResult | undefined> {
+		let pullRequest = this._findActivePullRequest();
+
+		if (!pullRequest) {
+			return new vscode.LanguageModelToolResult([new vscode.LanguageModelTextPart('There is no active pull request')]);
+		}
+
+		const status = await pullRequest.getStatusChecks();
+		const pullRequestInfo = {
+			title: pullRequest.title,
+			body: pullRequest.body,
+			author: pullRequest.author,
+			comments: pullRequest.comments.map(comment => {
+				return {
+					author: comment.user?.login,
+					body: comment.body,
+					commentState: comment.isResolved ? 'resolved' : 'unresolved',
+					file: comment.path
+				};
+			}),
+			state: pullRequest.state,
+			statusChecks: status[0]?.statuses.map((status) => {
+				return {
+					context: status.context,
+					description: status.description,
+					state: status.state,
+					name: status.workflowName,
+					targetUrl: status.targetUrl
+				};
+			}),
+			reviewRequirements: {
+				approvalsNeeded: status[1]?.count ?? 0,
+				currentApprovals: status[1]?.approvals.length ?? 0,
+				areChangesRequested: (status[1]?.requestedChanges.length ?? 0) > 0,
+			},
+			isDraft: pullRequest.isDraft,
+		};
+
+		return new vscode.LanguageModelToolResult([new vscode.LanguageModelTextPart(JSON.stringify(pullRequestInfo))]);
+
+	}
+
+}

src/lm/tools/tools.ts

@@ -8,6 +8,7 @@ import * as vscode from 'vscode';
 import { CredentialStore } from '../../github/credentials';
 import { RepositoriesManager } from '../../github/repositoriesManager';
 import { ChatParticipantState } from '../participants';
+import { ActivePullRequestTool } from './activePullRequestTool';
 import { DisplayIssuesTool } from './displayIssuesTool';
 import { FetchIssueTool } from './fetchIssueTool';
 import { FetchNotificationTool } from './fetchNotificationTool';
@@ -21,6 +22,7 @@ export function registerTools(context: vscode.ExtensionContext, credentialStore:
 	registerSummarizationTools(context);
 	registerSuggestFixTool(context, credentialStore, repositoriesManager, chatParticipantState);
 	registerSearchTools(context, credentialStore, repositoriesManager, chatParticipantState);
+	context.subscriptions.push(vscode.lm.registerTool(ActivePullRequestTool.toolId, new ActivePullRequestTool(repositoriesManager)));
 }
 
 function registerFetchingTools(context: vscode.ExtensionContext, credentialStore: CredentialStore, repositoriesManager: RepositoriesManager, chatParticipantState: ChatParticipantState) {