@mastra/client-js 0.12.0 → 0.12.1-alpha.1

package/dist/index.js

@@ -358,7 +358,11 @@ var Agent = class extends BaseResource {
       ...params,
       output: params.output ? zodToJsonSchema(params.output) : void 0,
       runtimeContext: parseClientRuntimeContext(params.runtimeContext),
-      clientTools: processClientTools(params.clientTools)
+      clientTools: processClientTools(params.clientTools),
+      structuredOutput: params.structuredOutput ? {
+        ...params.structuredOutput,
+        schema: zodToJsonSchema(params.structuredOutput.schema)
+      } : void 0
     };
     const { runId, resourceId, threadId, runtimeContext } = processedParams;
     const response = await this.request(
@@ -863,7 +867,7 @@ var Agent = class extends BaseResource {
               step,
               toolCallId: chunk.payload.toolCallId,
               toolName: chunk.payload.toolName,
-              args: void 0
+              args: chunk.payload.args
             };
             message.toolInvocations.push(invocation);
             updateToolInvocationPart(chunk.payload.toolCallId, invocation);
@@ -917,14 +921,14 @@ var Agent = class extends BaseResource {
           }
           case "step-finish": {
             step += 1;
-            currentTextPart = chunk.payload.isContinued ? currentTextPart : void 0;
+            currentTextPart = chunk.payload.stepResult.isContinued ? currentTextPart : void 0;
             currentReasoningPart = void 0;
             currentReasoningTextDetail = void 0;
             execUpdate();
             break;
           }
           case "finish": {
-            finishReason = chunk.payload.finishReason;
+            finishReason = chunk.payload.stepResult.reason;
             if (chunk.payload.usage != null) {
               usage = chunk.payload.usage;
             }
@@ -948,9 +952,28 @@ var Agent = class extends BaseResource {
       let toolCalls = [];
       let messages = [];
       const [streamForWritable, streamForProcessing] = response.body.tee();
-      streamForWritable.pipeTo(writable, {
-        preventClose: true
-      }).catch((error) => {
+      streamForWritable.pipeTo(
+        new WritableStream({
+          async write(chunk) {
+            try {
+              const text = new TextDecoder().decode(chunk);
+              if (text.includes("[DONE]")) {
+                return;
+              }
+            } catch {
+            }
+            const writer = writable.getWriter();
+            try {
+              await writer.write(chunk);
+            } finally {
+              writer.releaseLock();
+            }
+          }
+        }),
+        {
+          preventClose: true
+        }
+      ).catch((error) => {
         console.error("Error piping to writable stream:", error);
       });
       this.processChatResponse_vNext({
@@ -987,7 +1010,8 @@ var Agent = class extends BaseResource {
                     toolCallId: toolCall2?.toolCallId
                   }
                 );
-                const lastMessage = JSON.parse(JSON.stringify(messages[messages.length - 1]));
+                const lastMessageRaw = messages[messages.length - 1];
+                const lastMessage = lastMessageRaw != null ? JSON.parse(JSON.stringify(lastMessageRaw)) : void 0;
                 const toolInvocationPart = lastMessage?.parts?.find(
                   (part) => part.type === "tool-invocation" && part.toolInvocation?.toolCallId === toolCall2.toolCallId
                 );
@@ -1005,25 +1029,13 @@ var Agent = class extends BaseResource {
                   toolInvocation.state = "result";
                   toolInvocation.result = result;
                 }
-                const writer = writable.getWriter();
-                try {
-                  await writer.write(
-                    new TextEncoder().encode(
-                      "a:" + JSON.stringify({
-                        toolCallId: toolCall2.toolCallId,
-                        result
-                      }) + "\n"
-                    )
-                  );
-                } finally {
-                  writer.releaseLock();
-                }
                 const originalMessages = processedParams.messages;
                 const messageArray = Array.isArray(originalMessages) ? originalMessages : [originalMessages];
+                const updatedMessages = lastMessage != null ? [...messageArray, ...messages.filter((m) => m.id !== lastMessage.id), lastMessage] : [...messageArray, ...messages];
                 this.processStreamResponse_vNext(
                   {
                     ...processedParams,
-                    messages: [...messageArray, ...messages.filter((m) => m.id !== lastMessage.id), lastMessage]
+                    messages: updatedMessages
                   },
                   writable
                 ).catch((error) => {
@@ -1051,7 +1063,11 @@ var Agent = class extends BaseResource {
       ...params,
       output: params.output ? zodToJsonSchema(params.output) : void 0,
       runtimeContext: parseClientRuntimeContext(params.runtimeContext),
-      clientTools: processClientTools(params.clientTools)
+      clientTools: processClientTools(params.clientTools),
+      structuredOutput: params.structuredOutput ? {
+        ...params.structuredOutput,
+        schema: zodToJsonSchema(params.structuredOutput.schema)
+      } : void 0
     };
     const { readable, writable } = new TransformStream();
     const response = await this.processStreamResponse_vNext(processedParams, writable);
@@ -1897,6 +1913,57 @@ var Workflow = class extends BaseResource {
     });
     return response.body.pipeThrough(transformStream);
   }
+  /**
+   * Starts a workflow run and returns a stream
+   * @param params - Object containing the optional runId, inputData and runtimeContext
+   * @returns Promise containing the workflow execution results
+   */
+  async streamVNext(params) {
+    const searchParams = new URLSearchParams();
+    if (!!params?.runId) {
+      searchParams.set("runId", params.runId);
+    }
+    const runtimeContext = parseClientRuntimeContext(params.runtimeContext);
+    const response = await this.request(
+      `/api/workflows/${this.workflowId}/streamVNext?${searchParams.toString()}`,
+      {
+        method: "POST",
+        body: { inputData: params.inputData, runtimeContext },
+        stream: true
+      }
+    );
+    if (!response.ok) {
+      throw new Error(`Failed to stream vNext workflow: ${response.statusText}`);
+    }
+    if (!response.body) {
+      throw new Error("Response body is null");
+    }
+    let failedChunk = void 0;
+    const transformStream = new TransformStream({
+      start() {
+      },
+      async transform(chunk, controller) {
+        try {
+          const decoded = new TextDecoder().decode(chunk);
+          const chunks = decoded.split(RECORD_SEPARATOR2);
+          for (const chunk2 of chunks) {
+            if (chunk2) {
+              const newChunk = failedChunk ? failedChunk + chunk2 : chunk2;
+              try {
+                const parsedChunk = JSON.parse(newChunk);
+                controller.enqueue(parsedChunk);
+                failedChunk = void 0;
+              } catch {
+                failedChunk = newChunk;
+              }
+            }
+          }
+        } catch {
+        }
+      }
+    });
+    return response.body.pipeThrough(transformStream);
+  }
   /**
    * Resumes a suspended workflow step asynchronously and returns a promise that resolves when the workflow is complete
    * @param params - Object containing the runId, step, resumeData and runtimeContext
@@ -2459,7 +2526,7 @@ var Observability = class extends BaseResource {
   getTraces(params) {
     const { pagination, filters } = params;
     const { page, perPage, dateRange } = pagination || {};
-    const { name, spanType } = filters || {};
+    const { name, spanType, entityId, entityType } = filters || {};
     const searchParams = new URLSearchParams();
     if (page !== void 0) {
       searchParams.set("page", String(page));
@@ -2473,6 +2540,10 @@ var Observability = class extends BaseResource {
     if (spanType !== void 0) {
       searchParams.set("spanType", String(spanType));
     }
+    if (entityId && entityType) {
+      searchParams.set("entityId", entityId);
+      searchParams.set("entityType", entityType);
+    }
     if (dateRange) {
       const dateRangeStr = JSON.stringify({
         start: dateRange.start instanceof Date ? dateRange.start.toISOString() : dateRange.start,