@mastra/client-js 0.13.2 → 0.14.0-alpha.1

package/dist/index.js

@@ -223,11 +223,7 @@ async function executeToolCallAndRespond({
           }
         );
         const updatedMessages = [
-          {
-            role: "user",
-            content: params.messages
-          },
-          ...response.response.messages,
+          ...response.response.messages || [],
           {
             role: "tool",
             content: [
@@ -368,10 +364,6 @@ var Agent = class extends BaseResource {
             }
           );
           const updatedMessages = [
-            {
-              role: "user",
-              content: params.messages
-            },
             ...response.response.messages,
             {
               role: "tool",
@@ -1042,9 +1034,11 @@ var Agent = class extends BaseResource {
             if (toolCall) {
               toolCalls.push(toolCall);
             }
+            let shouldExecuteClientTool = false;
             for (const toolCall2 of toolCalls) {
               const clientTool = processedParams.clientTools?.[toolCall2.toolName];
               if (clientTool && clientTool.execute) {
+                shouldExecuteClientTool = true;
                 const result = await clientTool.execute(
                   {
                     context: toolCall2?.args,
@@ -1081,9 +1075,7 @@ var Agent = class extends BaseResource {
                   toolInvocation.state = "result";
                   toolInvocation.result = result;
                 }
-                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];
+                const updatedMessages = lastMessage != null ? [...messages.filter((m) => m.id !== lastMessage.id), lastMessage] : [...messages];
                 this.processStreamResponse_vNext(
                   {
                     ...processedParams,
@@ -1095,6 +1087,11 @@ var Agent = class extends BaseResource {
                 });
               }
             }
+            if (!shouldExecuteClientTool) {
+              setTimeout(() => {
+                writable.close();
+              }, 0);
+            }
           } else {
             setTimeout(() => {
               writable.close();
@@ -1259,12 +1256,10 @@ var Agent = class extends BaseResource {
                 } finally {
                   writer.releaseLock();
                 }
-                const originalMessages = processedParams.messages;
-                const messageArray = Array.isArray(originalMessages) ? originalMessages : [originalMessages];
                 this.processStreamResponse(
                   {
                     ...processedParams,
-                    messages: [...messageArray, ...messages.filter((m) => m.id !== lastMessage.id), lastMessage]
+                    messages: [...messages.filter((m) => m.id !== lastMessage.id), lastMessage]
                   },
                   writable
                 ).catch((error) => {
@@ -1511,193 +1506,6 @@ var Vector = class extends BaseResource {
   }
 };
 
-// src/resources/legacy-workflow.ts
-var RECORD_SEPARATOR = "";
-var LegacyWorkflow = class extends BaseResource {
-  constructor(options, workflowId) {
-    super(options);
-    this.workflowId = workflowId;
-  }
-  /**
-   * Retrieves details about the legacy workflow
-   * @param runtimeContext - Optional runtime context to pass as query parameter
-   * @returns Promise containing legacy workflow details including steps and graphs
-   */
-  details(runtimeContext) {
-    return this.request(`/api/workflows/legacy/${this.workflowId}${runtimeContextQueryString(runtimeContext)}`);
-  }
-  /**
-   * Retrieves all runs for a legacy workflow
-   * @param params - Parameters for filtering runs
-   * @param runtimeContext - Optional runtime context to pass as query parameter
-   * @returns Promise containing legacy workflow runs array
-   */
-  runs(params, runtimeContext) {
-    const runtimeContextParam = base64RuntimeContext(parseClientRuntimeContext(runtimeContext));
-    const searchParams = new URLSearchParams();
-    if (params?.fromDate) {
-      searchParams.set("fromDate", params.fromDate.toISOString());
-    }
-    if (params?.toDate) {
-      searchParams.set("toDate", params.toDate.toISOString());
-    }
-    if (params?.limit) {
-      searchParams.set("limit", String(params.limit));
-    }
-    if (params?.offset) {
-      searchParams.set("offset", String(params.offset));
-    }
-    if (params?.resourceId) {
-      searchParams.set("resourceId", params.resourceId);
-    }
-    if (runtimeContextParam) {
-      searchParams.set("runtimeContext", runtimeContextParam);
-    }
-    if (searchParams.size) {
-      return this.request(`/api/workflows/legacy/${this.workflowId}/runs?${searchParams}`);
-    } else {
-      return this.request(`/api/workflows/legacy/${this.workflowId}/runs`);
-    }
-  }
-  /**
-   * Creates a new legacy workflow run
-   * @returns Promise containing the generated run ID
-   */
-  createRun(params) {
-    const searchParams = new URLSearchParams();
-    if (!!params?.runId) {
-      searchParams.set("runId", params.runId);
-    }
-    return this.request(`/api/workflows/legacy/${this.workflowId}/create-run?${searchParams.toString()}`, {
-      method: "POST"
-    });
-  }
-  /**
-   * Starts a legacy workflow run synchronously without waiting for the workflow to complete
-   * @param params - Object containing the runId and triggerData
-   * @returns Promise containing success message
-   */
-  start(params) {
-    return this.request(`/api/workflows/legacy/${this.workflowId}/start?runId=${params.runId}`, {
-      method: "POST",
-      body: params?.triggerData
-    });
-  }
-  /**
-   * Resumes a suspended legacy workflow step synchronously without waiting for the workflow to complete
-   * @param stepId - ID of the step to resume
-   * @param runId - ID of the legacy workflow run
-   * @param context - Context to resume the legacy workflow with
-   * @returns Promise containing the legacy workflow resume results
-   */
-  resume({
-    stepId,
-    runId,
-    context
-  }) {
-    return this.request(`/api/workflows/legacy/${this.workflowId}/resume?runId=${runId}`, {
-      method: "POST",
-      body: {
-        stepId,
-        context
-      }
-    });
-  }
-  /**
-   * Starts a workflow run asynchronously and returns a promise that resolves when the workflow is complete
-   * @param params - Object containing the optional runId and triggerData
-   * @returns Promise containing the workflow execution results
-   */
-  startAsync(params) {
-    const searchParams = new URLSearchParams();
-    if (!!params?.runId) {
-      searchParams.set("runId", params.runId);
-    }
-    return this.request(`/api/workflows/legacy/${this.workflowId}/start-async?${searchParams.toString()}`, {
-      method: "POST",
-      body: params?.triggerData
-    });
-  }
-  /**
-   * Resumes a suspended legacy workflow step asynchronously and returns a promise that resolves when the workflow is complete
-   * @param params - Object containing the runId, stepId, and context
-   * @returns Promise containing the workflow resume results
-   */
-  resumeAsync(params) {
-    return this.request(`/api/workflows/legacy/${this.workflowId}/resume-async?runId=${params.runId}`, {
-      method: "POST",
-      body: {
-        stepId: params.stepId,
-        context: params.context
-      }
-    });
-  }
-  /**
-   * Creates an async generator that processes a readable stream and yields records
-   * separated by the Record Separator character (\x1E)
-   *
-   * @param stream - The readable stream to process
-   * @returns An async generator that yields parsed records
-   */
-  async *streamProcessor(stream) {
-    const reader = stream.getReader();
-    let doneReading = false;
-    let buffer = "";
-    try {
-      while (!doneReading) {
-        const { done, value } = await reader.read();
-        doneReading = done;
-        if (done && !value) continue;
-        try {
-          const decoded = value ? new TextDecoder().decode(value) : "";
-          const chunks = (buffer + decoded).split(RECORD_SEPARATOR);
-          buffer = chunks.pop() || "";
-          for (const chunk of chunks) {
-            if (chunk) {
-              if (typeof chunk === "string") {
-                try {
-                  const parsedChunk = JSON.parse(chunk);
-                  yield parsedChunk;
-                } catch {
-                }
-              }
-            }
-          }
-        } catch {
-        }
-      }
-      if (buffer) {
-        try {
-          yield JSON.parse(buffer);
-        } catch {
-        }
-      }
-    } finally {
-      reader.cancel().catch(() => {
-      });
-    }
-  }
-  /**
-   * Watches legacy workflow transitions in real-time
-   * @param runId - Optional run ID to filter the watch stream
-   * @returns AsyncGenerator that yields parsed records from the legacy workflow watch stream
-   */
-  async watch({ runId }, onRecord) {
-    const response = await this.request(`/api/workflows/legacy/${this.workflowId}/watch?runId=${runId}`, {
-      stream: true
-    });
-    if (!response.ok) {
-      throw new Error(`Failed to watch legacy workflow: ${response.statusText}`);
-    }
-    if (!response.body) {
-      throw new Error("Response body is null");
-    }
-    for await (const record of this.streamProcessor(response.body)) {
-      onRecord(record);
-    }
-  }
-};
-
 // src/resources/tool.ts
 var Tool = class extends BaseResource {
   constructor(options, toolId) {
@@ -1734,7 +1542,7 @@ var Tool = class extends BaseResource {
 };
 
 // src/resources/workflow.ts
-var RECORD_SEPARATOR2 = "";
+var RECORD_SEPARATOR = "";
 var Workflow = class extends BaseResource {
   constructor(options, workflowId) {
     super(options);
@@ -1758,7 +1566,7 @@ var Workflow = class extends BaseResource {
         if (done && !value) continue;
         try {
           const decoded = value ? new TextDecoder().decode(value) : "";
-          const chunks = (buffer + decoded).split(RECORD_SEPARATOR2);
+          const chunks = (buffer + decoded).split(RECORD_SEPARATOR);
           buffer = chunks.pop() || "";
           for (const chunk of chunks) {
             if (chunk) {
@@ -1896,10 +1704,20 @@ var Workflow = class extends BaseResource {
     return {
       runId,
       start: async (p) => {
-        return this.start({ runId, inputData: p.inputData, runtimeContext: p.runtimeContext });
+        return this.start({
+          runId,
+          inputData: p.inputData,
+          runtimeContext: p.runtimeContext,
+          tracingOptions: p.tracingOptions
+        });
       },
       startAsync: async (p) => {
-        return this.startAsync({ runId, inputData: p.inputData, runtimeContext: p.runtimeContext });
+        return this.startAsync({
+          runId,
+          inputData: p.inputData,
+          runtimeContext: p.runtimeContext,
+          tracingOptions: p.tracingOptions
+        });
       },
       watch: async (onRecord) => {
         return this.watch({ runId }, onRecord);
@@ -1908,10 +1726,22 @@ var Workflow = class extends BaseResource {
         return this.stream({ runId, inputData: p.inputData, runtimeContext: p.runtimeContext });
       },
       resume: async (p) => {
-        return this.resume({ runId, step: p.step, resumeData: p.resumeData, runtimeContext: p.runtimeContext });
+        return this.resume({
+          runId,
+          step: p.step,
+          resumeData: p.resumeData,
+          runtimeContext: p.runtimeContext,
+          tracingOptions: p.tracingOptions
+        });
       },
       resumeAsync: async (p) => {
-        return this.resumeAsync({ runId, step: p.step, resumeData: p.resumeData, runtimeContext: p.runtimeContext });
+        return this.resumeAsync({
+          runId,
+          step: p.step,
+          resumeData: p.resumeData,
+          runtimeContext: p.runtimeContext,
+          tracingOptions: p.tracingOptions
+        });
       },
       resumeStreamVNext: async (p) => {
         return this.resumeStreamVNext({
@@ -1932,7 +1762,7 @@ var Workflow = class extends BaseResource {
     const runtimeContext = parseClientRuntimeContext(params.runtimeContext);
     return this.request(`/api/workflows/${this.workflowId}/start?runId=${params.runId}`, {
       method: "POST",
-      body: { inputData: params?.inputData, runtimeContext }
+      body: { inputData: params?.inputData, runtimeContext, tracingOptions: params.tracingOptions }
     });
   }
   /**
@@ -1944,6 +1774,7 @@ var Workflow = class extends BaseResource {
     step,
     runId,
     resumeData,
+    tracingOptions,
     ...rest
   }) {
     const runtimeContext = parseClientRuntimeContext(rest.runtimeContext);
@@ -1952,7 +1783,8 @@ var Workflow = class extends BaseResource {
       body: {
         step,
         resumeData,
-        runtimeContext
+        runtimeContext,
+        tracingOptions
       }
     });
   }
@@ -1969,7 +1801,7 @@ var Workflow = class extends BaseResource {
     const runtimeContext = parseClientRuntimeContext(params.runtimeContext);
     return this.request(`/api/workflows/${this.workflowId}/start-async?${searchParams.toString()}`, {
       method: "POST",
-      body: { inputData: params.inputData, runtimeContext }
+      body: { inputData: params.inputData, runtimeContext, tracingOptions: params.tracingOptions }
     });
   }
   /**
@@ -1987,7 +1819,7 @@ var Workflow = class extends BaseResource {
       `/api/workflows/${this.workflowId}/stream?${searchParams.toString()}`,
       {
         method: "POST",
-        body: { inputData: params.inputData, runtimeContext },
+        body: { inputData: params.inputData, runtimeContext, tracingOptions: params.tracingOptions },
         stream: true
       }
     );
@@ -2004,7 +1836,7 @@ var Workflow = class extends BaseResource {
       async transform(chunk, controller) {
         try {
           const decoded = new TextDecoder().decode(chunk);
-          const chunks = decoded.split(RECORD_SEPARATOR2);
+          const chunks = decoded.split(RECORD_SEPARATOR);
           for (const chunk2 of chunks) {
             if (chunk2) {
               const newChunk = failedChunk ? failedChunk + chunk2 : chunk2;
@@ -2051,7 +1883,7 @@ var Workflow = class extends BaseResource {
       async transform(chunk, controller) {
         try {
           const decoded = new TextDecoder().decode(chunk);
-          const chunks = decoded.split(RECORD_SEPARATOR2);
+          const chunks = decoded.split(RECORD_SEPARATOR);
           for (const chunk2 of chunks) {
             if (chunk2) {
               const newChunk = failedChunk ? failedChunk + chunk2 : chunk2;
@@ -2085,7 +1917,12 @@ var Workflow = class extends BaseResource {
       `/api/workflows/${this.workflowId}/streamVNext?${searchParams.toString()}`,
       {
         method: "POST",
-        body: { inputData: params.inputData, runtimeContext, closeOnSuspend: params.closeOnSuspend },
+        body: {
+          inputData: params.inputData,
+          runtimeContext,
+          closeOnSuspend: params.closeOnSuspend,
+          tracingOptions: params.tracingOptions
+        },
         stream: true
       }
     );
@@ -2102,7 +1939,7 @@ var Workflow = class extends BaseResource {
       async transform(chunk, controller) {
         try {
           const decoded = new TextDecoder().decode(chunk);
-          const chunks = decoded.split(RECORD_SEPARATOR2);
+          const chunks = decoded.split(RECORD_SEPARATOR);
           for (const chunk2 of chunks) {
             if (chunk2) {
               const newChunk = failedChunk ? failedChunk + chunk2 : chunk2;
@@ -2133,7 +1970,8 @@ var Workflow = class extends BaseResource {
       body: {
         step: params.step,
         resumeData: params.resumeData,
-        runtimeContext
+        runtimeContext,
+        tracingOptions: params.tracingOptions
       }
     });
   }
@@ -2149,7 +1987,8 @@ var Workflow = class extends BaseResource {
       body: {
         step: params.step,
         resumeData: params.resumeData,
-        runtimeContext
+        runtimeContext,
+        tracingOptions: params.tracingOptions
       }
     });
   }
@@ -2189,7 +2028,7 @@ var Workflow = class extends BaseResource {
       async start(controller) {
         try {
           for await (const record of records) {
-            const json = JSON.stringify(record) + RECORD_SEPARATOR2;
+            const json = JSON.stringify(record) + RECORD_SEPARATOR;
             controller.enqueue(encoder.encode(json));
           }
           controller.close();
@@ -2312,7 +2151,7 @@ var MCPTool = class extends BaseResource {
 };
 
 // src/resources/agent-builder.ts
-var RECORD_SEPARATOR3 = "";
+var RECORD_SEPARATOR2 = "";
 var AgentBuilder = class extends BaseResource {
   constructor(options, actionId) {
     super(options);
@@ -2452,7 +2291,7 @@ var AgentBuilder = class extends BaseResource {
         if (done && !value) continue;
         try {
           const decoded = value ? new TextDecoder().decode(value) : "";
-          const chunks = (buffer + decoded).split(RECORD_SEPARATOR3);
+          const chunks = (buffer + decoded).split(RECORD_SEPARATOR2);
           buffer = chunks.pop() || "";
           for (const chunk of chunks) {
             if (chunk) {
@@ -2509,7 +2348,7 @@ var AgentBuilder = class extends BaseResource {
       async transform(chunk, controller) {
         try {
           const decoded = new TextDecoder().decode(chunk);
-          const chunks = decoded.split(RECORD_SEPARATOR3);
+          const chunks = decoded.split(RECORD_SEPARATOR2);
           for (const chunk2 of chunks) {
             if (chunk2) {
               const newChunk = failedChunk ? failedChunk + chunk2 : chunk2;
@@ -2558,7 +2397,7 @@ var AgentBuilder = class extends BaseResource {
       async transform(chunk, controller) {
         try {
           const decoded = new TextDecoder().decode(chunk);
-          const chunks = decoded.split(RECORD_SEPARATOR3);
+          const chunks = decoded.split(RECORD_SEPARATOR2);
           for (const chunk2 of chunks) {
             if (chunk2) {
               const newChunk = failedChunk ? failedChunk + chunk2 : chunk2;
@@ -2729,6 +2568,25 @@ var Observability = class extends BaseResource {
     const queryString = searchParams.toString();
     return this.request(`/api/observability/traces${queryString ? `?${queryString}` : ""}`);
   }
+  /**
+   * Retrieves scores by trace ID and span ID
+   * @param params - Parameters containing trace ID, span ID, and pagination options
+   * @returns Promise containing scores and pagination info
+   */
+  getScoresBySpan(params) {
+    const { traceId, spanId, page, perPage } = params;
+    const searchParams = new URLSearchParams();
+    if (page !== void 0) {
+      searchParams.set("page", String(page));
+    }
+    if (perPage !== void 0) {
+      searchParams.set("perPage", String(perPage));
+    }
+    const queryString = searchParams.toString();
+    return this.request(
+      `/api/observability/traces/${encodeURIComponent(traceId)}/${encodeURIComponent(spanId)}/scores${queryString ? `?${queryString}` : ""}`
+    );
+  }
   score(params) {
     return this.request(`/api/observability/traces/score`, {
       method: "POST",
@@ -2801,7 +2659,7 @@ var NetworkMemoryThread = class extends BaseResource {
 };
 
 // src/resources/vNextNetwork.ts
-var RECORD_SEPARATOR4 = "";
+var RECORD_SEPARATOR3 = "";
 var VNextNetwork = class extends BaseResource {
   constructor(options, networkId) {
     super(options);
@@ -2854,7 +2712,7 @@ var VNextNetwork = class extends BaseResource {
         if (done && !value) continue;
         try {
           const decoded = value ? new TextDecoder().decode(value) : "";
-          const chunks = (buffer + decoded).split(RECORD_SEPARATOR4);
+          const chunks = (buffer + decoded).split(RECORD_SEPARATOR3);
           buffer = chunks.pop() || "";
           for (const chunk of chunks) {
             if (chunk) {
@@ -3100,21 +2958,6 @@ var MastraClient = class extends BaseResource {
   getTool(toolId) {
     return new Tool(this.options, toolId);
   }
-  /**
-   * Retrieves all available legacy workflows
-   * @returns Promise containing map of legacy workflow IDs to legacy workflow details
-   */
-  getLegacyWorkflows() {
-    return this.request("/api/workflows/legacy");
-  }
-  /**
-   * Gets a legacy workflow instance by ID
-   * @param workflowId - ID of the legacy workflow to retrieve
-   * @returns Legacy Workflow instance
-   */
-  getLegacyWorkflow(workflowId) {
-    return new LegacyWorkflow(this.options, workflowId);
-  }
   /**
    * Retrieves all available workflows
    * @param runtimeContext - Optional runtime context to pass as query parameter
@@ -3414,7 +3257,7 @@ var MastraClient = class extends BaseResource {
    * @returns Promise containing the scorer
    */
   getScorer(scorerId) {
-    return this.request(`/api/scores/scorers/${scorerId}`);
+    return this.request(`/api/scores/scorers/${encodeURIComponent(scorerId)}`);
   }
   getScoresByScorerId(params) {
     const { page, perPage, scorerId, entityId, entityType } = params;
@@ -3432,7 +3275,7 @@ var MastraClient = class extends BaseResource {
       searchParams.set("perPage", String(perPage));
     }
     const queryString = searchParams.toString();
-    return this.request(`/api/scores/scorer/${scorerId}${queryString ? `?${queryString}` : ""}`);
+    return this.request(`/api/scores/scorer/${encodeURIComponent(scorerId)}${queryString ? `?${queryString}` : ""}`);
   }
   /**
    * Retrieves scores by run ID
@@ -3449,7 +3292,7 @@ var MastraClient = class extends BaseResource {
       searchParams.set("perPage", String(perPage));
     }
     const queryString = searchParams.toString();
-    return this.request(`/api/scores/run/${runId}${queryString ? `?${queryString}` : ""}`);
+    return this.request(`/api/scores/run/${encodeURIComponent(runId)}${queryString ? `?${queryString}` : ""}`);
   }
   /**
    * Retrieves scores by entity ID and type
@@ -3466,7 +3309,9 @@ var MastraClient = class extends BaseResource {
       searchParams.set("perPage", String(perPage));
     }
     const queryString = searchParams.toString();
-    return this.request(`/api/scores/entity/${entityType}/${entityId}${queryString ? `?${queryString}` : ""}`);
+    return this.request(
+      `/api/scores/entity/${encodeURIComponent(entityType)}/${encodeURIComponent(entityId)}${queryString ? `?${queryString}` : ""}`
+    );
   }
   /**
    * Saves a score
@@ -3492,6 +3337,9 @@ var MastraClient = class extends BaseResource {
   getAITraces(params) {
     return this.observability.getTraces(params);
   }
+  getScoresBySpan(params) {
+    return this.observability.getScoresBySpan(params);
+  }
   score(params) {
     return this.observability.score(params);
   }