@mastra/client-js 1.3.0 → 1.4.0

package/dist/index.js

@@ -1124,7 +1124,8 @@ function parseClientRequestContext(requestContext) {
 }
 function base64RequestContext(requestContext) {
   if (requestContext) {
-    return btoa(JSON.stringify(requestContext));
+    const bytes = new TextEncoder().encode(JSON.stringify(requestContext));
+    return btoa(Array.from(bytes, (byte) => String.fromCharCode(byte)).join(""));
   }
   return void 0;
 }
@@ -1282,7 +1283,7 @@ var BaseResource = class {
         const response = await fetchFn(`${baseUrl.replace(/\/$/, "")}${fullPath}`, {
           ...options,
           headers: {
-            ...options.body && !(options.body instanceof FormData) && (options.method === "POST" || options.method === "PUT" || options.method === "PATCH") ? { "content-type": "application/json" } : {},
+            ...options.body && !(options.body instanceof FormData) && (options.method === "POST" || options.method === "PUT" || options.method === "PATCH" || options.method === "DELETE") ? { "content-type": "application/json" } : {},
             ...headers,
             ...options.headers
             // TODO: Bring this back once we figure out what we/users need to do to make this work with cross-origin requests
@@ -2245,12 +2246,18 @@ var Agent = class extends BaseResource {
     return response;
   }
   async network(messages, params) {
+    const processedParams = {
+      ...params,
+      messages,
+      requestContext: parseClientRequestContext(params.requestContext),
+      structuredOutput: params.structuredOutput ? {
+        ...params.structuredOutput,
+        schema: zodToJsonSchema2(params.structuredOutput.schema)
+      } : void 0
+    };
     const response = await this.request(`/agents/${this.agentId}/network`, {
       method: "POST",
-      body: {
-        messages,
-        ...params
-      },
+      body: processedParams,
       stream: true
     });
     if (!response.body) {
@@ -3312,6 +3319,17 @@ var Workflow = class extends BaseResource {
       method: "DELETE"
     });
   }
+  /**
+   * Retrieves the input/output schema for the workflow
+   * @returns Promise containing parsed inputSchema and outputSchema, or null if not defined
+   */
+  async getSchema() {
+    const details = await this.details();
+    return {
+      inputSchema: details.inputSchema ? JSON.parse(details.inputSchema) : null,
+      outputSchema: details.outputSchema ? JSON.parse(details.outputSchema) : null
+    };
+  }
   /**
    * Creates a new workflow run
    * @param params - Optional object containing the optional runId
@@ -4070,6 +4088,52 @@ var StoredAgent = class extends BaseResource {
   }
 };
 
+// src/resources/stored-mcp-client.ts
+var StoredMCPClient = class extends BaseResource {
+  constructor(options, storedMCPClientId) {
+    super(options);
+    this.storedMCPClientId = storedMCPClientId;
+  }
+  /**
+   * Retrieves details about the stored MCP client
+   * @param requestContext - Optional request context to pass as query parameter
+   * @returns Promise containing stored MCP client details
+   */
+  details(requestContext) {
+    return this.request(
+      `/stored/mcp-clients/${encodeURIComponent(this.storedMCPClientId)}${requestContextQueryString(requestContext)}`
+    );
+  }
+  /**
+   * Updates the stored MCP client with the provided fields
+   * @param params - Fields to update
+   * @param requestContext - Optional request context to pass as query parameter
+   * @returns Promise containing the updated stored MCP client
+   */
+  update(params, requestContext) {
+    return this.request(
+      `/stored/mcp-clients/${encodeURIComponent(this.storedMCPClientId)}${requestContextQueryString(requestContext)}`,
+      {
+        method: "PATCH",
+        body: params
+      }
+    );
+  }
+  /**
+   * Deletes the stored MCP client
+   * @param requestContext - Optional request context to pass as query parameter
+   * @returns Promise containing deletion confirmation
+   */
+  delete(requestContext) {
+    return this.request(
+      `/stored/mcp-clients/${encodeURIComponent(this.storedMCPClientId)}${requestContextQueryString(requestContext)}`,
+      {
+        method: "DELETE"
+      }
+    );
+  }
+};
+
 // src/resources/stored-scorer.ts
 var StoredScorer = class extends BaseResource {
   constructor(options, storedScorerId) {
@@ -4116,6 +4180,55 @@ var StoredScorer = class extends BaseResource {
   }
 };
 
+// src/resources/tool-provider.ts
+var ToolProvider = class extends BaseResource {
+  constructor(options, providerId) {
+    super(options);
+    this.providerId = providerId;
+  }
+  /**
+   * Lists available toolkits from this provider
+   * @returns Promise containing list of toolkits
+   */
+  listToolkits() {
+    return this.request(`/tool-providers/${encodeURIComponent(this.providerId)}/toolkits`);
+  }
+  /**
+   * Lists available tools from this provider, with optional filtering
+   * @param params - Optional filtering and pagination parameters
+   * @returns Promise containing list of tools
+   */
+  listTools(params) {
+    const searchParams = new URLSearchParams();
+    if (params?.toolkit) {
+      searchParams.set("toolkit", params.toolkit);
+    }
+    if (params?.search) {
+      searchParams.set("search", params.search);
+    }
+    if (params?.page !== void 0) {
+      searchParams.set("page", String(params.page));
+    }
+    if (params?.perPage !== void 0) {
+      searchParams.set("perPage", String(params.perPage));
+    }
+    const queryString = searchParams.toString();
+    return this.request(
+      `/tool-providers/${encodeURIComponent(this.providerId)}/tools${queryString ? `?${queryString}` : ""}`
+    );
+  }
+  /**
+   * Gets the input schema for a specific tool
+   * @param toolSlug - The slug of the tool
+   * @returns Promise containing the tool's JSON schema
+   */
+  getToolSchema(toolSlug) {
+    return this.request(
+      `/tool-providers/${encodeURIComponent(this.providerId)}/tools/${encodeURIComponent(toolSlug)}/schema`
+    );
+  }
+};
+
 // src/resources/workspace.ts
 var WorkspaceSkillResource = class extends BaseResource {
   constructor(options, workspaceId, skillName) {
@@ -5048,6 +5161,76 @@ var MastraClient = class extends BaseResource {
     return new StoredScorer(this.options, storedScorerId);
   }
   // ============================================================================
+  // Stored MCP Clients
+  // ============================================================================
+  /**
+   * Lists all stored MCP clients with optional pagination
+   * @param params - Optional pagination and ordering parameters
+   * @returns Promise containing paginated list of stored MCP clients
+   */
+  listStoredMCPClients(params) {
+    const searchParams = new URLSearchParams();
+    if (params?.page !== void 0) {
+      searchParams.set("page", String(params.page));
+    }
+    if (params?.perPage !== void 0) {
+      searchParams.set("perPage", String(params.perPage));
+    }
+    if (params?.orderBy) {
+      if (params.orderBy.field) {
+        searchParams.set("orderBy[field]", params.orderBy.field);
+      }
+      if (params.orderBy.direction) {
+        searchParams.set("orderBy[direction]", params.orderBy.direction);
+      }
+    }
+    if (params?.authorId) {
+      searchParams.set("authorId", params.authorId);
+    }
+    if (params?.metadata) {
+      searchParams.set("metadata", JSON.stringify(params.metadata));
+    }
+    const queryString = searchParams.toString();
+    return this.request(`/stored/mcp-clients${queryString ? `?${queryString}` : ""}`);
+  }
+  /**
+   * Creates a new stored MCP client
+   * @param params - MCP client configuration
+   * @returns Promise containing the created stored MCP client
+   */
+  createStoredMCPClient(params) {
+    return this.request("/stored/mcp-clients", {
+      method: "POST",
+      body: params
+    });
+  }
+  /**
+   * Gets a stored MCP client instance by ID for further operations (details, update, delete)
+   * @param storedMCPClientId - ID of the stored MCP client
+   * @returns StoredMCPClient instance
+   */
+  getStoredMCPClient(storedMCPClientId) {
+    return new StoredMCPClient(this.options, storedMCPClientId);
+  }
+  // ============================================================================
+  // Tool Providers
+  // ============================================================================
+  /**
+   * Lists all registered tool providers
+   * @returns Promise containing list of tool provider info
+   */
+  listToolProviders() {
+    return this.request("/tool-providers");
+  }
+  /**
+   * Gets a tool provider instance by ID for further operations (listToolkits, listTools, getToolSchema)
+   * @param providerId - ID of the tool provider
+   * @returns ToolProvider instance
+   */
+  getToolProvider(providerId) {
+    return new ToolProvider(this.options, providerId);
+  }
+  // ============================================================================
   // System
   // ============================================================================
   /**
@@ -5092,6 +5275,201 @@ var MastraClient = class extends BaseResource {
   listEmbedders() {
     return this.request("/embedders");
   }
+  // ============================================================================
+  // Datasets
+  // ============================================================================
+  /**
+   * Lists all datasets with optional pagination
+   */
+  listDatasets(pagination) {
+    const searchParams = new URLSearchParams();
+    if (pagination?.page !== void 0) searchParams.set("page", String(pagination.page));
+    if (pagination?.perPage !== void 0) searchParams.set("perPage", String(pagination.perPage));
+    const qs = searchParams.toString();
+    return this.request(`/datasets${qs ? `?${qs}` : ""}`);
+  }
+  /**
+   * Gets a single dataset by ID
+   */
+  getDataset(datasetId) {
+    return this.request(`/datasets/${encodeURIComponent(datasetId)}`);
+  }
+  /**
+   * Creates a new dataset
+   */
+  createDataset(params) {
+    return this.request("/datasets", { method: "POST", body: params });
+  }
+  /**
+   * Updates a dataset
+   */
+  updateDataset(params) {
+    const { datasetId, ...body } = params;
+    return this.request(`/datasets/${encodeURIComponent(datasetId)}`, {
+      method: "PATCH",
+      body
+    });
+  }
+  /**
+   * Deletes a dataset
+   */
+  deleteDataset(datasetId) {
+    return this.request(`/datasets/${encodeURIComponent(datasetId)}`, {
+      method: "DELETE"
+    });
+  }
+  // ============================================================================
+  // Dataset Items
+  // ============================================================================
+  /**
+   * Lists items in a dataset with optional pagination, search, and version filter
+   */
+  listDatasetItems(datasetId, params) {
+    const searchParams = new URLSearchParams();
+    if (params?.page !== void 0) searchParams.set("page", String(params.page));
+    if (params?.perPage !== void 0) searchParams.set("perPage", String(params.perPage));
+    if (params?.search) searchParams.set("search", params.search);
+    if (params?.version != null) {
+      searchParams.set("version", String(params.version));
+    }
+    const qs = searchParams.toString();
+    return this.request(`/datasets/${encodeURIComponent(datasetId)}/items${qs ? `?${qs}` : ""}`);
+  }
+  /**
+   * Gets a single dataset item by ID
+   */
+  getDatasetItem(datasetId, itemId) {
+    return this.request(`/datasets/${encodeURIComponent(datasetId)}/items/${encodeURIComponent(itemId)}`);
+  }
+  /**
+   * Adds an item to a dataset
+   */
+  addDatasetItem(params) {
+    const { datasetId, ...body } = params;
+    return this.request(`/datasets/${encodeURIComponent(datasetId)}/items`, {
+      method: "POST",
+      body
+    });
+  }
+  /**
+   * Updates a dataset item
+   */
+  updateDatasetItem(params) {
+    const { datasetId, itemId, ...body } = params;
+    return this.request(`/datasets/${encodeURIComponent(datasetId)}/items/${encodeURIComponent(itemId)}`, {
+      method: "PATCH",
+      body
+    });
+  }
+  /**
+   * Deletes a dataset item
+   */
+  deleteDatasetItem(datasetId, itemId) {
+    return this.request(`/datasets/${encodeURIComponent(datasetId)}/items/${encodeURIComponent(itemId)}`, {
+      method: "DELETE"
+    });
+  }
+  /**
+   * Batch inserts items to a dataset
+   */
+  batchInsertDatasetItems(params) {
+    const { datasetId, ...body } = params;
+    return this.request(`/datasets/${encodeURIComponent(datasetId)}/items/batch`, {
+      method: "POST",
+      body
+    });
+  }
+  /**
+   * Batch deletes items from a dataset
+   */
+  batchDeleteDatasetItems(params) {
+    const { datasetId, ...body } = params;
+    return this.request(`/datasets/${encodeURIComponent(datasetId)}/items/batch`, {
+      method: "DELETE",
+      body
+    });
+  }
+  // ============================================================================
+  // Dataset Item Versions
+  // ============================================================================
+  /**
+   * Lists versions for a dataset item
+   */
+  getItemHistory(datasetId, itemId) {
+    return this.request(`/datasets/${encodeURIComponent(datasetId)}/items/${encodeURIComponent(itemId)}/history`);
+  }
+  /**
+   * Gets a specific version of a dataset item
+   */
+  getDatasetItemVersion(datasetId, itemId, datasetVersion) {
+    return this.request(
+      `/datasets/${encodeURIComponent(datasetId)}/items/${encodeURIComponent(itemId)}/versions/${datasetVersion}`
+    );
+  }
+  // ============================================================================
+  // Dataset Versions
+  // ============================================================================
+  /**
+   * Lists versions for a dataset
+   */
+  listDatasetVersions(datasetId, pagination) {
+    const searchParams = new URLSearchParams();
+    if (pagination?.page !== void 0) searchParams.set("page", String(pagination.page));
+    if (pagination?.perPage !== void 0) searchParams.set("perPage", String(pagination.perPage));
+    const qs = searchParams.toString();
+    return this.request(`/datasets/${encodeURIComponent(datasetId)}/versions${qs ? `?${qs}` : ""}`);
+  }
+  // ============================================================================
+  // Dataset Experiments
+  // ============================================================================
+  /**
+   * Lists experiments for a dataset
+   */
+  listDatasetExperiments(datasetId, pagination) {
+    const searchParams = new URLSearchParams();
+    if (pagination?.page !== void 0) searchParams.set("page", String(pagination.page));
+    if (pagination?.perPage !== void 0) searchParams.set("perPage", String(pagination.perPage));
+    const qs = searchParams.toString();
+    return this.request(`/datasets/${encodeURIComponent(datasetId)}/experiments${qs ? `?${qs}` : ""}`);
+  }
+  /**
+   * Gets a single dataset experiment by ID
+   */
+  getDatasetExperiment(datasetId, experimentId) {
+    return this.request(`/datasets/${encodeURIComponent(datasetId)}/experiments/${encodeURIComponent(experimentId)}`);
+  }
+  /**
+   * Lists results for a dataset experiment
+   */
+  listDatasetExperimentResults(datasetId, experimentId, pagination) {
+    const searchParams = new URLSearchParams();
+    if (pagination?.page !== void 0) searchParams.set("page", String(pagination.page));
+    if (pagination?.perPage !== void 0) searchParams.set("perPage", String(pagination.perPage));
+    const qs = searchParams.toString();
+    return this.request(
+      `/datasets/${encodeURIComponent(datasetId)}/experiments/${encodeURIComponent(experimentId)}/results${qs ? `?${qs}` : ""}`
+    );
+  }
+  /**
+   * Triggers a new dataset experiment
+   */
+  triggerDatasetExperiment(params) {
+    const { datasetId, ...body } = params;
+    return this.request(`/datasets/${encodeURIComponent(datasetId)}/experiments`, {
+      method: "POST",
+      body
+    });
+  }
+  /**
+   * Compares two dataset experiments for regression detection
+   */
+  compareExperiments(params) {
+    const { datasetId, ...body } = params;
+    return this.request(`/datasets/${encodeURIComponent(datasetId)}/compare`, {
+      method: "POST",
+      body
+    });
+  }
 };
 
 // src/tools.ts