Generates PR title and message using AI:

- prompt is from createCommitMessage
- only diff is used
(vscode-gitlens-private#43, #4177)

Author: sergeibbb

src/commands/createPullRequestOnRemote.ts

@@ -72,7 +72,22 @@ export class CreatePullRequestOnRemoteCommand extends GlCommandBase {
 			},
 			compare: {
 				branch: args.compare,
-				remote: { path: compareRemote.path, url: compareRemote.url },
+				remote: { path: compareRemote.path, url: compareRemote.url, name: compareRemote.name },
+			},
+			describePullRequest: async (
+				completedResource: RemoteResource & { type: RemoteResourceType.CreatePullRequest },
+			) => {
+				const base = completedResource.base;
+				const compare = completedResource.compare;
+				if (!base?.remote || !compare?.remote || !base?.branch || !compare?.branch) {
+					return undefined;
+				}
+				const baseRef = `${base.remote.name}/${base.branch}`;
+				const compareRef = `${compare.remote.name}/${compare.branch}`;
+				const result = await this.container.ai.generatePullRequestMessage(repo, baseRef, compareRef, {
+					source: 'home', // TODO provide the real source
+				});
+				return result?.parsed;
 			},
 		};
 

src/git/models/remoteResource.ts

@@ -35,12 +35,15 @@ export type RemoteResource =
 			type: RemoteResourceType.CreatePullRequest;
 			base: {
 				branch: string | undefined;
-				remote: { path: string; url: string };
+				remote: { path: string; url: string; name: string };
 			};
 			compare: {
 				branch: string;
-				remote: { path: string; url: string };
+				remote: { path: string; url: string; name: string };
 			};
+			describePullRequest?: (
+				completedResource: RemoteResource & { type: RemoteResourceType.CreatePullRequest },
+			) => Promise<{ summary: string; body: string } | undefined>;
 	  }
 	| {
 			type: RemoteResourceType.File;

src/git/remotes/github.ts

@@ -280,11 +280,15 @@ export class GitHubRemote extends RemoteProvider<GitHubRepositoryDescriptor> {
 		return this.encodeUrl(`${this.baseUrl}/compare/${base}${notation}${head}`);
 	}
 
-	protected override getUrlForCreatePullRequest(
+	protected override async getUrlForCreatePullRequest(
 		base: { branch?: string; remote: { path: string; url: string } },
 		head: { branch: string; remote: { path: string; url: string } },
-		options?: { title?: string; description?: string },
-	): string | undefined {
+		options?: {
+			title?: string;
+			description?: string;
+			describePullRequest?: () => Promise<{ summary: string; body: string } | undefined>;
+		},
+	): Promise<string | undefined> {
 		const query = new URLSearchParams({ expand: '1' });
 		if (options?.title) {
 			query.set('title', options.title);
@@ -293,6 +297,16 @@ export class GitHubRemote extends RemoteProvider<GitHubRepositoryDescriptor> {
 			query.set('body', options.description);
 		}
 
+		if ((!options?.title || !options?.description) && options?.describePullRequest) {
+			const result = await options.describePullRequest();
+			if (result?.summary) {
+				query.set('title', result.summary);
+			}
+			if (result?.body) {
+				query.set('body', result.body);
+			}
+		}
+
 		if (base.remote.url === head.remote.url) {
 			return base.branch
 				? `${this.encodeUrl(`${this.baseUrl}/compare/${base.branch}...${head.branch}`)}?${query.toString()}`

src/git/remotes/remoteProvider.ts

@@ -132,7 +132,9 @@ export abstract class RemoteProvider<T extends ResourceDescriptor = ResourceDesc
 				return this.getUrlForComparison(resource.base, resource.compare, resource.notation ?? '...');
 			}
 			case RemoteResourceType.CreatePullRequest: {
-				return this.getUrlForCreatePullRequest(resource.base, resource.compare);
+				return this.getUrlForCreatePullRequest(resource.base, resource.compare, {
+					describePullRequest: resource.describePullRequest?.bind(null, resource),
+				});
 			}
 			case RemoteResourceType.File:
 				return this.getUrlForFile(
@@ -193,7 +195,11 @@ export abstract class RemoteProvider<T extends ResourceDescriptor = ResourceDesc
 	protected abstract getUrlForCreatePullRequest(
 		base: { branch?: string; remote: { path: string; url: string } },
 		head: { branch: string; remote: { path: string; url: string } },
-		options?: { title?: string; description?: string },
+		options?: {
+			title?: string;
+			description?: string;
+			describePullRequest?: () => Promise<{ summary: string; body: string } | undefined>;
+		},
 	): Promise<string | undefined> | string | undefined;
 
 	protected abstract getUrlForFile(fileName: string, branch?: string, sha?: string, range?: Range): string;

src/plus/ai/aiProviderService.ts

@@ -518,6 +518,44 @@ export class AIProviderService implements Disposable {
 		return result != null ? { ...result, parsed: parseSummarizeResult(result.content) } : undefined;
 	}
 
+	async generatePullRequestMessage(
+		repo: Repository,
+		baseRef: string,
+		compareRef: string,
+		source: Source,
+		options?: {
+			cancellation?: CancellationToken;
+			context?: string;
+			generating?: Deferred<AIModel>;
+			progress?: ProgressOptions;
+		},
+	): Promise<AISummarizeResult | undefined> {
+		const diff = await repo.git.diff().getDiff?.(compareRef, baseRef);
+
+		const result = await this.sendRequest(
+			'generate-commitMessage',
+			() => ({
+				diff: diff?.contents ?? '[no code changes]',
+				context: options?.context ?? '',
+				instructions: configuration.get('ai.generateCommitMessage.customInstructions') ?? '',
+			}),
+			m => `Generating commit message with ${m.name}...`,
+			source,
+			m => ({
+				key: 'ai/generate',
+				data: {
+					type: 'commitMessage',
+					'model.id': m.id,
+					'model.provider.id': m.provider.id,
+					'model.provider.name': m.provider.name,
+					'retry.count': 0,
+				},
+			}),
+			options,
+		);
+		return result != null ? { ...result, parsed: parseSummarizeResult(result.content) } : undefined;
+	}
+
 	async generateDraftMessage(
 		changesOrRepo: string | string[] | Repository,
 		sourceContext: Source & { type: AIGenerateDraftEventData['draftType'] },

src/quickpicks/remoteProviderPicker.ts

@@ -71,7 +71,10 @@ export class CopyOrOpenRemoteCommandQuickPickItem extends CommandQuickPickItem {
 
 					resource = {
 						...resource,
-						base: { branch: branch, remote: { path: this.remote.path, url: this.remote.url } },
+						base: {
+							branch: branch,
+							remote: { path: this.remote.path, url: this.remote.url, name: this.remote.name },
+						},
 					};
 
 					if (