sessions list paging

Author: rebornix

src/github/copilotApi.ts

@@ -192,11 +192,9 @@ export class CopilotApi {
 		return copilotSteps;
 	}
 
-	public async getAllSessions(pullRequestId: number | undefined): Promise<SessionInfo[]> {
+	public async getAllSessions(pullRequestId: number): Promise<SessionInfo[]> {
 		const response = await this.makeApiCall(
-			pullRequestId
-				? `/agents/sessions/resource/pull/${pullRequestId}`
-				: `/agents/sessions?page_size=100`,
+			`/agents/sessions/resource/pull/${pullRequestId}`,
 			{
 				headers: {
 					Authorization: `Bearer ${this.token}`,
@@ -210,6 +208,32 @@ export class CopilotApi {
 		return sessions.sessions;
 	}
 
+	public async getAllSessionsForAllRepositories(): Promise<SessionInfo[]> {
+		let hasNextPage = false;
+		const sessionInfos: SessionInfo[] = [];
+		const page_size = 20;
+		let page = 1;
+		do {
+			const response = await this.makeApiCall(
+				`/agents/sessions?page_size=${page_size}&page_number=${page}`,
+				{
+					headers: {
+						Authorization: `Bearer ${this.token}`,
+						Accept: 'application/json',
+					},
+				});
+			if (!response.ok) {
+				throw new Error(`Failed to fetch sessions: ${response.statusText}`);
+			}
+			const sessions = await response.json();
+			sessionInfos.push(...sessions.sessions);
+			hasNextPage = sessions.sessions.length === page_size;
+			page++;
+		} while (hasNextPage);
+
+		return sessionInfos;
+	}
+
 	public async getPullRequestFromSession(globalId): Promise<SessionPullRequestInfo | undefined> {
 		try {
 			const { data } = await this.graphql.query({

src/github/copilotRemoteAgent.ts

@@ -882,7 +882,7 @@ export class CopilotRemoteAgentManager extends Disposable {
 			const currentRepositories = this.repositoriesManager.folderManagers.map(folder => folder.gitHubRepositories).flat();
 
 			this.codingAgentPRsPromise = this.codingAgentPRsPromise ?? new Promise<vscode.ChatSessionItem[]>(async (resolve) => {
-				const sessions = await capi.getAllSessions(undefined);
+				const sessions = await capi.getAllSessionsForAllRepositories();
 				const sessionMap = new Map<string, SessionInfo[]>();
 
 				for (const session of sessions) {
@@ -952,13 +952,13 @@ export class CopilotRemoteAgentManager extends Disposable {
 				});
 
 				resolve(filteredPRs);
+				this.codingAgentPRsPromise = undefined;
 			});
 
 			return this.codingAgentPRsPromise;
 		} catch (error) {
 			Logger.error(`Failed to provide coding agents information: ${error}`, CopilotRemoteAgentManager.ID);
 		} finally {
-			this.codingAgentPRsPromise = undefined;
 		}
 		return [];
 	}