@mastra/client-js 1.21.0-alpha.6 → 1.21.0-alpha.7

package/dist/index.js

@@ -1289,6 +1289,7 @@ var Agent = class extends BaseResource {
     update,
     onToolCall,
     onFinish,
+    onStreamChunk,
     getCurrentDate = () => /* @__PURE__ */ new Date(),
     lastMessage
   }) {
@@ -1356,6 +1357,7 @@ var Agent = class extends BaseResource {
       // TODO: casting as any here because the stream types were all typed as any before in core.
       // but this is completely wrong and this fn is probably broken. Remove ":any" and you'll see a bunch of type errors
       onChunk: async (chunk) => {
+        onStreamChunk?.(chunk);
         switch (chunk.type) {
           case "tripwire": {
             message.parts.push({
@@ -1457,6 +1459,7 @@ var Agent = class extends BaseResource {
                 execUpdate();
               }
             }
+            break;
           }
           case "tool-call-input-streaming-start": {
             if (message.toolInvocations == null) {
@@ -1573,6 +1576,7 @@ var Agent = class extends BaseResource {
     try {
       let toolCalls = [];
       let messages = [];
+      let streamRunId = processedParams.runId;
       const [streamForController, streamForProcessing] = response.body.tee();
       const pipePromise = streamForController.pipeTo(
         new WritableStream({
@@ -1626,25 +1630,76 @@ var Agent = class extends BaseResource {
               const clientTool = processedParams.clientTools?.[toolCall2.toolName];
               if (clientTool && clientTool.execute) {
                 shouldExecuteClientTool = true;
-                const { result, observability } = await executeClientToolWithObservability({
-                  clientTool,
-                  args: toolCall2?.args,
-                  toolName: toolCall2.toolName,
-                  parentContext: getClientToolObservabilityContext(toolCall2),
-                  executeContext: {
-                    requestContext: processedParams.requestContext,
-                    tracingContext: { currentSpan: void 0 },
-                    agent: {
-                      agentId: this.agentId,
-                      messages: response.messages,
-                      toolCallId: toolCall2?.toolCallId,
-                      suspend: async () => {
-                      },
-                      threadId,
-                      resourceId
+                const runId = streamRunId ?? toolCall2.toolCallId;
+                let result;
+                let observability;
+                let synthetic;
+                try {
+                  ({ result, observability } = await executeClientToolWithObservability({
+                    clientTool,
+                    args: toolCall2?.args,
+                    toolName: toolCall2.toolName,
+                    parentContext: getClientToolObservabilityContext(toolCall2),
+                    executeContext: {
+                      requestContext: processedParams.requestContext,
+                      tracingContext: { currentSpan: void 0 },
+                      agent: {
+                        agentId: this.agentId,
+                        messages: response.messages,
+                        toolCallId: toolCall2?.toolCallId,
+                        suspend: async () => {
+                        },
+                        threadId,
+                        resourceId
+                      }
                     }
-                  }
-                });
+                  }));
+                  synthetic = {
+                    type: "tool-result",
+                    runId,
+                    from: "AGENT",
+                    payload: {
+                      toolCallId: toolCall2.toolCallId,
+                      toolName: toolCall2.toolName,
+                      result,
+                      isError: false,
+                      providerExecuted: false
+                    }
+                  };
+                } catch (error) {
+                  synthetic = {
+                    type: "tool-error",
+                    runId,
+                    from: "AGENT",
+                    payload: {
+                      toolCallId: toolCall2.toolCallId,
+                      toolName: toolCall2.toolName,
+                      error,
+                      args: toolCall2?.args,
+                      providerExecuted: false
+                    }
+                  };
+                  result = { error: error instanceof Error ? error.message : String(error) };
+                }
+                try {
+                  await pipePromise;
+                } catch {
+                }
+                try {
+                  const errorForSerialization = synthetic.type === "tool-error" ? synthetic.payload.error : void 0;
+                  const serializedError = errorForSerialization instanceof Error ? {
+                    name: errorForSerialization.name,
+                    message: errorForSerialization.message,
+                    stack: errorForSerialization.stack
+                  } : errorForSerialization;
+                  const payloadForWire = synthetic.type === "tool-error" ? { ...synthetic, payload: { ...synthetic.payload, error: serializedError } } : synthetic;
+                  const sseLine = `data: ${JSON.stringify(payloadForWire)}
+
+`;
+                  controller.enqueue(new TextEncoder().encode(sseLine));
+                } catch (enqueueErr) {
+                  console.error("Failed to enqueue synthetic tool-result chunk:", enqueueErr);
+                }
                 const lastMessageRaw = messages[messages.length - 1];
                 const lastMessage = lastMessageRaw != null ? JSON.parse(JSON.stringify(lastMessageRaw)) : void 0;
                 const toolInvocationPart = lastMessage?.parts?.find(
@@ -1694,6 +1749,11 @@ var Agent = class extends BaseResource {
             controller.close();
           }
         },
+        onStreamChunk: (chunk) => {
+          if (!streamRunId && typeof chunk.runId === "string") {
+            streamRunId = chunk.runId;
+          }
+        },
         lastMessage: void 0
       }).catch(async (error) => {
         console.error("Error processing stream response:", error);