Update chat's newEditSession command to take an input prompt (microsoft#241796)

* Update chat's newEditSession command to take an input prompt

* Use acceptInput instead

* PR feedback - making a type and supporting partial queries

Author: cmbrose

src/vs/workbench/contrib/chat/browser/actions/chatClearActions.ts

@@ -15,7 +15,7 @@ import { KeybindingWeight } from '../../../../../platform/keybinding/common/keyb
 import { ActiveEditorContext } from '../../../../common/contextkeys.js';
 import { IViewsService } from '../../../../services/views/common/viewsService.js';
 import { isChatViewTitleActionContext } from '../../common/chatActions.js';
-import { ChatAgentLocation } from '../../common/chatAgents.js';
+import { ChatAgentLocation, IChatAgentService } from '../../common/chatAgents.js';
 import { ChatContextKeys } from '../../common/chatContextKeys.js';
 import { hasAppliedChatEditsContextKey, hasUndecidedChatEditingResourceContextKey, IChatEditingSession, WorkingSetEntryState } from '../../common/chatEditingService.js';
 import { ChatViewId, EditsViewId, IChatWidget, IChatWidgetService } from '../chat.js';
@@ -30,6 +30,24 @@ export const ACTION_ID_NEW_CHAT = `workbench.action.chat.newChat`;
 export const ACTION_ID_NEW_EDIT_SESSION = `workbench.action.chat.newEditSession`;
 export const ChatDoneActionId = 'workbench.action.chat.done';
 
+export interface INewEditSessionActionContext {
+	/**
+	 * An initial prompt to write to the chat.
+	 */
+	inputValue?: string;
+
+	/**
+	 * Selects opening in agent mode or not. If not set, the current mode is used.
+	 * This is ignored when coming from a chat view title context.
+	 */
+	agentMode?: boolean;
+
+	/**
+	 * Whether the inputValue is partial and should wait for further user input. If false or not set, the prompt is sent immediately.
+	 */
+	isPartialQuery?: boolean;
+}
+
 export function registerNewChatActions() {
 	registerAction2(class NewChatEditorAction extends Action2 {
 		constructor() {
@@ -175,34 +193,54 @@ export function registerNewChatActions() {
 		}
 
 		async runEditingSessionAction(accessor: ServicesAccessor, editingSession: IChatEditingSession, chatWidget: IChatWidget, ...args: any[]) {
-			const context = args[0];
+			const context: INewEditSessionActionContext | undefined = args[0];
 			const accessibilitySignalService = accessor.get(IAccessibilitySignalService);
 			const widgetService = accessor.get(IChatWidgetService);
 			const dialogService = accessor.get(IDialogService);
 			const viewsService = accessor.get(IViewsService);
+			const agentService = accessor.get(IChatAgentService);
+
 			if (!(await this._handleCurrentEditingSession(editingSession, dialogService))) {
 				return;
 			}
-			if (isChatViewTitleActionContext(context)) {
+
+			const isChatViewTitleAction = isChatViewTitleActionContext(context);
+
+			let widget: IChatWidget | undefined;
+			if (isChatViewTitleAction) {
 				// Is running in the Chat view title
-				announceChatCleared(accessibilitySignalService);
-				const widget = widgetService.getWidgetBySessionId(context.sessionId);
-				if (widget) {
-					await editingSession.stop(true);
-					widget.clear();
-					widget.attachmentModel.clear();
-					widget.focusInput();
-				}
+				widget = widgetService.getWidgetBySessionId(context.sessionId);
 			} else {
 				// Is running from f1 or keybinding
 				const chatView = await viewsService.openView(EditsViewId) as ChatViewPane;
-				const widget = chatView.widget;
+				widget = chatView.widget;
+			}
 
-				announceChatCleared(accessibilitySignalService);
-				await editingSession.stop(true);
-				widget.clear();
-				widget.attachmentModel.clear();
-				widget.focusInput();
+			announceChatCleared(accessibilitySignalService);
+
+			if (!widget) {
+				return;
+			}
+
+			await editingSession.stop(true);
+			widget.clear();
+			widget.attachmentModel.clear();
+			widget.focusInput();
+
+			if (!context) {
+				return;
+			}
+
+			if (!isChatViewTitleAction && typeof context.agentMode === 'boolean') {
+				agentService.toggleToolsAgentMode(context.agentMode);
+			}
+
+			if (context.inputValue) {
+				if (context.isPartialQuery) {
+					widget.setInput(context.inputValue);
+				} else {
+					widget.acceptInput(context.inputValue);
+				}
 			}
 		}
 	});