adopt provideNewChatSessionItem (#7632)

Author: joshspicer

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

@@ -33,16 +33,29 @@ declare module 'vscode' {
 		 */
 		readonly onDidChangeChatSessionItems: Event<void>;
 
-		// /**
-		//  * Create a new chat session item
-		//  */
-		// provideNewChatSessionItem(context: {
-		// 	// This interface should be extracted
-		// 	readonly triggerChat?: {
-		// 		readonly prompt: string;
-		// 		readonly history: ReadonlyArray<ChatRequestTurn | ChatResponseTurn>;
-		// 	};
-		// }, token: CancellationToken): Thenable<ChatSessionItem> | ChatSessionItem;
+		/**
+		 * Creates a new chat session.
+		 *
+		 * @param options Options for the new session including an optional initial prompt and history
+		 * @param token A cancellation token
+		 * @returns Metadata for the chat session
+		 */
+		provideNewChatSessionItem?(options: {
+			/**
+			 * Initial prompt to initiate the session
+			 */
+			readonly prompt?: string;
+
+			/**
+			 * History to initialize the session with
+			 */
+			readonly history?: ReadonlyArray<ChatRequestTurn | ChatResponseTurn>;
+
+			/**
+			 * Additional metadata to use for session creation
+			 */
+			metadata?: any;
+		}, token: CancellationToken): ProviderResult<ChatSessionItem>;
 
 		/**
 		 * Provides a list of chat sessions.
@@ -109,6 +122,7 @@ declare module 'vscode' {
 		 * If not set, then the session will be considered read-only and no requests can be made.
 		 */
 		// TODO: Should we introduce our own type for `ChatRequestHandler` since not all field apply to chat sessions?
+		// TODO: Revisit this to align with code.
 		readonly requestHandler: ChatRequestHandler | undefined;
 	}
 

src/extension.ts

@@ -433,6 +433,9 @@ async function deferredActivate(context: vscode.ExtensionContext, showPRControll
 				return await copilotRemoteAgentManager.provideChatSessionContent(id, token);
 			};
 			onDidChangeChatSessionItems = copilotRemoteAgentManager.onDidChangeChatSessions;
+			provideNewChatSessionItem = async (options: { prompt?: string; history: ReadonlyArray<vscode.ChatRequestTurn | vscode.ChatResponseTurn>; metadata?: any; }, token: vscode.CancellationToken): Promise<vscode.ChatSessionItem> => {
+				return await copilotRemoteAgentManager.provideNewChatSessionItem(options, token);
+			};
 		}();
 
 		context.subscriptions.push(vscode.chat?.registerChatSessionItemProvider(

src/github/copilotRemoteAgent.ts

@@ -666,6 +666,40 @@ export class CopilotRemoteAgentManager extends Disposable {
 		return this._stateModel.getCounts();
 	}
 
+	public async provideNewChatSessionItem(options: { prompt?: string; history: ReadonlyArray<vscode.ChatRequestTurn | vscode.ChatResponseTurn>; metadata?: any; }, _token: vscode.CancellationToken): Promise<ChatSessionWithPR> {
+		const { prompt } = options;
+		if (!prompt) {
+			throw new Error(`Prompt is expected to provide a new chat session item`);
+		}
+		const result = await this.invokeRemoteAgent(
+			prompt,
+			prompt,
+			false,
+		);
+		if (result.state !== 'success') {
+			Logger.error(`Failed to provide new chat session item: ${result.error}`, CopilotRemoteAgentManager.ID);
+			throw new Error(`Failed to provide new chat session item: ${result.error}`);
+		}
+
+		const { number } = result;
+
+		const session = await this.findPullRequestById(number, true);
+		if (!session) {
+			throw new Error(`Failed to find session for pull request: ${number}`);
+		}
+		const timeline = await session.getCopilotTimelineEvents(session);
+		const status = copilotEventToStatus(mostRecentCopilotEvent(timeline));
+		const tooltip = await issueMarkdown(session, this.context, this.repositoriesManager);
+		return {
+			id: `${session.number}`,
+			label: session.title || `Session ${session.number}`,
+			iconPath: this.getIconForSession(status),
+			description: `${dateFromNow(session.createdAt)}`,
+			pullRequest: session,
+			tooltip,
+		};
+	}
+
 	public async provideChatSessions(token: vscode.CancellationToken): Promise<ChatSessionWithPR[]> {
 		try {
 			const capi = await this.copilotApi;
@@ -700,8 +734,6 @@ export class CopilotRemoteAgentManager extends Disposable {
 		return [];
 	}
 
-
-
 	public async provideChatSessionContent(id: string, token: vscode.CancellationToken): Promise<vscode.ChatSession> {
 		try {
 			const capi = await this.copilotApi;