Keep explicitly expanded/collapsed reasoning state in MatrixService

Author: lukemelia

packages/host/app/components/ai-assistant/message/index.gts

@@ -75,6 +75,7 @@ interface Signature {
       errors: { id: string; error: Error }[] | undefined;
     };
     index: number;
+    eventId: string;
     monacoSDK: MonacoSDK;
     registerScroller: (args: {
       index: number;
@@ -200,6 +201,7 @@ function isPresent(val: SafeString | string | null | undefined) {
 
 export default class AiAssistantMessage extends Component<Signature> {
   @service private declare cardService: CardService;
+  @service private declare matrixService: MatrixService;
   get isReasoningExpandedByDefault() {
     let result =
       this.args.isStreaming &&
@@ -208,15 +210,18 @@ export default class AiAssistantMessage extends Component<Signature> {
       !isThinkingMessage(this.args.reasoningContent);
     return result;
   }
-  @tracked _explicitlySetReasoningExpanded: boolean | undefined;
   get isReasoningExpanded() {
     return (
-      this._explicitlySetReasoningExpanded ?? this.isReasoningExpandedByDefault
+      this.matrixService.reasoningExpandedState.get(this.args.eventId) ??
+      this.isReasoningExpandedByDefault
     );
   }
   updateReasoningExpanded = (ev: MouseEvent) => {
     ev.preventDefault();
-    this._explicitlySetReasoningExpanded = !this.isReasoningExpanded;
+    this.matrixService.reasoningExpandedState.set(
+      this.args.eventId,
+      !this.isReasoningExpanded,
+    );
   };
 
   <template>

packages/host/app/components/matrix/room-message.gts

@@ -108,6 +108,7 @@ export default class RoomMessage extends Component<Signature> {
         @reasoningContent={{this.message.reasoningContent}}
         @monacoSDK={{@monacoSDK}}
         @datetime={{this.message.created}}
+        @eventId={{this.message.eventId}}
         @index={{@index}}
         @registerScroller={{@registerScroller}}
         @isFromAssistant={{this.isFromAssistant}}

packages/host/app/services/matrix-service.ts

@@ -142,6 +142,7 @@ export default class MatrixService extends Service {
   cardsToSend: TrackedMap<string, CardDef[] | undefined> = new TrackedMap();
   filesToSend: TrackedMap<string, FileDef[] | undefined> = new TrackedMap();
   failedCommandState: TrackedMap<string, Error> = new TrackedMap();
+  reasoningExpandedState: TrackedMap<string, boolean> = new TrackedMap();
   flushTimeline: Promise<void> | undefined;
   flushMembership: Promise<void> | undefined;
   flushRoomState: Promise<void> | undefined;