@mastra/client-js 0.10.23 → 0.11.0-alpha.1

package/dist/index.js

@@ -1,11 +1,11 @@
 import { AbstractAgent, EventType } from '@ag-ui/client';
 import { Observable } from 'rxjs';
 import { processDataStream, parsePartialJson } from '@ai-sdk/ui-utils';
-import { ZodSchema } from 'zod';
-import originalZodToJsonSchema from 'zod-to-json-schema';
-import { isVercelTool } from '@mastra/core/tools/is-vercel-tool';
 import { v4 } from '@lukeed/uuid';
 import { RuntimeContext } from '@mastra/core/runtime-context';
+import { isVercelTool } from '@mastra/core/tools/is-vercel-tool';
+import { z } from 'zod';
+import originalZodToJsonSchema from 'zod-to-json-schema';
 
 // src/adapters/agui.ts
 var AGUIAdapter = class extends AbstractAgent {
@@ -206,12 +206,29 @@ function convertMessagesToMastraMessages(messages) {
   }
   return result;
 }
+function parseClientRuntimeContext(runtimeContext) {
+  if (runtimeContext) {
+    if (runtimeContext instanceof RuntimeContext) {
+      return Object.fromEntries(runtimeContext.entries());
+    }
+    return runtimeContext;
+  }
+  return void 0;
+}
+function isZodType(value) {
+  return typeof value === "object" && value !== null && "_def" in value && "parse" in value && typeof value.parse === "function" && "safeParse" in value && typeof value.safeParse === "function";
+}
 function zodToJsonSchema(zodSchema) {
-  if (!(zodSchema instanceof ZodSchema)) {
+  if (!isZodType(zodSchema)) {
     return zodSchema;
   }
+  if ("toJSONSchema" in z) {
+    return z.toJSONSchema(zodSchema);
+  }
   return originalZodToJsonSchema(zodSchema, { $refStrategy: "none" });
 }
+
+// src/utils/process-client-tools.ts
 function processClientTools(clientTools) {
   if (!clientTools) {
     return void 0;
@@ -240,6 +257,42 @@ function processClientTools(clientTools) {
   );
 }
 
+// src/utils/process-mastra-stream.ts
+async function processMastraStream({
+  stream,
+  onChunk
+}) {
+  const reader = stream.getReader();
+  const decoder = new TextDecoder();
+  let buffer = "";
+  try {
+    while (true) {
+      const { done, value } = await reader.read();
+      if (done) break;
+      buffer += decoder.decode(value, { stream: true });
+      const lines = buffer.split("\n\n");
+      buffer = lines.pop() || "";
+      for (const line of lines) {
+        if (line.startsWith("data: ")) {
+          const data = line.slice(6);
+          if (data === "[DONE]") {
+            console.log("\u{1F3C1} Stream finished");
+            return;
+          }
+          try {
+            const json = JSON.parse(data);
+            await onChunk(json);
+          } catch (error) {
+            console.error("\u274C JSON parse error:", error, "Data:", data);
+          }
+        }
+      }
+    }
+  } finally {
+    reader.releaseLock();
+  }
+}
+
 // src/resources/base.ts
 var BaseResource = class {
   options;
@@ -254,7 +307,7 @@ var BaseResource = class {
    */
   async request(path, options = {}) {
     let lastError = null;
-    const { baseUrl, retries = 3, backoffMs = 100, maxBackoffMs = 1e3, headers = {} } = this.options;
+    const { baseUrl, retries = 3, backoffMs = 100, maxBackoffMs = 1e3, headers = {}, credentials } = this.options;
     let delay = backoffMs;
     for (let attempt = 0; attempt <= retries; attempt++) {
       try {
@@ -268,6 +321,7 @@ var BaseResource = class {
             // 'x-mastra-client-type': 'js',
           },
           signal: this.options.abortSignal,
+          credentials: options.credentials ?? credentials,
           body: options.body instanceof FormData ? options.body : options.body ? JSON.stringify(options.body) : void 0
         });
         if (!response.ok) {
@@ -300,51 +354,6 @@ var BaseResource = class {
     throw lastError || new Error("Request failed");
   }
 };
-function parseClientRuntimeContext(runtimeContext) {
-  if (runtimeContext) {
-    if (runtimeContext instanceof RuntimeContext) {
-      return Object.fromEntries(runtimeContext.entries());
-    }
-    return runtimeContext;
-  }
-  return void 0;
-}
-
-// src/utils/process-mastra-stream.ts
-async function processMastraStream({
-  stream,
-  onChunk
-}) {
-  const reader = stream.getReader();
-  const decoder = new TextDecoder();
-  let buffer = "";
-  try {
-    while (true) {
-      const { done, value } = await reader.read();
-      if (done) break;
-      buffer += decoder.decode(value, { stream: true });
-      const lines = buffer.split("\n\n");
-      buffer = lines.pop() || "";
-      for (const line of lines) {
-        if (line.startsWith("data: ")) {
-          const data = line.slice(6);
-          if (data === "[DONE]") {
-            console.log("\u{1F3C1} Stream finished");
-            return;
-          }
-          try {
-            const json = JSON.parse(data);
-            await onChunk(json);
-          } catch (error) {
-            console.error("\u274C JSON parse error:", error, "Data:", data);
-          }
-        }
-      }
-    }
-  } finally {
-    reader.releaseLock();
-  }
-}
 
 // src/resources/agent.ts
 async function executeToolCallAndRespond({
@@ -1772,7 +1781,7 @@ var LegacyWorkflow = class extends BaseResource {
 };
 
 // src/resources/tool.ts
-var Tool2 = class extends BaseResource {
+var Tool = class extends BaseResource {
   constructor(options, toolId) {
     super(options);
     this.toolId = toolId;
@@ -2041,7 +2050,7 @@ var Workflow = class extends BaseResource {
                 const parsedChunk = JSON.parse(newChunk);
                 controller.enqueue(parsedChunk);
                 failedChunk = void 0;
-              } catch (error) {
+              } catch {
                 failedChunk = newChunk;
               }
             }
@@ -2225,6 +2234,53 @@ var MCPTool = class extends BaseResource {
   }
 };
 
+// src/resources/observability.ts
+var Observability = class extends BaseResource {
+  constructor(options) {
+    super(options);
+  }
+  /**
+   * Retrieves a specific AI trace by ID
+   * @param traceId - ID of the trace to retrieve
+   * @returns Promise containing the AI trace with all its spans
+   */
+  getTrace(traceId) {
+    return this.request(`/api/observability/traces/${traceId}`);
+  }
+  /**
+   * Retrieves paginated list of AI traces with optional filtering
+   * @param params - Parameters for pagination and filtering
+   * @returns Promise containing paginated traces and pagination info
+   */
+  getTraces(params) {
+    const { pagination, filters } = params;
+    const { page, perPage, dateRange } = pagination || {};
+    const { name, spanType } = filters || {};
+    const searchParams = new URLSearchParams();
+    if (page !== void 0) {
+      searchParams.set("page", String(page));
+    }
+    if (perPage !== void 0) {
+      searchParams.set("perPage", String(perPage));
+    }
+    if (name) {
+      searchParams.set("name", name);
+    }
+    if (spanType !== void 0) {
+      searchParams.set("spanType", String(spanType));
+    }
+    if (dateRange) {
+      const dateRangeStr = JSON.stringify({
+        start: dateRange.start instanceof Date ? dateRange.start.toISOString() : dateRange.start,
+        end: dateRange.end instanceof Date ? dateRange.end.toISOString() : dateRange.end
+      });
+      searchParams.set("dateRange", dateRangeStr);
+    }
+    const queryString = searchParams.toString();
+    return this.request(`/api/observability/traces${queryString ? `?${queryString}` : ""}`);
+  }
+};
+
 // src/resources/network-memory-thread.ts
 var NetworkMemoryThread = class extends BaseResource {
   constructor(options, threadId, networkId) {
@@ -2428,8 +2484,10 @@ var VNextNetwork = class extends BaseResource {
 
 // src/client.ts
 var MastraClient = class extends BaseResource {
+  observability;
   constructor(options) {
     super(options);
+    this.observability = new Observability(options);
   }
   /**
    * Retrieves all available agents
@@ -2558,7 +2616,7 @@ var MastraClient = class extends BaseResource {
    * @returns Tool instance
    */
   getTool(toolId) {
-    return new Tool2(this.options, toolId);
+    return new Tool(this.options, toolId);
   }
   /**
    * Retrieves all available legacy workflows
@@ -2940,6 +2998,12 @@ var MastraClient = class extends BaseResource {
   getModelProviders() {
     return this.request(`/api/model-providers`);
   }
+  getAITrace(traceId) {
+    return this.observability.getTrace(traceId);
+  }
+  getAITraces(params) {
+    return this.observability.getTraces(params);
+  }
 };
 
 export { MastraClient };