Populate thinking progress to chat editor (#7743)

* Populate thinking progress to chat editor

* fix test

Author: rebornix

common/sessionParsing.ts

@@ -282,9 +282,10 @@ export function parseToolCallDetails(
 			} : undefined
 		};
 	} else if (name === 'think') {
+		const thought = (args as unknown as { thought?: string }).thought || content || 'Thought';
 		return {
-			toolName: 'Thought',
-			invocationMessage: content || 'Thought',
+			toolName: 'think',
+			invocationMessage: thought,
 		};
 	} else if (name === 'report_progress') {
 		const details: ParsedToolCallDetails = {

src/github/copilotRemoteAgent.ts

@@ -1333,7 +1333,7 @@ export class CopilotRemoteAgentManager extends Disposable {
 		return undefined;
 	}
 
-	private createToolInvocationPart(pullRequest: PullRequestModel, toolCall: any, deltaContent: string = ''): vscode.ChatToolInvocationPart | undefined {
+	private createToolInvocationPart(pullRequest: PullRequestModel, toolCall: any, deltaContent: string = ''): vscode.ChatToolInvocationPart | vscode.ChatResponseThinkingProgressPart | undefined {
 		if (!toolCall.function?.name || !toolCall.id) {
 			return undefined;
 		}
@@ -1352,6 +1352,10 @@ export class CopilotRemoteAgentManager extends Disposable {
 			const toolDetails = parseToolCallDetails(toolCall, deltaContent);
 			toolPart.toolName = toolDetails.toolName;
 
+			if (toolCall.toolName === 'think') {
+				return new vscode.ChatResponseThinkingProgressPart(toolCall.invocationMessage);
+			}
+
 			if (toolCall.function.name === 'bash') {
 				toolPart.invocationMessage = new vscode.MarkdownString(`\`\`\`bash\n${toolDetails.invocationMessage}\n\`\`\``);
 			} else {

src/github/copilotRemoteAgent/chatSessionContentBuilder.ts

@@ -301,7 +301,7 @@ export class ChatSessionContentBuilder {
 		delta: AssistantDelta,
 		choice: Choice,
 		pullRequest: PullRequestModel,
-		responseParts: Array<vscode.ChatResponseMarkdownPart | vscode.ChatToolInvocationPart | vscode.ChatResponseMultiDiffPart>,
+		responseParts: Array<vscode.ChatResponseMarkdownPart | vscode.ChatToolInvocationPart | vscode.ChatResponseMultiDiffPart | vscode.ChatResponseThinkingProgressPart>,
 		currentResponseContent: string,
 	): string {
 		if (delta.role === 'assistant') {
@@ -369,7 +369,7 @@ export class ChatSessionContentBuilder {
 		return currentResponseContent;
 	}
 
-	private createToolInvocationPart(pullRequest: PullRequestModel, toolCall: ToolCall, deltaContent: string = ''): vscode.ChatToolInvocationPart | undefined {
+	private createToolInvocationPart(pullRequest: PullRequestModel, toolCall: ToolCall, deltaContent: string = ''): vscode.ChatToolInvocationPart | vscode.ChatResponseThinkingProgressPart | undefined {
 		if (!toolCall.function?.name || !toolCall.id) {
 			return undefined;
 		}
@@ -388,6 +388,10 @@ export class ChatSessionContentBuilder {
 			const toolDetails = parseToolCallDetails(toolCall, deltaContent);
 			toolPart.toolName = toolDetails.toolName;
 
+			if (toolPart.toolName === 'think') {
+				return new vscode.ChatResponseThinkingProgressPart(toolDetails.invocationMessage);
+			}
+
 			if (toolCall.function.name === 'bash') {
 				toolPart.invocationMessage = new vscode.MarkdownString(`\`\`\`bash\n${toolDetails.invocationMessage}\n\`\`\``);
 			} else {

src/test/common/sessionParsing.test.ts

@@ -134,7 +134,7 @@ another non-data line`;
 
 			const result = parseToolCallDetails(toolCall, 'I need to analyze this code');
 
-			assert.strictEqual(result.toolName, 'Thought');
+			assert.strictEqual(result.toolName, 'think');
 			assert.strictEqual(result.invocationMessage, 'I need to analyze this code');
 		});