fix(chat): limit copied content to orchestrator output

Author: j15z

apps/sim/app/workspace/[workspaceId]/home/components/message-content/index.ts

@@ -1,6 +1,6 @@
 export {
   assistantMessageHasRenderableContent,
-  getRenderableMessageText,
+  getOrchestratorMessageText,
   MessageContent,
 } from './message-content'
 export type { MessagePhase } from './utils'

apps/sim/app/workspace/[workspaceId]/home/components/message-content/message-content.test.ts

@@ -27,7 +27,7 @@ import type { ContentBlock } from '../../types'
 import {
   assistantMessageHasVisibleExecutingTool,
   deriveThinkingLabel,
-  getRenderableMessageText,
+  getOrchestratorMessageText,
   parseBlocks,
   shouldSmoothTextSegment,
 } from './message-content'
@@ -101,8 +101,8 @@ function toolEnvelope(
   } as PersistedStreamEventEnvelope
 }
 
-describe('getRenderableMessageText', () => {
-  it('omits span-based subagent text that has no visible agent group', () => {
+describe('getOrchestratorMessageText', () => {
+  it('copies only orchestrator text from span-based messages', () => {
     const blocks: ContentBlock[] = [
       subagentStart('research', 'span-visible', 'main'),
       {
@@ -120,10 +120,10 @@ describe('getRenderableMessageText', () => {
       mainText('Main answer.'),
     ]
 
-    expect(getRenderableMessageText(blocks, 'Fallback.')).toBe('Visible research. \n\nMain answer.')
+    expect(getOrchestratorMessageText(blocks, 'Fallback.')).toBe('Main answer.')
   })
 
-  it('omits legacy subagent text that has no parent group', () => {
+  it('copies only orchestrator text from legacy messages', () => {
     const blocks: ContentBlock[] = [
       { type: 'subagent_text', content: 'Hidden orphan. ', timestamp: 1 },
       {
@@ -141,11 +141,12 @@ describe('getRenderableMessageText', () => {
       mainText('Main answer.'),
     ]
 
-    expect(getRenderableMessageText(blocks, 'Fallback.')).toBe('Visible research. \n\nMain answer.')
+    expect(getOrchestratorMessageText(blocks, 'Fallback.')).toBe('Main answer.')
   })
 
-  it('separates rendered blocks when streamed text has no boundary whitespace', () => {
+  it('separates orchestrator text blocks around excluded subagent output', () => {
     const blocks: ContentBlock[] = [
+      mainText('Starting answer.'),
       subagentStart('research', 'span-visible', 'main'),
       {
         type: 'subagent_text',
@@ -156,7 +157,7 @@ describe('getRenderableMessageText', () => {
       mainText('Main answer.'),
     ]
 
-    expect(getRenderableMessageText(blocks, 'Fallback.')).toBe('Visible research.\n\nMain answer.')
+    expect(getOrchestratorMessageText(blocks, 'Fallback.')).toBe('Starting answer.\n\nMain answer.')
   })
 })
 

apps/sim/app/workspace/[workspaceId]/home/components/message-content/message-content.tsx

@@ -496,27 +496,16 @@ function joinRenderableText(parts: string[]): string {
   return parts.filter(Boolean).join('\n\n')
 }
 
-function getAgentGroupText(items: AgentGroupItem[]): string {
-  return joinRenderableText(
-    items.map((item) => {
-      if (item.type === 'text') return item.content
-      if (item.type === 'agent_group') return getAgentGroupText(item.group.items)
-      return ''
-    })
-  )
-}
-
-/** Returns only the text represented by the same segment tree the transcript renders. */
-export function getRenderableMessageText(blocks: ContentBlock[], fallbackContent: string): string {
+/** Returns only top-level orchestrator text, excluding agent groups and other UI segments. */
+export function getOrchestratorMessageText(
+  blocks: ContentBlock[],
+  fallbackContent: string
+): string {
   const parsed = blocks.length > 0 ? parseBlocks(blocks) : []
   if (parsed.length === 0) return fallbackContent
 
   return joinRenderableText(
-    parsed.map((segment) => {
-      if (segment.type === 'text') return segment.content
-      if (segment.type === 'agent_group') return getAgentGroupText(segment.items)
-      return ''
-    })
+    parsed.map((segment) => (segment.type === 'text' ? segment.content : ''))
   )
 }
 

apps/sim/app/workspace/[workspaceId]/home/components/mothership-chat/copyable-markdown.test.ts

@@ -5,7 +5,7 @@ vi.mock('@/lib/auth/auth-client', () => ({
 }))
 
 import type { WorkspaceFileRecord } from '@/lib/uploads/contexts/workspace'
-import { getRenderableMessageText } from '@/app/workspace/[workspaceId]/home/components/message-content'
+import { getOrchestratorMessageText } from '@/app/workspace/[workspaceId]/home/components/message-content'
 import {
   prepareCopyableMarkdown,
   serializeCopyableMarkdown,
@@ -173,7 +173,7 @@ describe('toCopyableMarkdown', () => {
     expect(refreshWorkspaceFiles).not.toHaveBeenCalled()
   })
 
-  it('copies workspace resources from visible content blocks', () => {
+  it('copies workspace resources from orchestrator content blocks', () => {
     const contentBlocks: ContentBlock[] = [
       { type: 'text', content: 'Read ' },
       { type: 'thinking', content: 'Do not copy this.' },
@@ -184,7 +184,7 @@ describe('toCopyableMarkdown', () => {
       },
     ]
 
-    const content = getRenderableMessageText(contentBlocks, 'Fallback without the resource.')
+    const content = getOrchestratorMessageText(contentBlocks, 'Fallback without the resource.')
 
     expect(toCopyableMarkdown(content, WORKSPACE_FILES)).toBe(
       'Read [notes.md](sim:file/files/notes.md) for details.'

apps/sim/app/workspace/[workspaceId]/home/components/mothership-chat/mothership-chat.tsx

@@ -20,7 +20,7 @@ import { ChatMessageAttachments } from '@/app/workspace/[workspaceId]/home/compo
 import { ChatSurfaceProvider } from '@/app/workspace/[workspaceId]/home/components/chat-surface-context'
 import {
   assistantMessageHasRenderableContent,
-  getRenderableMessageText,
+  getOrchestratorMessageText,
   MessageContent,
   type MessagePhase,
 } from '@/app/workspace/[workspaceId]/home/components/message-content'
@@ -237,7 +237,7 @@ const AssistantMessageRow = memo(function AssistantMessageRow({
   }, [phase])
 
   const getCopyContent = useCallback(
-    () => getRenderableMessageText(blocks, message.content),
+    () => getOrchestratorMessageText(blocks, message.content),
     [blocks, message.content]
   )
   const prepareContentForCopy = useCallback(
@@ -302,7 +302,7 @@ const AssistantMessageRow = memo(function AssistantMessageRow({
             <MessageActions
               content={message.content}
               getCopyContent={getCopyContent}
-              hasCopyContent={Boolean(getRenderableMessageText(blocks, message.content).trim())}
+              hasCopyContent={Boolean(getOrchestratorMessageText(blocks, message.content).trim())}
               prepareContentForCopy={prepareContentForCopy}
               userQuery={precedingUserContent}
               requestId={message.requestId}