@mastra/playground-ui 5.1.19 → 5.1.20-alpha.0

package/dist/index.cjs.js

@@ -5424,27 +5424,46 @@ function MastraRuntimeProvider({
     if (messages.length === 0 || currentThreadId !== threadId || hasNewInitialMessages && currentThreadId === threadId) {
       if (initialMessages && threadId && memory) {
         const convertedMessages = initialMessages?.map((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
           }));
-          const reasoning = message?.parts?.find(({ type }) => type === "reasoning")?.details?.map((detail) => detail?.text)?.join(" ");
+          const formattedParts = (message.parts || []).map((part) => {
+            if (part.type === "reasoning") {
+              return {
+                type: "reasoning",
+                text: part.reasoning || part?.details?.filter((detail) => detail.type === "text")?.map((detail) => detail.text).join(" ")
+              };
+            }
+            if (part.type === "tool-invocation") {
+              if (part.toolInvocation.state === "result") {
+                return {
+                  type: "tool-call",
+                  toolCallId: part.toolInvocation.toolCallId,
+                  toolName: part.toolInvocation.toolName,
+                  args: part.toolInvocation.args,
+                  result: part.toolInvocation.result
+                };
+              }
+            }
+            if (part.type === "file") {
+              return {
+                type: "file",
+                mimeType: part.mimeType,
+                data: part.data
+              };
+            }
+            if (part.type === "text") {
+              return {
+                type: "text",
+                text: part.text
+              };
+            }
+          }).filter(Boolean);
           return {
             ...message,
-            content: [
-              ...reasoning ? [{ type: "reasoning", text: reasoning }] : [],
-              ...typeof message.content === "string" ? [{ type: "text", text: message.content }] : [],
-              ...toolInvocationsAsContentParts,
-              ...attachmentsAsContentParts
-            ]
+            content: [...formattedParts, ...attachmentsAsContentParts]
           };
         }).filter(Boolean);
         setMessages(convertedMessages);