Add session to agents (#4216)

Co-authored-by: Douwe Osinga <douwe@squareup.com>
Co-authored-by: Jack Amadeo <jackamadeo@squareup.com>
Co-authored-by: Jack Amadeo <jackamadeo@block.xyz>

ui/desktop/src/components/ProgressiveMessageList.tsx

@@ -14,7 +14,7 @@
  * - Configurable batch size and delay
  */
 
-import { useState, useEffect, useCallback, useRef } from 'react';
+import { useCallback, useEffect, useRef, useState } from 'react';
 import { Message } from '../types/message';
 import GooseMessage from './GooseMessage';
 import UserMessage from './UserMessage';
@@ -22,10 +22,11 @@ import { CompactionMarker } from './context_management/CompactionMarker';
 import { useContextManager } from './context_management/ContextManager';
 import { NotificationEvent } from '../hooks/useMessageStream';
 import LoadingGoose from './LoadingGoose';
+import { ChatType } from '../types/chat';
 
 interface ProgressiveMessageListProps {
   messages: Message[];
-  chat?: { id: string; messageHistoryIndex: number }; // Make optional for session history
+  chat?: Pick<ChatType, 'sessionId' | 'messageHistoryIndex'>;
   toolCallNotifications?: Map<string, NotificationEvent[]>; // Make optional
   append?: (value: string) => void; // Make optional
   appendMessage?: (message: Message) => void; // Make optional
@@ -152,8 +153,7 @@ export default function ProgressiveMessageList({
   // Render messages up to the current rendered count
   const renderMessages = useCallback(() => {
     const messagesToRender = messages.slice(0, renderedCount);
-
-    const renderedMessages = messagesToRender
+    return messagesToRender
       .map((message, index) => {
         // Use custom render function if provided
         if (renderMessage) {
@@ -170,7 +170,7 @@ export default function ProgressiveMessageList({
 
         const isUser = isUserMessage(message);
 
-        const result = (
+        return (
           <div
             key={message.id && `${message.id}-${message.content.length}`}
             className={`relative ${index === 0 ? 'mt-0' : 'mt-4'} ${isUser ? 'user' : 'assistant'}`}
@@ -192,6 +192,7 @@ export default function ProgressiveMessageList({
                   <CompactionMarker message={message} />
                 ) : (
                   <GooseMessage
+                    sessionId={chat.sessionId}
                     messageHistoryIndex={chat.messageHistoryIndex}
                     message={message}
                     messages={messages}
@@ -210,12 +211,8 @@ export default function ProgressiveMessageList({
             )}
           </div>
         );
-
-        return result;
       })
-      .filter(Boolean); // Filter out null values
-
-    return renderedMessages;
+      .filter(Boolean);
   }, [
     messages,
     renderedCount,