Support pagination in workflow runs (#7134)

* Support pagination in workflow runs

* Missing updates

Author: osortega

src/github/copilotApi.ts

@@ -8,10 +8,10 @@ import JSZip from 'jszip';
 import * as vscode from 'vscode';
 import { AuthProvider } from '../common/authentication';
 import Logger from '../common/logger';
-import { Remote } from '../common/remote';
 import { OctokitCommon } from './common';
 import { CredentialStore } from './credentials';
 import { LoggingOctokit } from './loggingOctokit';
+import { PullRequestModel } from './pullRequestModel';
 import { hasEnterpriseUri } from './utils';
 
 const LEARN_MORE_URL = 'https://docs.github.com/en/copilot/how-tos/agents/copilot-coding-agent';
@@ -109,18 +109,35 @@ export class CopilotApi {
 		}
 	}
 
-	public async getWorkflowRunsFromAction(remote: Remote): Promise<OctokitCommon.ListWorkflowRunsForRepo> {
-		const runs = await this.octokit.api.actions.listWorkflowRunsForRepo(
-			{
-				owner: remote.owner,
-				repo: remote.repositoryName,
-				event: 'dynamic'
+	public async getWorkflowRunsFromAction(pullRequest: PullRequestModel): Promise<OctokitCommon.ListWorkflowRunsForRepo> {
+		const createdDate = new Date(pullRequest.createdAt);
+		const created = `>=${createdDate.getFullYear()}-${String(createdDate.getMonth() + 1).padStart(2, '0')}-${String(createdDate.getDate()).padStart(2, '0')}`;
+		const allRuns: any[] = [];
+		let page = 1;
+		let hasMore = true;
+		const per_page = 100;
+		while (hasMore) {
+			const runs = await this.octokit.api.actions.listWorkflowRunsForRepo({
+				owner: pullRequest.remote.owner,
+				repo: pullRequest.remote.repositoryName,
+				event: 'dynamic',
+				created,
+				per_page,
+				page
+			});
+			if (runs.status !== 200) {
+				throw new Error(`Failed to fetch workflow runs: ${runs.status}`);
+			}
+			if (Array.isArray(runs.data.workflow_runs)) {
+				allRuns.push(...runs.data.workflow_runs);
+				hasMore = runs.data.total_count > allRuns.length;
+				page++;
+			} else {
+				hasMore = false;
 			}
-		);
-		if (runs.status !== 200) {
-			throw new Error(`Failed to fetch workflow runs: ${runs.status}`);
 		}
-		return runs.data.workflow_runs;
+		// Return only the workflow_runs array for compatibility
+		return allRuns as unknown as OctokitCommon.ListWorkflowRunsForRepo;
 	}
 
 	public async getLogsFromZipUrl(logsUrl: string): Promise<string[]> {

src/github/copilotRemoteAgent.ts

@@ -9,16 +9,17 @@ import { AuthProvider } from '../common/authentication';
 import { COPILOT_LOGINS } from '../common/copilot';
 import { commands } from '../common/executeCommands';
 import { Disposable } from '../common/lifecycle';
-import { GitHubRemote, Remote } from '../common/remote';
+import Logger from '../common/logger';
+import { GitHubRemote } from '../common/remote';
 import { CODING_AGENT, CODING_AGENT_AUTO_COMMIT_AND_PUSH, CODING_AGENT_ENABLED } from '../common/settingKeys';
 import { toOpenPullRequestWebviewUri } from '../common/uri';
 import { CopilotApi, RemoteAgentJobPayload } from './copilotApi';
 import { CopilotPRWatcher, CopilotStateModel } from './copilotPrWatcher';
 import { CredentialStore } from './credentials';
 import { FolderRepositoryManager } from './folderRepositoryManager';
-import { RepositoriesManager } from './repositoriesManager';
 import { GitHubRepository } from './githubRepository';
-import Logger from '../common/logger';
+import { PullRequestModel } from './pullRequestModel';
+import { RepositoriesManager } from './repositoriesManager';
 
 type RemoteAgentSuccessResult = { link: string; state: 'success'; number: number; webviewUri: vscode.Uri; llmDetails: string };
 type RemoteAgentErrorResult = { error: string; state: 'error' };
@@ -397,15 +398,15 @@ export class CopilotRemoteAgentManager extends Disposable {
 		}
 	}
 
-	async getSessionLogsFromAction(remote: Remote, pullRequestId: number) {
+	async getSessionLogsFromAction(pullRequest: PullRequestModel) {
 		const capi = await this.copilotApi;
 		if (!capi) {
 			return [];
 		}
-		const runs = await capi.getWorkflowRunsFromAction(remote);
+		const runs = await capi.getWorkflowRunsFromAction(pullRequest);
 		const padawanRuns = runs
 			.filter(run => run.path && run.path.startsWith('dynamic/copilot-swe-agent'))
-			.filter(run => run.pull_requests?.some(pr => pr.id === pullRequestId));
+			.filter(run => run.pull_requests?.some(pr => pr.id === pullRequest.id));
 
 		const lastRun = this.getLatestRun(padawanRuns);
 

src/lm/tools/activePullRequestTool.ts

@@ -73,7 +73,7 @@ export class ActivePullRequestTool implements vscode.LanguageModelTool<FetchIssu
 		model: vscode.LanguageModelChat,
 		cancellationToken: vscode.CancellationToken
 	) {
-		const logs = await this.copilotRemoteAgentManager.getSessionLogsFromAction(pullRequest.remote, pullRequest.id);
+		const logs = await this.copilotRemoteAgentManager.getSessionLogsFromAction(pullRequest);
 		// Summarize the Copilot agent's thinking process using the model
 		const messages = [
 			vscode.LanguageModelChatMessage.Assistant('You are an expert summarizer. The following logs show the thinking process and performed actions of a GitHub Copilot agent that was in charge of working on the current pull request. Read the logs and always maintain the thinking process. You can remove information on the tool call results that you think are not necessary for building context.'),