fix: escape heading on user message item

web/screens/Thread/ThreadCenterPanel/TextMessage/MarkdownTextMessage.tsx

@@ -19,11 +19,17 @@
 import { getLanguageFromExtension } from '@/utils/codeLanguageExtension'
 
 export const MarkdownTextMessage = memo(
-  ({ text }: { id: string; text: string }) => {
+  ({ text, isUser }: { id: string; text: string; isUser: boolean }) => {
     const clipboard = useClipboard({ timeout: 1000 })
 
+    // Escapes headings
+    function preprocessMarkdown(text: string): string {
+      if (!isUser) return text
+      return text.replace(/^#{1,6} /gm, (match) => `\\${match}`)
+    }
+
     function extractCodeLines(node: { children: { children: any[] }[] }) {
       const codeLines: any[] = []
 
       // Helper function to extract text recursively from children
       function getTextFromNode(node: {
@@ -31,38 +37,38 @@
         value: any
         children: any[]
       }): string {
         if (node.type === 'text') {
           return node.value
         } else if (node.children) {
           return node.children.map(getTextFromNode).join('')
         }
         return ''
       }
 
       // Traverse each line in the <code> block
       node.children[0].children.forEach(
         (lineNode: {
           type: string
           tagName: string
           value: any
           children: any[]
         }) => {
           if (lineNode.type === 'element' && lineNode.tagName === 'span') {
             const lineContent = getTextFromNode(lineNode)
             codeLines.push(lineContent)
           }
         }
       )
 
       // Join the lines with newline characters for proper formatting
       return codeLines.join('\n')
     }
     function wrapCodeBlocksWithoutVisit() {
       return (tree: { children: any[] }) => {
         tree.children = tree.children.map((node) => {
           if (node.tagName === 'pre' && node.children[0]?.tagName === 'code') {
             const language =
               node.children[0].properties.className?.[1]?.replace(
                 'language-',
                 ''
               )
@@ -204,7 +210,7 @@
             wrapCodeBlocksWithoutVisit,
           ]}
         >
-          {text}
+          {preprocessMarkdown(text)}
         </Markdown>
       </>
     )

web/screens/Thread/ThreadCenterPanel/TextMessage/index.tsx

@@ -141,7 +141,11 @@ const MessageContainer: React.FC<
                 )}
                 dir="ltr"
               >
-                <MarkdownTextMessage id={props.id} text={text} />
+                <MarkdownTextMessage
+                  id={props.id}
+                  text={text}
+                  isUser={isUser}
+                />
               </div>
             )}
           </>