@mastra/client-js 0.14.0-alpha.0 → 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) => {
@@ -1709,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);
@@ -1721,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({
@@ -1745,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 }
     });
   }
   /**
@@ -1757,6 +1774,7 @@ var Workflow = class extends BaseResource {
     step,
     runId,
     resumeData,
+    tracingOptions,
     ...rest
   }) {
     const runtimeContext = parseClientRuntimeContext(rest.runtimeContext);
@@ -1765,7 +1783,8 @@ var Workflow = class extends BaseResource {
       body: {
         step,
         resumeData,
-        runtimeContext
+        runtimeContext,
+        tracingOptions
       }
     });
   }
@@ -1782,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 }
     });
   }
   /**
@@ -1800,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
       }
     );
@@ -1898,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
       }
     );
@@ -1946,7 +1970,8 @@ var Workflow = class extends BaseResource {
       body: {
         step: params.step,
         resumeData: params.resumeData,
-        runtimeContext
+        runtimeContext,
+        tracingOptions: params.tracingOptions
       }
     });
   }
@@ -1962,7 +1987,8 @@ var Workflow = class extends BaseResource {
       body: {
         step: params.step,
         resumeData: params.resumeData,
-        runtimeContext
+        runtimeContext,
+        tracingOptions: params.tracingOptions
       }
     });
   }
@@ -2542,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",
@@ -3212,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;
@@ -3230,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
@@ -3247,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
@@ -3264,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
@@ -3290,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);
   }