sessions: set session title to first message text immediately

src/vs/platform/agentHost/node/agentService.ts

@@ -176,11 +176,16 @@ export class AgentService extends Disposable implements IAgentService {
 			return s;
 		}));
 
-		// Overlay live session status from the state manager
+		// Overlay live session state from the state manager
 		const withStatus = result.map(s => {
 			const liveState = this._stateManager.getSessionState(s.session.toString());
 			if (liveState) {
-				return { ...s, status: liveState.summary.status, model: liveState.summary.model ?? s.model };
+				return {
+					...s,
+					summary: liveState.summary.title ?? s.summary,
+					status: liveState.summary.status,
+					model: liveState.summary.model ?? s.model,
+				};
 			}
 			return s;
 		});

src/vs/platform/agentHost/node/agentSideEffects.ts

@@ -555,6 +555,19 @@ export class AgentSideEffects extends Disposable {
 				for (const mapper of this._eventMappers.values()) {
 					mapper.reset(action.session);
 				}
+
+				// On the very first turn, immediately set the session title to the
+				// user's message so the UI shows a meaningful title right away
+				// instead of "New Session" while waiting for the AI-generated title.
+				const state = this._stateManager.getSessionState(action.session);
+				if (state && state.turns.length === 0) {
+					this._stateManager.dispatchServerAction({
+						type: ActionType.SessionTitleChanged,
+						session: action.session,
+						title: action.userMessage.text,
+					});
+				}
+
 				const agent = this._options.getAgent(action.session);
 				if (!agent) {
 					this._stateManager.dispatchServerAction({

src/vs/sessions/contrib/localAgentHost/browser/localAgentHostSessionsProvider.ts

@@ -280,6 +280,10 @@ export class LocalAgentHostSessionsProvider extends Disposable implements IAgent
 				this._handleSessionAdded(n.summary);
 			} else if (n.type === 'notify/sessionRemoved') {
 				this._handleSessionRemoved(n.session);
+			} else if (n.type === 'notify/sessionSummaryChanged') {
+				if (n.changes.title !== undefined) {
+					this._handleTitleChanged(n.session, n.changes.title);
+				}
 			}
 		}));
 

src/vs/sessions/contrib/remoteAgentHost/browser/remoteAgentHostSessionsProvider.ts

@@ -396,6 +396,10 @@ export class RemoteAgentHostSessionsProvider extends Disposable implements IAgen
 				this._handleSessionAdded(n.summary);
 			} else if (n.type === 'notify/sessionRemoved') {
 				this._handleSessionRemoved(n.session);
+			} else if (n.type === 'notify/sessionSummaryChanged') {
+				if (n.changes.title !== undefined) {
+					this._handleTitleChanged(n.session, n.changes.title);
+				}
 			}
 		}));