@mastra/playground-ui 5.1.2 → 5.1.3

package/dist/index.cjs.js

@@ -4647,22 +4647,26 @@ function MastraRuntimeProvider({
     if (messages.length === 0 || currentThreadId !== threadId || hasNewInitialMessages && currentThreadId === threadId) {
       if (initialMessages && threadId && memory) {
         const convertedMessages = initialMessages?.map((message) => {
-          if (message?.toolInvocations?.length > 0) {
-            return {
-              ...message,
-              content: [
-                ...typeof message.content === "string" ? [{ type: "text", text: message.content }] : [],
-                ...message.toolInvocations.map((toolInvocation) => ({
-                  type: "tool-call",
-                  toolCallId: toolInvocation?.toolCallId,
-                  toolName: toolInvocation?.toolName,
-                  args: toolInvocation?.args,
-                  result: toolInvocation?.result
-                }))
-              ]
-            };
-          }
-          return message;
+          const toolInvocationsAsContentParts = (message.toolInvocations || []).map((toolInvocation) => ({
+            type: "tool-call",
+            toolCallId: toolInvocation?.toolCallId,
+            toolName: toolInvocation?.toolName,
+            args: toolInvocation?.args,
+            result: toolInvocation?.result
+          }));
+          const attachmentsAsContentParts = (message.experimental_attachments || []).map((image) => ({
+            type: image.contentType.startsWith(`image/`) ? "image" : image.contentType.startsWith(`audio/`) ? "audio" : "file",
+            mimeType: image.contentType,
+            image: image.url
+          }));
+          return {
+            ...message,
+            content: [
+              ...typeof message.content === "string" ? [{ type: "text", text: message.content }] : [],
+              ...toolInvocationsAsContentParts,
+              ...attachmentsAsContentParts
+            ]
+          };
         }).filter(Boolean);
         setMessages(convertedMessages);
         setCurrentThreadId(threadId);