chat session content provider

Author: rebornix

src/@types/vscode.proposed.chatSessionsProvider.d.ts

@@ -22,8 +22,6 @@ declare module 'vscode' {
 		 * Provides a list of chat sessions.
 		 */
 		provideChatSessionItems(token: CancellationToken): ProviderResult<ChatSessionItem[]>;
-
-		provideChatSessionContent(id: string, token: CancellationToken): Thenable<ChatSession>;
 	}
 
 	export interface ChatSessionItem {
@@ -47,7 +45,7 @@ declare module 'vscode' {
 		/**
 		 * The content that was received from the chat participant. Only the stream parts that represent actual content (not metadata) are represented.
 		 */
-		readonly response: ReadonlyArray<ChatResponseMarkdownPart | ChatResponseFileTreePart | ChatResponseAnchorPart | ChatResponseCommandButtonPart | ExtendedChatResponsePart>;
+		readonly response: ReadonlyArray<ChatResponseMarkdownPart | ChatResponseFileTreePart | ChatResponseAnchorPart | ChatResponseCommandButtonPart | ExtendedChatResponsePart | ChatToolInvocationPart>;
 
 		/**
 		 * The result that was received from the chat participant.
@@ -64,9 +62,6 @@ declare module 'vscode' {
 		 */
 		readonly command?: string;
 
-		/**
-		 * @hidden
-		 */
 		constructor(response: ReadonlyArray<ChatResponseMarkdownPart | ChatResponseFileTreePart | ChatResponseAnchorPart | ChatResponseCommandButtonPart | ExtendedChatResponsePart>, result: ChatResult, participant: string);
 	}
 
@@ -101,7 +96,23 @@ declare module 'vscode' {
 		readonly requestHandler: ChatRequestHandler | undefined;
 	}
 
+	export interface ChatSessionContentProvider {
+		/**
+		 * Resolves a chat session into a full `ChatSession` object.
+		 *
+		 * @param uri The URI of the chat session to open. Uris as structured as `vscode-chat-session:<chatSessionType>/id`
+		 * @param token A cancellation token that can be used to cancel the operation.
+		 */
+		provideChatSessionContent(id: string, token: CancellationToken): Thenable<ChatSession>;
+	}
+
 	export namespace chat {
 		export function registerChatSessionItemProvider(chatSessionType: string, provider: ChatSessionItemProvider): Disposable;
+
+		/**
+		 * @param chatSessionType A unique identifier for the chat session type. This is used to differentiate between different chat session providers.
+		 */
+		export function registerChatSessionContentProvider(chatSessionType: string, provider: ChatSessionContentProvider): Disposable;
+
 	}
 }

src/extension.ts

@@ -412,19 +412,26 @@ async function deferredActivate(context: vscode.ExtensionContext, showPRControll
 	const copilotRemoteAgentManager = new CopilotRemoteAgentManager(credentialStore, reposManager, telemetry);
 	context.subscriptions.push(copilotRemoteAgentManager);
 	if (vscode.chat?.registerChatSessionItemProvider) {
+		const provider = new class implements vscode.ChatSessionContentProvider, vscode.ChatSessionItemProvider {
+			label = vscode.l10n.t('GitHub Copilot Coding Agent');
+			provideChatSessionItems = async (token) => {
+				return await copilotRemoteAgentManager.provideChatSessions(token);
+			};
+			provideChatSessionContent = async (id, token) => {
+				return await copilotRemoteAgentManager.provideChatSessionContent(id, token);
+			};
+			// Events not used yet, but required by interface.
+			onDidChangeChatSessionItems = new vscode.EventEmitter<void>().event;
+		}();
+
 		context.subscriptions.push(vscode.chat?.registerChatSessionItemProvider(
 			'copilot-swe-agent',
-			{
-				label: vscode.l10n.t('GitHub Copilot Coding Agent'),
-				provideChatSessionItems: async (token) => {
-					return await copilotRemoteAgentManager.provideChatSessions(token);
-				},
-				provideChatSessionContent: async (id, token) => {
-					return await copilotRemoteAgentManager.provideChatSessionContent(id, token);
-				},
-				// Events not used yet, but required by interface.
-				onDidChangeChatSessionItems: new vscode.EventEmitter<void>().event,
-			}
+			provider
+		));
+
+		context.subscriptions.push(vscode.chat?.registerChatSessionContentProvider(
+			'copilot-swe-agent',
+			provider
 		));
 	}