@mastra/client-js 0.16.9 → 0.16.10

package/dist/index.js

@@ -92,7 +92,6 @@ async function sharedProcessMastraStream({
         if (line.startsWith("data: ")) {
           const data = line.slice(6);
           if (data === "[DONE]") {
-            console.info("\u{1F3C1} Stream finished");
             return;
           }
           let json;
@@ -209,11 +208,11 @@ async function executeToolCallAndRespond({
       return response;
     }
     for (const toolCall of toolCalls) {
-      const clientTool = params.clientTools?.[toolCall.toolName];
+      const clientTool = params.clientTools?.[toolCall.payload.toolName];
       if (clientTool && clientTool.execute) {
         const result = await clientTool.execute(
           {
-            context: toolCall?.args,
+            context: toolCall?.payload.args,
             runId,
             resourceId,
             threadId,
@@ -224,7 +223,7 @@ async function executeToolCallAndRespond({
           },
           {
             messages: response.messages,
-            toolCallId: toolCall?.toolCallId
+            toolCallId: toolCall?.payload.toolCallId
           }
         );
         const updatedMessages = [
@@ -234,8 +233,8 @@ async function executeToolCallAndRespond({
             content: [
               {
                 type: "tool-result",
-                toolCallId: toolCall.toolCallId,
-                toolName: toolCall.toolName,
+                toolCallId: toolCall.payload.toolCallId,
+                toolName: toolCall.payload.toolName,
                 result
               }
             ]
@@ -248,6 +247,7 @@ async function executeToolCallAndRespond({
       }
     }
   }
+  return response;
 }
 var AgentVoice = class extends BaseResource {
   constructor(options, agentId) {
@@ -980,7 +980,7 @@ var Agent = class extends BaseResource {
     });
     onFinish?.({ message, finishReason, usage });
   }
-  async processStreamResponse(processedParams, writable, route = "stream") {
+  async processStreamResponse(processedParams, controller, route = "stream") {
     const response = await this.request(`/api/agents/${this.agentId}/${route}`, {
       method: "POST",
       body: processedParams,
@@ -992,29 +992,30 @@ var Agent = class extends BaseResource {
     try {
       let toolCalls = [];
       let messages = [];
-      const [streamForWritable, streamForProcessing] = response.body.tee();
-      streamForWritable.pipeTo(
+      const [streamForController, streamForProcessing] = response.body.tee();
+      const pipePromise = streamForController.pipeTo(
         new WritableStream({
           async write(chunk) {
-            let writer;
             try {
-              writer = writable.getWriter();
               const text = new TextDecoder().decode(chunk);
               const lines = text.split("\n\n");
-              const readableLines = lines.filter((line) => line !== "[DONE]").join("\n\n");
-              await writer.write(new TextEncoder().encode(readableLines));
-            } catch {
-              await writer?.write(chunk);
-            } finally {
-              writer?.releaseLock();
+              const readableLines = lines.filter((line) => line.trim() !== "[DONE]" && line.trim() !== "data: [DONE]").join("\n\n");
+              if (readableLines) {
+                const encoded = new TextEncoder().encode(readableLines);
+                controller.enqueue(encoded);
+              }
+            } catch (error) {
+              console.error("Error enqueueing to controller:", error);
+              controller.enqueue(chunk);
             }
           }
-        }),
-        {
-          preventClose: true
-        }
+        })
       ).catch((error) => {
-        console.error("Error piping to writable stream:", error);
+        console.error("Error piping to controller:", error);
+        try {
+          controller.close();
+        } catch {
+        }
       });
       this.processChatResponse_vNext({
         stream: streamForProcessing,
@@ -1074,31 +1075,36 @@ var Agent = class extends BaseResource {
                   toolInvocation.result = result;
                 }
                 const updatedMessages = lastMessage != null ? [...messages.filter((m) => m.id !== lastMessage.id), lastMessage] : [...messages];
-                this.processStreamResponse(
-                  {
-                    ...processedParams,
-                    messages: updatedMessages
-                  },
-                  writable
-                ).catch((error) => {
-                  console.error("Error processing stream response:", error);
-                });
+                try {
+                  await this.processStreamResponse(
+                    {
+                      ...processedParams,
+                      messages: updatedMessages
+                    },
+                    controller
+                  );
+                } catch (error) {
+                  console.error("Error processing recursive stream response:", error);
+                }
               }
             }
             if (!shouldExecuteClientTool) {
-              setTimeout(() => {
-                writable.close();
-              }, 0);
+              await pipePromise;
+              controller.close();
             }
           } else {
-            setTimeout(() => {
-              writable.close();
-            }, 0);
+            await pipePromise;
+            controller.close();
           }
         },
         lastMessage: void 0
-      }).catch((error) => {
+      }).catch(async (error) => {
         console.error("Error processing stream response:", error);
+        try {
+          await pipePromise;
+          controller.close();
+        } catch {
+        }
       });
     } catch (error) {
       console.error("Error processing stream response:", error);
@@ -1149,8 +1155,13 @@ var Agent = class extends BaseResource {
         schema: zodToJsonSchema(params.structuredOutput.schema)
       } : void 0
     };
-    const { readable, writable } = new TransformStream();
-    const response = await this.processStreamResponse(processedParams, writable);
+    let readableController;
+    const readable = new ReadableStream({
+      start(controller) {
+        readableController = controller;
+      }
+    });
+    const response = await this.processStreamResponse(processedParams, readableController);
     const streamResponse = new Response(readable, {
       status: response.status,
       statusText: response.statusText,
@@ -1167,8 +1178,13 @@ var Agent = class extends BaseResource {
     return streamResponse;
   }
   async approveToolCall(params) {
-    const { readable, writable } = new TransformStream();
-    const response = await this.processStreamResponse(params, writable, "approve-tool-call");
+    let readableController;
+    const readable = new ReadableStream({
+      start(controller) {
+        readableController = controller;
+      }
+    });
+    const response = await this.processStreamResponse(params, readableController, "approve-tool-call");
     const streamResponse = new Response(readable, {
       status: response.status,
       statusText: response.statusText,
@@ -1185,8 +1201,13 @@ var Agent = class extends BaseResource {
     return streamResponse;
   }
   async declineToolCall(params) {
-    const { readable, writable } = new TransformStream();
-    const response = await this.processStreamResponse(params, writable, "decline-tool-call");
+    let readableController;
+    const readable = new ReadableStream({
+      start(controller) {
+        readableController = controller;
+      }
+    });
+    const response = await this.processStreamResponse(params, readableController, "decline-tool-call");
     const streamResponse = new Response(readable, {
       status: response.status,
       statusText: response.statusText,