diff --git a/lightrag_webui/src/components/retrieval/ChatMessage.tsx b/lightrag_webui/src/components/retrieval/ChatMessage.tsx
index b70e7bf0..42dd9faf 100644
--- a/lightrag_webui/src/components/retrieval/ChatMessage.tsx
+++ b/lightrag_webui/src/components/retrieval/ChatMessage.tsx
@@ -70,10 +70,14 @@ export const ChatMessage = ({ message }: { message: MessageWithError }) => { //
     ? 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) {