Add notification when closing chat session in progress

src/vs/workbench/contrib/chat/browser/agentSessions/chatCloseNotification.ts

@@ -0,0 +1,43 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as nls from '../../../../../nls.js';
+import { ServicesAccessor } from '../../../../../platform/instantiation/common/instantiation.js';
+import { INotificationService, NeverShowAgainScope, Severity } from '../../../../../platform/notification/common/notification.js';
+import { IViewsService } from '../../../../services/views/common/viewsService.js';
+import { AGENT_SESSIONS_VIEW_ID } from './agentSessions.js';
+
+const STORAGE_KEY = 'chat.closeWithActiveResponse.doNotShowAgain2';
+
+/**
+ * Shows a notification when closing a chat with an active response, informing the user
+ * that the chat will continue running in the background. The notification includes a button
+ * to open the Agent Sessions view and a "Don't Show Again" option.
+ */
+export function showCloseActiveChatNotification(
+	accessor: ServicesAccessor
+): void {
+	const notificationService = accessor.get(INotificationService);
+	const viewsService = accessor.get(IViewsService);
+
+	notificationService.prompt(
+		Severity.Info,
+		nls.localize('chat.closeWithActiveResponse', "A chat session is in progress. It will continue running in the background."),
+		[
+			{
+				label: nls.localize('chat.openAgentSessions', "Open Agent Sessions"),
+				run: async () => {
+					await viewsService.openView(AGENT_SESSIONS_VIEW_ID, true);
+				}
+			}
+		],
+		{
+			neverShowAgain: {
+				id: STORAGE_KEY,
+				scope: NeverShowAgainScope.APPLICATION
+			}
+		}
+	);
+}

src/vs/workbench/contrib/chat/browser/chatEditorInput.ts

@@ -5,7 +5,6 @@
 
 import { CancellationToken } from '../../../../base/common/cancellation.js';
 import { Codicon } from '../../../../base/common/codicons.js';
-import { Emitter } from '../../../../base/common/event.js';
 import { Disposable, MutableDisposable, toDisposable } from '../../../../base/common/lifecycle.js';
 import { Schemas } from '../../../../base/common/network.js';
 import { isEqual } from '../../../../base/common/resources.js';
@@ -25,6 +24,7 @@ import { IChatSessionsService, localChatSessionType } from '../common/chatSessio
 import { LocalChatSessionUri } from '../common/chatUri.js';
 import { ChatAgentLocation, ChatEditorTitleMaxLength } from '../common/constants.js';
 import { IClearEditingSessionConfirmationOptions } from './actions/chatActions.js';
+import { showCloseActiveChatNotification } from './agentSessions/chatCloseNotification.js';
 import type { IChatEditorOptions } from './chatEditor.js';
 
 const ChatEditorIcon = registerIcon('chat-editor-label-icon', Codicon.chatSparkle, nls.localize('chatEditorLabelIcon', 'Icon of the chat editor label.'));
@@ -77,6 +77,7 @@ export class ChatEditorInput extends EditorInput implements IEditorCloseHandler
 		@IChatService private readonly chatService: IChatService,
 		@IDialogService private readonly dialogService: IDialogService,
 		@IChatSessionsService private readonly chatSessionsService: IChatSessionsService,
+		@IInstantiationService private readonly instantiationService: IInstantiationService,
 	) {
 		super();
 
@@ -316,12 +317,18 @@ export class ChatEditorInput extends EditorInput implements IEditorCloseHandler
 		}
 		return false;
 	}
+
+	override dispose(): void {
+		// Check if we're disposing a model with an active request
+		if (this.modelRef.value?.object.requestInProgress.get()) {
+			this.instantiationService.invokeFunction(showCloseActiveChatNotification);
+		}
+
+		super.dispose();
+	}
 }
 
 export class ChatEditorModel extends Disposable {
-	private _onWillDispose = this._register(new Emitter<void>());
-	readonly onWillDispose = this._onWillDispose.event;
-
 	private _isResolved = false;
 
 	constructor(

src/vs/workbench/contrib/chat/browser/chatViewPane.ts

@@ -36,6 +36,7 @@ import { IChatModelReference, IChatService } from '../common/chatService.js';
 import { IChatSessionsExtensionPoint, IChatSessionsService, localChatSessionType } from '../common/chatSessionsService.js';
 import { LocalChatSessionUri } from '../common/chatUri.js';
 import { ChatAgentLocation, ChatModeKind } from '../common/constants.js';
+import { showCloseActiveChatNotification } from './agentSessions/chatCloseNotification.js';
 import { ChatWidget } from './chatWidget.js';
 import { ChatViewWelcomeController, IViewWelcomeDelegate } from './viewsWelcome/chatViewWelcomeController.js';
 
@@ -148,6 +149,11 @@ export class ChatViewPane extends ViewPane implements IViewWelcomeDelegate {
 	}
 
 	private async updateModel(modelRef?: IChatModelReference | undefined) {
+		// Check if we're disposing a model with an active request
+		if (this.modelRef.value?.object.requestInProgress.get()) {
+			this.instantiationService.invokeFunction(showCloseActiveChatNotification);
+		}
+
 		this.modelRef.value = undefined;
 
 		const ref = modelRef ?? (this.chatService.transferredSessionData?.sessionId && this.chatService.transferredSessionData?.location === this.chatOptions.location