From 36aa13c4f11cf69a102d7a0e94efb4e1238da23f Mon Sep 17 00:00:00 2001
From: one <wangan.cs@gmail.com>
Date: Tue, 15 Apr 2025 06:01:23 +0800
Subject: [PATCH] refactor: merge rehype plugins

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

diff --git a/src/renderer/src/pages/home/Markdown/Markdown.tsx b/src/renderer/src/pages/home/Markdown/Markdown.tsx
index 05b51265..9dd69389 100644
--- a/src/renderer/src/pages/home/Markdown/Markdown.tsx
+++ b/src/renderer/src/pages/home/Markdown/Markdown.tsx
@@ -36,8 +36,6 @@ const Markdown: FC<Props> = ({ message }) => {
   const { t } = useTranslation()
   const { renderInputMessageAsMarkdown, mathEngine } = useSettings()
 
-  const rehypeMath = useMemo(() => (mathEngine === 'KaTeX' ? rehypeKatex : rehypeMathjax), [mathEngine])
-
   const messageContent = useMemo(() => {
     const empty = isEmpty(message.content)
     const paused = message.status === 'paused'
@@ -46,8 +44,8 @@ const Markdown: FC<Props> = ({ message }) => {
   }, [message, t])
 
   const rehypePlugins = useMemo(() => {
-    return [rehypeRaw, [rehypeSanitize, sanitizeSchema], rehypeMath]
-  }, [rehypeMath])
+    return [rehypeRaw, [rehypeSanitize, sanitizeSchema], mathEngine === 'KaTeX' ? rehypeKatex : rehypeMathjax]
+  }, [mathEngine])
 
   const components = useMemo(() => {
     const baseComponents = {