Route omni voice mode to omni input

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

Copilot-Session: be6b07e4-795c-42ba-88ef-329916399579

Author: meganrogge

src/vs/workbench/contrib/chat/browser/voiceInputMode/voiceInputModeActionViewItem.ts

@@ -309,6 +309,8 @@ export interface IVoiceInputModePillOptions {
 	readonly isActive?: IObservable<boolean>;
 	/** Whether the shared Voice Mode transport belongs to this input. */
 	readonly isVoiceActive?: IObservable<boolean>;
+	/** Claim Voice Mode for this host instead of targeting the last focused chat session. */
+	readonly activateVoiceMode?: () => void | Promise<void>;
 }
 
 /**
@@ -768,12 +770,20 @@ export class VoiceInputModeActionViewItem extends BaseActionViewItem {
 		const controller = this.voiceSessionController;
 		if (controller.isConnected.get() || controller.isConnecting.get()) {
 			if (this._options?.isVoiceActive?.get() === false) {
-				await retargetVoiceToCurrentSession(this.commandService, controller);
+				if (this._options.activateVoiceMode) {
+					await this._options.activateVoiceMode();
+				} else {
+					await retargetVoiceToCurrentSession(this.commandService, controller);
+				}
 				return;
 			}
 			controller.disconnect();
 		} else {
-			await retargetVoiceToCurrentSession(this.commandService, controller);
+			if (this._options?.activateVoiceMode) {
+				await this._options.activateVoiceMode();
+			} else {
+				await retargetVoiceToCurrentSession(this.commandService, controller);
+			}
 			const targetWindow = getWindow(this._voiceCell);
 			controller.connect(targetWindow).catch(() => { /* connect failures are surfaced/logged by the controller */ });
 		}

src/vs/workbench/contrib/chat/browser/widget/input/chatInputPart.ts

@@ -3348,6 +3348,10 @@ export class ChatInputPart extends Disposable implements IHistoryNavigationWidge
 					return this.instantiationService.createInstance(VoiceInputModeActionViewItem, action, {
 						isActive: isVoiceInputActive,
 						isVoiceActive: isVoiceSessionActive,
+						activateVoiceMode: isOmniInput ? () => {
+							this.voiceSessionController.setOmniInputActive(true);
+							this.voiceSessionController.setDraftTarget();
+						} : undefined,
 					});
 				}
 				if ((action.id === ChatSubmitAction.ID || action.id === ChatEditingSessionSubmitAction.ID) && action instanceof MenuItemAction) {