fix(cli): show agent prompts in previews

Author: brandonkachen

cli/src/components/message-block.tsx

@@ -198,7 +198,9 @@ export const MessageBlock = ({
     const firstLine = lines[0] || ''
 
     const streamingPreview = isStreaming
-      ? `${sanitizePreview(firstLine)}...`
+      ? agentBlock.initialPrompt
+        ? sanitizePreview(agentBlock.initialPrompt)
+        : `${sanitizePreview(firstLine)}...`
       : ''
 
     const finishedPreview =

cli/src/hooks/use-send-message.ts

@@ -621,22 +621,22 @@ export const useSendMessage = ({
                       return msg
                     }
 
-                    const existingBlocks: ContentBlock[] = msg.blocks
-                      ? [...msg.blocks]
-                      : []
+                const existingBlocks: ContentBlock[] = msg.blocks
+                  ? [...msg.blocks]
+                  : []
 
-                    const newAgentBlocks: ContentBlock[] = agents.map(
-                      (agent: any, index: number) => ({
-                        type: 'agent',
-                        agentId: `${toolCallId}-${index}`,
-                        agentName: agent.agent_type || 'Agent',
-                        agentType: agent.agent_type || 'unknown',
-                        content: agent.prompt || '',
-                        status: 'running' as const,
-                        blocks: [] as ContentBlock[],
-                        initialPrompt: agent.prompt || '',
-                      }),
-                    )
+                const newAgentBlocks: ContentBlock[] = agents.map(
+                  (agent: any, index: number) => ({
+                    type: 'agent',
+                    agentId: `${toolCallId}-${index}`,
+                    agentName: agent.agent_type || 'Agent',
+                    agentType: agent.agent_type || 'unknown',
+                    content: '',
+                    status: 'running' as const,
+                    blocks: [] as ContentBlock[],
+                    initialPrompt: agent.prompt || '',
+                  }),
+                )
 
                     return {
                       ...msg,