From e02c967f5bf05f2f4d737577e1bd78016d7470a0 Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Sun, 6 Apr 2025 09:11:23 +0800
Subject: [PATCH] fix(Markdown): Conditionally apply MarkdownShadowDOMRenderer
 for style components based on message content

---
 src/renderer/src/pages/home/Markdown/Markdown.tsx | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/renderer/src/pages/home/Markdown/Markdown.tsx b/src/renderer/src/pages/home/Markdown/Markdown.tsx
index 25faabbc..ecf70dc0 100644
--- a/src/renderer/src/pages/home/Markdown/Markdown.tsx
+++ b/src/renderer/src/pages/home/Markdown/Markdown.tsx
@@ -67,8 +67,7 @@ const Markdown: FC<Props> = ({ message, citationsData }) => {
       },
       code: CodeBlock,
       img: ImagePreview,
-      pre: (props: any) => <pre style={{ overflow: 'visible' }} {...props} />,
-      style: MarkdownShadowDOMRenderer as any
+      pre: (props: any) => <pre style={{ overflow: 'visible' }} {...props} />
     } as Partial<Components>
     return baseComponents
   }, [citationsData])
@@ -77,6 +76,10 @@ const Markdown: FC<Props> = ({ message, citationsData }) => {
     return <p style={{ marginBottom: 5, whiteSpace: 'pre-wrap' }}>{messageContent}</p>
   }
 
+  if (messageContent.includes('<style>')) {
+    components.style = MarkdownShadowDOMRenderer as any
+  }
+
   return (
     <ReactMarkdown
       rehypePlugins={rehypePlugins}