From 129eabd4e5d4fec852cd17db3ca86fc9eca79356 Mon Sep 17 00:00:00 2001
From: Justin Chen <54879025+justschen@users.noreply.github.com>
Date: Tue, 24 Jun 2025 01:52:45 -0700
Subject: [PATCH] make sure to handle edit input when changing modes (#252266)

---
 .../contrib/chat/browser/actions/chatExecuteActions.ts         | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/vs/workbench/contrib/chat/browser/actions/chatExecuteActions.ts b/src/vs/workbench/contrib/chat/browser/actions/chatExecuteActions.ts
index c0c5c7dc4c7..22070275294 100644
--- a/src/vs/workbench/contrib/chat/browser/actions/chatExecuteActions.ts
+++ b/src/vs/workbench/contrib/chat/browser/actions/chatExecuteActions.ts
@@ -224,6 +224,9 @@ class ToggleChatModeAction extends Action2 {
 		context.chatWidget.input.setChatMode2(switchToMode);
 
 		if (chatModeCheck.needToClearSession) {
+			if (context.chatWidget.viewModel?.editing) {
+				context.chatWidget.input.dispose();
+			}
 			await commandService.executeCommand(ACTION_ID_NEW_CHAT);
 		}
 	}