Merge branch 'bot_ui_mermaid_wait_stream_ends' into 'master'

Bot UI: Don't render mermaid until message streaming is complete

See merge request postgres-ai/database-lab!911

Author: Tanya301

ui/packages/platform/src/pages/Bot/Messages/Message/Message.tsx

@@ -22,12 +22,14 @@ type BaseMessageProps = {
   formattedTime?: string;
   aiModel?: string
   stateMessage?: StateMessage | null
+  isCurrentStreamMessage?: boolean
 }
 
 type AiMessageProps = BaseMessageProps & {
   isAi: true;
   content: string;
   aiModel: string
+  isCurrentStreamMessage?: boolean
 }
 
 type HumanMessageProps = BaseMessageProps & {
@@ -256,7 +258,8 @@ export const Message = React.memo((props: MessageProps) => {
     created_at,
     isLoading,
     aiModel,
-    stateMessage
+    stateMessage,
+    isCurrentStreamMessage
   } = props;
 
   const [isDebugVisible, setDebugVisible] = useState(false);
@@ -280,7 +283,7 @@ export const Message = React.memo((props: MessageProps) => {
       if (!inline) {
         return (
           <>
-            {matchMermaid && <MermaidDiagram chart={String(children).replace(/\n$/, '')} />}
+            {matchMermaid && !isCurrentStreamMessage && <MermaidDiagram chart={String(children).replace(/\n$/, '')} />}
             <CodeBlock value={String(children).replace(/\n$/, '')} language={match?.[1]} />
           </>
         )

ui/packages/platform/src/pages/Bot/Messages/Messages.tsx

@@ -281,6 +281,7 @@ export const Messages = React.memo(() => {
               isAi
               content={currentStreamMessage.content}
               aiModel={currentStreamMessage.ai_model}
+              isCurrentStreamMessage
             />
           }
           {isWaitingForAnswer &&