fix: set args for Retry action in chat editing widget (microsoft#232318)

Author: joyceerhl

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

@@ -208,13 +208,17 @@ export function registerChatTitleActions() {
 		}
 
 		async run(accessor: ServicesAccessor, ...args: any[]) {
-			const item = args[0];
+			const chatWidgetService = accessor.get(IChatWidgetService);
+
+			let item = args[0];
+			if (typeof item === 'object' && !!item && 'sessionId' in item) {
+				item = chatWidgetService.getWidgetBySessionId(item.sessionId)?.viewModel?.getItems().at(-1);
+			}
 			if (!isResponseVM(item)) {
 				return;
 			}
 
 			const chatService = accessor.get(IChatService);
-			const chatWidgetService = accessor.get(IChatWidgetService);
 			const chatEditingService = accessor.get(IChatEditingService);
 			const chatModel = chatService.getSession(item.sessionId);
 			const chatRequests = chatModel?.getRequests();

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

@@ -993,6 +993,9 @@ export class ChatInputPart extends Disposable implements IHistoryNavigationWidge
 
 		this._chatEditsActionsDisposables.add(this.instantiationService.createInstance(MenuWorkbenchButtonBar, actionsContainer, MenuId.ChatEditingWidgetToolbar, {
 			telemetrySource: this.options.menus.telemetrySource,
+			menuOptions: {
+				arg: { sessionId: chatEditingSession.chatSessionId },
+			},
 			buttonConfigProvider: (action) => {
 				if (action.id === ChatEditingShowChangesAction.ID || action.id === ChatEditingSaveAllAction.ID) {
 					return { showIcon: true, showLabel: false, isSecondary: true };