From aeaa0b32f9e51613633dafb3c3be16d4fc679e82 Mon Sep 17 00:00:00 2001
From: yangdx <gzdaniel@me.com>
Date: Mon, 1 Dec 2025 12:16:43 +0800
Subject: [PATCH] Add mhchem extension support for chemistry formulas in
 ChatMessage

---
 lightrag_webui/src/components/retrieval/ChatMessage.tsx | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/lightrag_webui/src/components/retrieval/ChatMessage.tsx b/lightrag_webui/src/components/retrieval/ChatMessage.tsx
index 7570f503..c9b67a87 100644
--- a/lightrag_webui/src/components/retrieval/ChatMessage.tsx
+++ b/lightrag_webui/src/components/retrieval/ChatMessage.tsx
@@ -76,10 +76,14 @@ export const ChatMessage = ({
     ? message.content
     : (displayContent !== undefined ? displayContent : (message.content || ''))
 
-  // Load KaTeX dynamically
+  // Load KaTeX dynamically with mhchem extension for chemistry formulas
   useEffect(() => {
     const loadKaTeX = async () => {
       try {
+        // First load mhchem extension (must be loaded before rehype-katex)
+        // This enables \ce and \pu commands for chemistry formulas
+        await import('katex/contrib/mhchem');
+        // Then load rehype-katex
         const { default: rehypeKatex } = await import('rehype-katex');
         setKatexPlugin(() => rehypeKatex);
       } catch (error) {