Stream coding agent stages in discrete steps

Author: joshspicer

src/github/copilotApi.ts

@@ -79,7 +79,6 @@ export class CopilotApi {
 		return `vscode-pull-request-github/${extensionVersion}`;
 	}
 
-
 	async postRemoteAgentJob(
 		owner: string,
 		name: string,

src/github/copilotRemoteAgent.ts

@@ -94,7 +94,6 @@ export class CopilotRemoteAgentManager extends Disposable {
 				historySummaryLength: historySummary?.length.toString() || '0',
 				source,
 			});
-			stream.progress(vscode.l10n.t('Delegating to coding agent'));
 			const result = await this.invokeRemoteAgent(
 				promptSummary || prompt,
 				[
@@ -103,6 +102,7 @@ export class CopilotRemoteAgentManager extends Disposable {
 				].join('\n\n').trim(),
 				token,
 				false,
+				stream,
 			);
 			if (result.state !== 'success') {
 				Logger.error(`Failed to provide new chat session item: ${result.error}`, CopilotRemoteAgentManager.ID);
@@ -194,7 +194,7 @@ export class CopilotRemoteAgentManager extends Disposable {
 				stream.markdown(result);
 				stream.markdown('\n\n');
 
-				stream.progress(vscode.l10n.t('Waiting for coding agent to respond'));
+				stream.progress(vscode.l10n.t('Attaching to session'));
 
 				// Wait for new session and stream its progress
 				const newSession = await this.waitForNewSession(pullRequest, stream, token, true);
@@ -705,7 +705,7 @@ export class CopilotRemoteAgentManager extends Disposable {
 		return vscode.l10n.t('🚀 Coding agent will continue work in [#{0}]({1}).  Track progress [here]({2}).', number, link, webviewUri.toString());
 	}
 
-	async invokeRemoteAgent(prompt: string, problemContext?: string, token?: vscode.CancellationToken, autoPushAndCommit = true): Promise<RemoteAgentResult> {
+	async invokeRemoteAgent(prompt: string, problemContext?: string, token?: vscode.CancellationToken, autoPushAndCommit = true, chatStream?: vscode.ChatResponseStream): Promise<RemoteAgentResult> {
 		const capiClient = await this.copilotApi;
 		if (!capiClient) {
 			return { error: vscode.l10n.t('Failed to initialize Copilot API'), state: 'error' };
@@ -731,6 +731,7 @@ export class CopilotRemoteAgentManager extends Disposable {
 				return { error: vscode.l10n.t('Uncommitted changes detected. Please commit or stash your changes before starting the remote agent. Enable \'{0}\' to push your changes automatically.', CODING_AGENT_AUTO_COMMIT_AND_PUSH), state: 'error' };
 			}
 			try {
+				chatStream?.progress(vscode.l10n.t('Waiting for local changes'));
 				head_ref = await this.gitOperationsManager.commitAndPushChanges(repoInfo);
 			} catch (error) {
 				return { error: error.message, state: 'error' };
@@ -755,6 +756,7 @@ export class CopilotRemoteAgentManager extends Disposable {
 		const { problemStatement, isTruncated } = truncatePrompt(prompt, problemContext);
 
 		if (isTruncated) {
+			chatStream?.progress(vscode.l10n.t('Truncating context'));
 			const truncationResult = await vscode.window.showWarningMessage(
 				vscode.l10n.t('Prompt size exceeded'), { modal: true, detail: vscode.l10n.t('Your prompt will be truncated to fit within coding agent\'s context window. This may affect the quality of the response.') }, CONTINUE_TRUNCATION);
 			const userCancelled = token?.isCancellationRequested || !truncationResult || truncationResult !== CONTINUE_TRUNCATION;
@@ -784,10 +786,12 @@ export class CopilotRemoteAgentManager extends Disposable {
 		};
 
 		try {
+			chatStream?.progress(vscode.l10n.t('Delegating to coding agent'));
 			const response = await capiClient.postRemoteAgentJob(owner, repo, payload, isTruncated);
 
 			// For v1 API, we need to fetch the job details to get the PR info
 			// Since the PR might not be created immediately, we need to poll for it
+			chatStream?.progress(vscode.l10n.t('Creating pull request'));
 			const jobInfo = await this.waitForJobWithPullRequest(capiClient, owner, repo, response.job_id, token);
 			if (!jobInfo || !jobInfo.pull_request) {
 				return { error: vscode.l10n.t('Failed to retrieve pull request information from job'), state: 'error' };
@@ -803,6 +807,7 @@ export class CopilotRemoteAgentManager extends Disposable {
 			const webviewUri = await toOpenPullRequestWebviewUri({ owner, repo, pullRequestNumber: number });
 			const prLlmString = `The remote agent has begun work and has created a pull request. Details about the pull request are being shown to the user. If the user wants to track progress or iterate on the agent's work, they should use the pull request.`;
 
+			chatStream?.progress(vscode.l10n.t('Attaching to session'));
 			await this.waitForQueuedToInProgress(response.session_id, token);
 			return {
 				state: 'success',