Marshaled sessions

Author: osortega

src/vs/workbench/api/common/extHostChatSessions.ts

@@ -75,7 +75,7 @@ export class ExtHostChatSessions extends Disposable implements ExtHostChatSessio
 		commands.registerArgumentProcessor({
 			processArgument: (arg) => {
 				if (arg && arg.$mid === MarshalledId.ChatSessionContext) {
-					const id = arg.id;
+					const id = arg.session.id;
 					const sessionContent = this._sessionMap.get(id);
 					if (sessionContent) {
 						return sessionContent;

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

@@ -756,7 +756,7 @@ export function registerChatActions() {
 						const contextItem = context.item as ICodingAgentPickerItem;
 						commandService.executeCommand(buttonItem.id, {
 							uri: contextItem.uri,
-							session: contextItem.session,
+							session: contextItem.session?.session,
 							$mid: MarshalledId.ChatSessionContext
 						});
 

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

@@ -24,6 +24,7 @@ import { IMenuService, MenuId, MenuRegistry } from '../../../../platform/actions
 import { getActionBarActions } from '../../../../platform/actions/browser/menuEntryActionViewItem.js';
 import { Disposable, DisposableStore } from '../../../../base/common/lifecycle.js';
 import { ViewPaneContainer } from '../../../browser/parts/views/viewPaneContainer.js';
+import { MarshalledId } from '../../../../base/common/marshallingIds.js';
 import { ViewPane, IViewPaneOptions } from '../../../browser/parts/views/viewPane.js';
 import { Extensions, IViewContainersRegistry, IViewDescriptorService, ViewContainerLocation, IViewsRegistry, IViewDescriptor } from '../../../common/views.js';
 import { IExtensionService } from '../../../services/extensions/common/extensions.js';
@@ -698,7 +699,13 @@ class SessionsRenderer extends Disposable implements ITreeRenderer<IChatSessionI
 		const setupActionBar = () => {
 			templateData.actionBar.clear();
 
-			const actions = menu.getActions({ arg: session, shouldForwardArgs: true });
+			// Create marshalled context for command execution
+			const marshalledSession = {
+				session: session,
+				$mid: MarshalledId.ChatSessionContext
+			};
+
+			const actions = menu.getActions({ arg: marshalledSession, shouldForwardArgs: true });
 
 			const { primary } = getActionBarActions(
 				actions,