@mastra/playground-ui 5.1.9 → 5.1.10-alpha.0

package/dist/index.cjs.js

@@ -4867,6 +4867,12 @@ class PDFAttachmentAdapter {
 const convertMessage$2 = (message) => {
   return message;
 };
+const handleFinishReason = (finishReason) => {
+  switch (finishReason) {
+    case "tool-calls":
+      throw new Error("Stream finished with reason tool-calls, try increasing maxSteps");
+  }
+};
 const convertToAIAttachments = async (attachments) => {
   const promises = attachments.filter((attachment) => attachment.type === "image" || attachment.type === "document").map(async (attachment) => {
     if (attachment.type === "document") {
@@ -5055,6 +5061,7 @@ function MastraRuntimeProvider({
             { role: "assistant", content: [] }
           );
           setMessages((currentConversation) => [...currentConversation, latestMessage]);
+          handleFinishReason(generateResponse.finishReason);
         }
       } else {
         let updater = function() {
@@ -5184,6 +5191,9 @@ function MastraRuntimeProvider({
           },
           onErrorPart(error) {
             throw new Error(error);
+          },
+          onFinishMessagePart({ finishReason }) {
+            handleFinishReason(finishReason);
           }
         });
       }
@@ -5196,7 +5206,7 @@ function MastraRuntimeProvider({
       setIsRunning(false);
       setMessages((currentConversation) => [
         ...currentConversation,
-        { role: "assistant", content: [{ type: "text", text: `Error: ${error}` }] }
+        { role: "assistant", content: [{ type: "text", text: `${error}` }] }
       ]);
     }
   };