From 60d88a3514489086a09ea1f04a978f032c1cfbcf Mon Sep 17 00:00:00 2001
From: minicx <minicx@disroot.org>
Date: Fri, 13 Dec 2024 01:42:22 +0300
Subject: [PATCH] fix(llm-markdown-code-rendering)

- Added replace for bad markdown code generated by LLM model

Signed-off-by: minicx <minicx@disroot.org>
---
 ui/components/MessageBox.tsx | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/ui/components/MessageBox.tsx b/ui/components/MessageBox.tsx
index f23127c..d6fb508 100644
--- a/ui/components/MessageBox.tsx
+++ b/ui/components/MessageBox.tsx
@@ -51,11 +51,15 @@ const MessageBox = ({
       message.sources.length > 0
     ) {
       return setParsedMessage(
-        message.content.replace(
-          regex,
-          (_, number) =>
-            `<a href="${message.sources?.[number - 1]?.metadata?.url}" target="_blank" className="bg-light-secondary dark:bg-dark-secondary px-1 rounded ml-1 no-underline text-xs text-black/70 dark:text-white/70 relative">${number}</a>`,
-        ),
+        message.content
+          .replace(
+            regex,
+            (_, number) =>
+              `<a href="${message.sources?.[number - 1]?.metadata?.url}" target="_blank" className="bg-light-secondary dark:bg-dark-secondary px-1 rounded ml-1 no-underline text-xs text-black/70 dark:text-white/70 relative">${number}</a>`,
+          )
+          .replace(/(\*\*.*?\*\*\s*)(\s*```[\s\S]*?```)/g, (match, p1, p2) => {
+            return `${p1}\n${p2.trim()}`;
+          }),
       );
     }