Update Markdown.tsx

Gerome-Elassaad · Gerome-Elassaad · commit 81e5700fced0 · 2025-12-04T14:26:29.000+11:00
diff --git a/app/components/chat/Markdown.tsx b/app/components/chat/Markdown.tsx
@@ -5,6 +5,7 @@ import { createScopedLogger } from '~/utils/logger';
 import { rehypePlugins, remarkPlugins, allowedHTMLElements } from '~/utils/markdown';
 import { Artifact } from './Artifact';
 import { ThinkingArtifact } from './ThinkingArtifact';
+import { TestArtifact } from './TestArtifact';
 import { CodeBlock } from './CodeBlock';
 import { ThinkingProcess } from './ThinkingProcess';
 
@@ -45,6 +46,16 @@ export const Markdown = memo(({ children, html = false, limitedMarkdown = false
           return <ThinkingArtifact messageId={messageId} />;
         }
 
+        if (className?.includes('__testArtifact__')) {
+          const messageId = node?.properties.dataMessageId as string;
+
+          if (!messageId) {
+            logger.error(`Invalid message id ${messageId}`);
+          }
+
+          return <TestArtifact messageId={messageId} />;
+        }
+
         if (className?.includes('__codinitThought__')) {
           return <ThoughtBox title="Thought process">{children}</ThoughtBox>;
         }
