From 333c43fbdf33ad61be89eedecf5b3073785903bc Mon Sep 17 00:00:00 2001
From: choizhang <choizhang@qq.com>
Date: Tue, 22 Apr 2025 01:25:18 +0800
Subject: [PATCH] refactor: Fixed redundant blank characters in message content

---
 lightrag_webui/src/api/lightrag.ts                     |  1 -
 .../src/components/retrieval/ChatMessage.tsx           | 10 +++++-----
 2 files changed, 5 insertions(+), 6 deletions(-)

diff --git a/lightrag_webui/src/api/lightrag.ts b/lightrag_webui/src/api/lightrag.ts
index 641b8486..0528d11d 100644
--- a/lightrag_webui/src/api/lightrag.ts
+++ b/lightrag_webui/src/api/lightrag.ts
@@ -333,7 +333,6 @@ export const queryTextStream = async (
           try {
             const parsed = JSON.parse(line);
             if (parsed.response) {
-              console.log('Received chunk:', parsed.response); // Log for debugging
               onChunk(parsed.response);
             } else if (parsed.error && onError) {
               onError(parsed.error);
diff --git a/lightrag_webui/src/components/retrieval/ChatMessage.tsx b/lightrag_webui/src/components/retrieval/ChatMessage.tsx
index a55d244a..5bb657ca 100644
--- a/lightrag_webui/src/components/retrieval/ChatMessage.tsx
+++ b/lightrag_webui/src/components/retrieval/ChatMessage.tsx
@@ -1,4 +1,4 @@
-import { ReactNode, useCallback, useEffect, useState, useRef } from 'react'
+import { ReactNode, useCallback, useEffect, useRef } from 'react'
 import { Message } from '@/api/lightrag'
 import useTheme from '@/hooks/useTheme'
 import Button from '@/components/ui/Button'
@@ -24,6 +24,8 @@ export type MessageWithError = Message & {
 
 export const ChatMessage = ({ message }: { message: MessageWithError }) => {
   const { t } = useTranslation()
+  // Remove extra spaces around bold text
+  message.content = message.content.replace(/\*\ {3}/g, '').replace(/\ {4}\*\*/g, '**')
 
   const handleCopyMarkdown = useCallback(async () => {
     if (message.content) {
@@ -47,9 +49,9 @@ export const ChatMessage = ({ message }: { message: MessageWithError }) => {
     >
       <pre className="relative break-words whitespace-pre-wrap">
         <ReactMarkdown
-          className="dark:prose-invert max-w-none text-sm" // Removed text-base as it might conflict
+          className="dark:prose-invert max-w-none text-base text-sm"
           remarkPlugins={[remarkGfm, remarkMath]}
-          // Removed rehypeReact as it's often not needed with custom components and can cause issues
+          rehypePlugins={[rehypeReact]}
           skipHtml={false}
           components={{
             code: CodeHighlight
@@ -168,8 +170,6 @@ const CodeHighlight = ({ className, children, node, ...props }: CodeHighlightPro
             .filter(line => !line.trim().startsWith('linkStyle')) // Keep filtering linkStyle
             .join('\n');
 
-          console.log("Rendering Mermaid with debounced, filtered content:", processedContent);
-
           const mermaidId = `mermaid-${Date.now()}`;
           mermaid.render(mermaidId, processedContent)
             .then(({ svg, bindFunctions }) => {