Ensure user selected tools is setup correctly (microsoft#250027)

Author: DonJayamanne

src/vs/workbench/contrib/chat/browser/chatContentParts/chatConfirmationContentPart.ts

@@ -10,6 +10,8 @@ import { IInstantiationService } from '../../../../../platform/instantiation/com
 import { IChatProgressRenderableResponseContent } from '../../common/chatModel.js';
 import { IChatConfirmation, IChatSendRequestOptions, IChatService } from '../../common/chatService.js';
 import { isResponseVM } from '../../common/chatViewModel.js';
+import { ChatMode } from '../../common/constants.js';
+import { ILanguageModelToolsService } from '../../common/languageModelToolsService.js';
 import { IChatWidgetService } from '../chat.js';
 import { ChatConfirmationWidget } from './chatConfirmationWidget.js';
 import { IChatContentPart, IChatContentPartRenderContext } from './chatContentParts.js';
@@ -26,6 +28,7 @@ export class ChatConfirmationContentPart extends Disposable implements IChatCont
 		@IInstantiationService private readonly instantiationService: IInstantiationService,
 		@IChatService private readonly chatService: IChatService,
 		@IChatWidgetService chatWidgetService: IChatWidgetService,
+		@ILanguageModelToolsService private readonly toolsService: ILanguageModelToolsService,
 	) {
 		super();
 
@@ -56,6 +59,15 @@ export class ChatConfirmationContentPart extends Disposable implements IChatCont
 				const widget = chatWidgetService.getWidgetBySessionId(element.sessionId);
 				options.userSelectedModelId = widget?.input.currentLanguageModel;
 				options.mode = widget?.input.currentMode;
+				if (widget?.input.currentMode2.customTools) {
+					options.userSelectedTools = this.toolsService.toEnablementMap(widget.input.currentMode2.customTools);
+				} else if (widget?.input.currentMode === ChatMode.Agent) {
+					options.userSelectedTools = {};
+					for (const [tool, enablement] of widget.input.selectedToolsModel.asEnablementMap()) {
+						options.userSelectedTools[tool.id] = enablement;
+					}
+				}
+
 				if (await this.chatService.sendRequest(element.sessionId, prompt, options)) {
 					confirmation.isUsed = true;
 					confirmationWidget.setShowButtons(false);