Merge pull request microsoft#183124 from microsoft/aiday/interactiveEditorIssue4164

Fixing the incorrect pointing of the collapse/expand arrows

Author: aiday-mar

src/vs/workbench/contrib/interactiveEditor/browser/interactiveEditorController.ts

@@ -526,6 +526,7 @@ export class InteractiveEditorController implements IEditorContribution {
 			this._zone.widget.updateStatus('');
 			this._zone.widget.updateMarkdownMessage(renderedMarkdown.element);
 			this._zone.widget.updateToolbar(true);
+			this._zone.widget.updateMarkdownMessageExpansionState(this._activeSession.lastExpansionState);
 
 		} else if (response instanceof EditResponse) {
 			// edit response -> complex...
@@ -625,7 +626,10 @@ export class InteractiveEditorController implements IEditorContribution {
 	}
 
 	updateExpansionState(expand: boolean) {
-		this._zone.widget.updateToggleState(expand);
+		if (this._activeSession) {
+			this._zone.widget.updateMarkdownMessageExpansionState(expand);
+			this._activeSession.lastExpansionState = expand;
+		}
 	}
 
 	feedbackLast(helpful: boolean) {

src/vs/workbench/contrib/interactiveEditor/browser/interactiveEditorSession.ts

@@ -59,6 +59,7 @@ type TelemetryDataClassification = {
 export class Session {
 
 	private _lastInput: string | undefined;
+	private _lastExpansionState: boolean | undefined;
 	private _lastTextModelChanges: LineRangeMapping[] | undefined;
 	private _lastSnapshot: ITextSnapshot | undefined;
 	private readonly _exchange: SessionExchange[] = [];
@@ -92,6 +93,14 @@ export class Session {
 		return this._lastInput;
 	}
 
+	get lastExpansionState() {
+		return this._lastExpansionState ?? false;
+	}
+
+	set lastExpansionState(state: boolean) {
+		this._lastExpansionState = state;
+	}
+
 	get lastSnapshot(): ITextSnapshot | undefined {
 		return this._lastSnapshot;
 	}

src/vs/workbench/contrib/interactiveEditor/browser/interactiveEditorWidget.ts

@@ -444,7 +444,7 @@ export class InteractiveEditorWidget {
 		this._inputEditor.focus();
 	}
 
-	updateToggleState(expand: boolean) {
+	updateMarkdownMessageExpansionState(expand: boolean) {
 		this._ctxMessageCropState.set(expand ? 'expanded' : 'cropped');
 		this._elements.message.style.webkitLineClamp = expand ? '10' : '3';
 		this._onDidChangeHeight.fire();