Fix layout in tool invocation codeblocks (microsoft#250162)

Author: roblourens

src/vs/workbench/contrib/chat/browser/chatContentParts/chatToolInputOutputContentPart.ts

@@ -107,11 +107,13 @@ export class ChatCollapsibleInputOutputContentPart extends Disposable {
 		private readonly output: IChatCollapsibleOutputData | undefined,
 		isError: boolean,
 		initiallyExpanded: boolean,
+		width: number,
 		@IContextKeyService private readonly contextKeyService: IContextKeyService,
 		@IInstantiationService private readonly _instantiationService: IInstantiationService,
 		@IContextMenuService private readonly _contextMenuService: IContextMenuService,
 	) {
 		super();
+		this._currentWidth = width;
 
 		const titleEl = dom.h('.chat-confirmation-widget-title-inner');
 		const iconEl = dom.h('.chat-confirmation-widget-title-icon');

src/vs/workbench/contrib/chat/browser/chatContentParts/toolInvocationParts/chatInputOutputMarkdownProgressPart.ts

@@ -42,6 +42,7 @@ export class ChatInputOutputMarkdownProgressPart extends BaseChatToolInvocationS
 		input: string,
 		output: IToolResultInputOutputDetails['output'] | undefined,
 		isError: boolean,
+		currentWidthDelegate: () => number,
 		@IInstantiationService instantiationService: IInstantiationService,
 		@IModelService modelService: IModelService,
 		@ILanguageService languageService: ILanguageService,
@@ -116,6 +117,7 @@ export class ChatInputOutputMarkdownProgressPart extends BaseChatToolInvocationS
 			},
 			isError,
 			ChatInputOutputMarkdownProgressPart._expandedByDefault.get(toolInvocation) ?? false,
+			currentWidthDelegate(),
 		));
 		this._codeblocks.push(...collapsibleListPart.codeblocks);
 		this._register(collapsibleListPart.onDidChangeHeight(() => this._onDidChangeHeight.fire()));

src/vs/workbench/contrib/chat/browser/chatContentParts/toolInvocationParts/chatToolInvocationPart.ts

@@ -111,7 +111,8 @@ export class ChatToolInvocationPart extends Disposable implements IChatContentPa
 				this.toolInvocation.originMessage,
 				this.toolInvocation.resultDetails.input,
 				this.toolInvocation.resultDetails.output,
-				!!this.toolInvocation.resultDetails.isError
+				!!this.toolInvocation.resultDetails.isError,
+				this.currentWidthDelegate
 			);
 		}
 
@@ -126,7 +127,8 @@ export class ChatToolInvocationPart extends Disposable implements IChatContentPa
 				this.toolInvocation.originMessage,
 				typeof this.toolInvocation.toolSpecificData.rawInput === 'string' ? this.toolInvocation.toolSpecificData.rawInput : JSON.stringify(this.toolInvocation.toolSpecificData.rawInput, null, 2),
 				undefined,
-				false
+				false,
+				this.currentWidthDelegate
 			);
 		}