From 247cf4411bb91d336f5f791a2ba5b2a15fde6ba3 Mon Sep 17 00:00:00 2001
From: Bassim Shahidy <122117267+AVGVSTVS96@users.noreply.github.com>
Date: Sat, 29 Jun 2024 14:43:07 -0400
Subject: [PATCH] Use Shiki for code highlighting, add language label to code
 blocks

---
 src/components/ChatUI/CodeHighlight.tsx | 9 +--------
 1 file changed, 1 insertion(+), 8 deletions(-)

diff --git a/src/components/ChatUI/CodeHighlight.tsx b/src/components/ChatUI/CodeHighlight.tsx
index 388e9b3d..e52f785f 100644
--- a/src/components/ChatUI/CodeHighlight.tsx
+++ b/src/components/ChatUI/CodeHighlight.tsx
@@ -71,14 +71,7 @@ export const CodeHighlight = ({
           {language}
         </span>
       ) : null}
-      <SyntaxHighlighter
-        className='rounded-lg py-5!'
-        language={language}
-        style={rainglowAzureContrast as SyntaxHighlighterProps['style']}
-        PreTag='div'
-        {...props}>
-        {String(children).trimEnd()}
-      </SyntaxHighlighter>
+      {highlightedCode}
     </div>
   ) : (
     <code className={className} {...props}>