fix(chat): preserve literal tags when copying

Author: j15z

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

@@ -24,27 +24,21 @@ import { useSubmitCopilotFeedback } from '@/hooks/queries/copilot-feedback'
 import { useForkMothershipChat } from '@/hooks/queries/mothership-chats'
 import { useFolderStore } from '@/stores/folders/store'
 
-const SPECIAL_TAGS = 'thinking|options|usage_upgrade|credential|mothership-error|file|question'
-
-export function toCopyableMarkdown(raw: string): string {
-  return raw
-    .replace(new RegExp(`<\\/?(${SPECIAL_TAGS})(?:>[\\s\\S]*?<\\/(${SPECIAL_TAGS})>|>)`, 'g'), '')
-    .trim()
-}
-
 const ICON_CLASS = 'size-[14px]'
 const BUTTON_CLASS =
   'flex size-[26px] items-center justify-center rounded-[6px] text-[var(--text-icon)] transition-colors hover-hover:bg-[var(--surface-hover)] focus-visible:outline-none'
 
 interface MessageActionsProps {
   content: string
+  prepareContentForCopy?: (content: string) => string
   userQuery?: string
   requestId?: string
   messageId?: string
 }
 
 export const MessageActions = memo(function MessageActions({
   content,
+  prepareContentForCopy,
   userQuery,
   requestId,
   messageId,
@@ -70,7 +64,7 @@ export const MessageActions = memo(function MessageActions({
 
   const copyToClipboard = () => {
     if (!content) return
-    const markdown = toCopyableMarkdown(content)
+    const markdown = prepareContentForCopy?.(content) ?? content
     if (!markdown) return
     void copyMessage(markdown)
   }

…/message-actions/message-actions.test.ts …othership-chat/copyable-markdown.test.tsapps/sim/app/workspace/[workspaceId]/components/message-actions/message-actions.test.ts renamed to apps/sim/app/workspace/[workspaceId]/home/components/mothership-chat/copyable-markdown.test.ts apps/sim/app/workspace/[workspaceId]/components/message-actions/message-actions.test.ts renamed to apps/sim/app/workspace/[workspaceId]/home/components/mothership-chat/copyable-markdown.test.ts

@@ -1,5 +1,5 @@
 import { describe, expect, it } from 'vitest'
-import { toCopyableMarkdown } from '@/app/workspace/[workspaceId]/components/message-actions/message-actions'
+import { toCopyableMarkdown } from '@/app/workspace/[workspaceId]/home/components/mothership-chat/copyable-markdown'
 
 describe('toCopyableMarkdown', () => {
   it('preserves message Markdown, including fenced code and its language', () => {
@@ -23,12 +23,25 @@ describe('toCopyableMarkdown', () => {
   it('removes internal structured tags without flattening surrounding Markdown', () => {
     const message = [
       'Before **formatted text**.',
-      '<credential>remove this UI payload</credential>',
+      '<credential>{"type":"service_account","provider":"gmail"}</credential>',
       'After [a link](https://example.com).',
     ].join('\n')
 
     expect(toCopyableMarkdown(message)).toBe(
       ['Before **formatted text**.', '', 'After [a link](https://example.com).'].join('\n')
     )
   })
+
+  it('preserves tag-shaped text that the chat renders literally', () => {
+    const message = [
+      'Document `<credential>example</credential>`.',
+      '',
+      '```html',
+      '<file>example</file>',
+      '<question>example</question>',
+      '```',
+    ].join('\n')
+
+    expect(toCopyableMarkdown(message)).toBe(message)
+  })
 })

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

@@ -0,0 +1,13 @@
+import { sanitizeChatDisplayContent } from '@/app/workspace/[workspaceId]/home/components/message-content/components/chat-content/chat-sanitize'
+import { parseSpecialTags } from '@/app/workspace/[workspaceId]/home/components/message-content/components/special-tags'
+
+export function toCopyableMarkdown(raw: string): string {
+  const displayContent = sanitizeChatDisplayContent(raw)
+  const { segments } = parseSpecialTags(displayContent, false)
+
+  return segments
+    .reduce((markdown, segment) => {
+      return segment.type === 'text' ? markdown + segment.content : markdown
+    }, '')
+    .trim()
+}

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

@@ -29,6 +29,7 @@ import {
   parseLastCredentialTag,
   parseLastQuestionTag,
 } from '@/app/workspace/[workspaceId]/home/components/message-content/components/special-tags'
+import { toCopyableMarkdown } from '@/app/workspace/[workspaceId]/home/components/mothership-chat/copyable-markdown'
 import { nextSizerFloor } from '@/app/workspace/[workspaceId]/home/components/mothership-chat/sizer-floor'
 import { QueuedMessages } from '@/app/workspace/[workspaceId]/home/components/queued-messages'
 import {
@@ -281,6 +282,7 @@ const AssistantMessageRow = memo(function AssistantMessageRow({
           actionsEligible ? (
             <MessageActions
               content={message.content}
+              prepareContentForCopy={toCopyableMarkdown}
               userQuery={precedingUserContent}
               requestId={message.requestId}
               messageId={message.id}