@mastra/client-js 1.46.0-alpha.3 → 1.46.0-alpha.5

package/dist/index.js

@@ -670,7 +670,10 @@ async function executeToolCallAndRespond({ response, params, agentId, resourceId
 					content: [toolResultContent]
 				}];
 				const updatedMessages = threadId ? newMessages : [...Array.isArray(params.messages) ? params.messages : [], ...newMessages];
-				const respondOptions = { ...params };
+				const respondOptions = {
+					...params,
+					clientTools: params.clientToolsResolver?.() ?? params.clientTools
+				};
 				delete respondOptions.messages;
 				return respondFn(updatedMessages, respondOptions);
 			}
@@ -749,11 +752,13 @@ var AgentVoice = class extends BaseResource {
 var Agent = class extends BaseResource {
 	agentId;
 	version;
+	routeOverrides;
 	voice;
-	constructor(options, agentId, version) {
+	constructor(options, agentId, version, routeOverrides) {
 		super(options);
 		this.agentId = agentId;
 		this.version = version;
+		this.routeOverrides = routeOverrides;
 		this.voice = new AgentVoice(options, this.agentId, this.version);
 	}
 	getQueryString(requestContext, delimiter = "?") {
@@ -821,7 +826,8 @@ var Agent = class extends BaseResource {
 					streamOptions: {
 						...streamOptions,
 						requestContext: parseClientRequestContext(streamOptions.requestContext),
-						clientTools: processClientTools(streamOptions.clientTools)
+						clientTools: processClientTools(streamOptions.clientToolsResolver?.() ?? streamOptions.clientTools),
+						clientToolsResolver: void 0
 					}
 				}
 			},
@@ -998,7 +1004,11 @@ var Agent = class extends BaseResource {
 						resourceId,
 						threadId
 					});
-					const activeClientTools = activeRuntimeOptions?.clientTools;
+					if (!activeRuntimeOptions) {
+						agent.deleteSignalRuntimeOptions(runId);
+						return;
+					}
+					const activeClientTools = activeRuntimeOptions.clientToolsResolver?.() ?? activeRuntimeOptions.clientTools;
 					if (!activeClientTools) {
 						agent.deleteSignalRuntimeOptions(runId);
 						return;
@@ -1291,7 +1301,7 @@ var Agent = class extends BaseResource {
 			output: params.output ? zodToJsonSchema$1(params.output) : void 0,
 			experimental_output: params.experimental_output ? zodToJsonSchema$1(params.experimental_output) : void 0,
 			requestContext: parseClientRequestContext(params.requestContext),
-			clientTools: processClientTools(params.clientTools)
+			clientTools: processClientTools(params.clientToolsResolver?.() ?? params.clientTools)
 		};
 		const { resourceId, threadId, requestContext } = processedParams;
 		const response = await this.request(`/agents/${this.agentId}/generate-legacy`, {
@@ -1302,7 +1312,7 @@ var Agent = class extends BaseResource {
 			const toolCalls = response.toolCalls;
 			if (!toolCalls || !Array.isArray(toolCalls)) return response;
 			for (const toolCall of toolCalls) {
-				const clientTool = params.clientTools?.[toolCall.toolName];
+				const clientTool = processedParams.clientTools?.[toolCall.toolName];
 				if (clientTool && clientTool.execute) {
 					const { result, observability } = await executeClientToolWithObservability({
 						clientTool,
@@ -1346,10 +1356,11 @@ var Agent = class extends BaseResource {
 			...options,
 			messages
 		};
+		const resolvedClientTools = params.clientToolsResolver?.() ?? params.clientTools;
 		const processedParams = {
 			...params,
 			requestContext: parseClientRequestContext(params.requestContext),
-			clientTools: processClientTools(params.clientTools),
+			clientTools: processClientTools(resolvedClientTools),
 			structuredOutput: params.structuredOutput ? {
 				...params.structuredOutput,
 				schema: standardSchemaToJSONSchema(toStandardSchema(params.structuredOutput.schema))
@@ -1365,7 +1376,10 @@ var Agent = class extends BaseResource {
 		});
 		if (response.finishReason === "tool-calls") return executeToolCallAndRespond({
 			response,
-			params,
+			params: {
+				...params,
+				clientTools: resolvedClientTools
+			},
 			agentId: this.agentId,
 			resourceId,
 			threadId,
@@ -1532,12 +1546,18 @@ var Agent = class extends BaseResource {
 				execUpdate();
 			},
 			async onToolCallPart(value) {
+				const partialToolCall = partialToolCalls[value.toolCallId];
+				const streamedArgs = partialToolCall?.text ? parsePartialJson(partialToolCall.text).value : void 0;
+				const toolCall = streamedArgs === void 0 ? value : {
+					...value,
+					args: streamedArgs
+				};
 				const invocation = {
 					state: "call",
 					step,
-					...value
+					...toolCall
 				};
-				if (partialToolCalls[value.toolCallId] != null) message.toolInvocations[partialToolCalls[value.toolCallId].index] = invocation;
+				if (partialToolCall != null) message.toolInvocations[partialToolCalls[value.toolCallId].index] = invocation;
 				else {
 					if (message.toolInvocations == null) message.toolInvocations = [];
 					message.toolInvocations.push(invocation);
@@ -1545,12 +1565,12 @@ var Agent = class extends BaseResource {
 				updateToolInvocationPart(value.toolCallId, invocation, value.providerMetadata);
 				execUpdate();
 				if (onToolCall) {
-					const result = await onToolCall({ toolCall: value });
+					const result = await onToolCall({ toolCall });
 					if (result != null) {
 						const invocation = {
 							state: "result",
 							step,
-							...value,
+							...toolCall,
 							result
 						};
 						message.toolInvocations[message.toolInvocations.length - 1] = invocation;
@@ -1618,7 +1638,7 @@ var Agent = class extends BaseResource {
 			output: params.output ? zodToJsonSchema$1(params.output) : void 0,
 			experimental_output: params.experimental_output ? zodToJsonSchema$1(params.experimental_output) : void 0,
 			requestContext: parseClientRequestContext(params.requestContext),
-			clientTools: processClientTools(params.clientTools)
+			clientTools: processClientTools(params.clientToolsResolver?.() ?? params.clientTools)
 		};
 		const { readable, writable } = new TransformStream();
 		const response = await this.processStreamResponseLegacy(processedParams, writable);
@@ -1751,12 +1771,18 @@ var Agent = class extends BaseResource {
 						execUpdate();
 						break;
 					case "tool-call": {
+						const partialToolCall = partialToolCalls[chunk.payload.toolCallId];
+						const streamedArgs = partialToolCall?.text ? parsePartialJson(partialToolCall.text).value : void 0;
+						const toolCall = streamedArgs === void 0 ? chunk.payload : {
+							...chunk.payload,
+							args: streamedArgs
+						};
 						const invocation = {
 							state: "call",
 							step,
-							...chunk.payload
+							...toolCall
 						};
-						if (partialToolCalls[chunk.payload.toolCallId] != null) message.toolInvocations[partialToolCalls[chunk.payload.toolCallId].index] = invocation;
+						if (partialToolCall != null) message.toolInvocations[partialToolCalls[chunk.payload.toolCallId].index] = invocation;
 						else {
 							if (message.toolInvocations == null) message.toolInvocations = [];
 							message.toolInvocations.push(invocation);
@@ -1764,12 +1790,12 @@ var Agent = class extends BaseResource {
 						updateToolInvocationPart(chunk.payload.toolCallId, invocation, chunk.payload.providerMetadata);
 						execUpdate();
 						if (onToolCall) {
-							const result = await onToolCall({ toolCall: chunk.payload });
+							const result = await onToolCall({ toolCall });
 							if (result != null) {
 								const invocation = {
 									state: "result",
 									step,
-									...chunk.payload,
+									...toolCall,
 									result
 								};
 								message.toolInvocations[message.toolInvocations.length - 1] = invocation;
@@ -1872,7 +1898,8 @@ var Agent = class extends BaseResource {
 			const { messages: _messages, ...resumeStreamBody } = processedParams;
 			requestBody = resumeStreamBody;
 		}
-		const response = await this.request(`/agents/${this.agentId}/${route}`, {
+		const requestPath = route === "stream" && this.routeOverrides?.stream ? this.routeOverrides.stream : `/agents/${this.agentId}/${route}`;
+		const response = await this.request(requestPath, {
 			method: "POST",
 			body: requestBody,
 			stream: true
@@ -2042,7 +2069,8 @@ var Agent = class extends BaseResource {
 							try {
 								await this.processStreamResponse({
 									...processedParams,
-									messages: updatedMessages
+									messages: updatedMessages,
+									clientTools: processClientTools(processedParams.clientToolsResolver?.() ?? processedParams.clientTools)
 								}, controller, recursionRoute);
 							} catch (error) {
 								console.error("Error processing recursive stream response:", error);
@@ -2162,7 +2190,7 @@ var Agent = class extends BaseResource {
 		const processedParams = {
 			...params,
 			requestContext: parseClientRequestContext(params.requestContext),
-			clientTools: processClientTools(params.clientTools),
+			clientTools: processClientTools(params.clientToolsResolver?.() ?? params.clientTools),
 			structuredOutput
 		};
 		let readableController;
@@ -2196,7 +2224,7 @@ var Agent = class extends BaseResource {
 		const processedParams = {
 			...params,
 			requestContext: parseClientRequestContext(params.requestContext),
-			clientTools: processClientTools(params.clientTools),
+			clientTools: processClientTools(params.clientToolsResolver?.() ?? params.clientTools),
 			structuredOutput
 		};
 		let readableController;
@@ -2345,7 +2373,7 @@ var Agent = class extends BaseResource {
 			...options,
 			resumeData,
 			requestContext: parseClientRequestContext(options.requestContext),
-			clientTools: processClientTools(options.clientTools),
+			clientTools: processClientTools(options.clientToolsResolver?.() ?? options.clientTools),
 			structuredOutput: options.structuredOutput ? {
 				...options.structuredOutput,
 				schema: standardSchemaToJSONSchema(toStandardSchema(options.structuredOutput.schema))
@@ -2408,7 +2436,7 @@ var Agent = class extends BaseResource {
 			...options,
 			resumeData,
 			requestContext: parseClientRequestContext(options.requestContext),
-			clientTools: processClientTools(options.clientTools),
+			clientTools: processClientTools(options.clientToolsResolver?.() ?? options.clientTools),
 			structuredOutput: options.structuredOutput ? {
 				...options.structuredOutput,
 				schema: standardSchemaToJSONSchema(toStandardSchema(options.structuredOutput.schema))
@@ -2468,7 +2496,7 @@ var Agent = class extends BaseResource {
 		const { resource, thread } = memory ?? {};
 		const threadId = processedParams.threadId ?? (typeof thread === "string" ? thread : thread?.id);
 		const resourceId = processedParams.resourceId ?? resource;
-		const response = await this.request(`/agents/${this.agentId}/stream-legacy`, {
+		const response = await this.request(this.routeOverrides?.stream ?? `/agents/${this.agentId}/stream-legacy`, {
 			method: "POST",
 			body: processedParams,
 			stream: true
@@ -2547,6 +2575,7 @@ var Agent = class extends BaseResource {
 								];
 								this.processStreamResponseLegacy({
 									...processedParams,
+									clientTools: processClientTools(processedParams.clientToolsResolver?.() ?? processedParams.clientTools),
 									messages: updatedMessages
 								}, writable).catch((error) => {
 									console.error("Error processing stream response:", error);
@@ -4325,7 +4354,7 @@ var Observability = class extends BaseResource {
 	* Queries completed logical traces using recursive trace and related-record predicates.
 	*
 	* @param params - Advanced trace query, including its required time range
-	* @returns Matching lightweight traces or distinct thread groups
+	* @returns Matching lightweight traces
 	*/
 	queryTraces(params) {
 		return this.request("/observability/traces/query", {
@@ -6388,8 +6417,8 @@ var MastraClient = class extends BaseResource {
 	* @param version - Optional version selector for stored agent overrides
 	* @returns Agent instance
 	*/
-	getAgent(agentId, version) {
-		return new Agent(this.options, agentId, version);
+	getAgent(agentId, version, routeOverrides) {
+		return new Agent(this.options, agentId, version, routeOverrides);
 	}
 	/**
 	* Lists the agent controllers hosted on the connected Mastra instance.
@@ -7132,6 +7161,12 @@ var MastraClient = class extends BaseResource {
 		return new DynamicWorkflow(this.options, dynamicWorkflowId);
 	}
 	/**
+	* Retrieves workflow builder settings for UI gating.
+	*/
+	getWorkflowBuilderSettings() {
+		return this.request("/editor/workflow-builder/settings");
+	}
+	/**
 	* Lists all stored prompt blocks with optional pagination
 	* @param params - Optional pagination and ordering parameters
 	* @returns Promise containing paginated list of stored prompt blocks
@@ -7461,10 +7496,12 @@ var MastraClient = class extends BaseResource {
 	/**
 	* Lists all datasets with optional pagination
 	*/
-	listDatasets(pagination) {
+	listDatasets(params) {
 		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));
+		if (params?.page !== void 0) searchParams.set("page", String(params.page));
+		if (params?.perPage !== void 0) searchParams.set("perPage", String(params.perPage));
+		if (params?.targetType !== void 0) searchParams.set("targetType", params.targetType);
+		for (const id of params?.targetIds ?? []) searchParams.append("targetIds", id);
 		const qs = searchParams.toString();
 		return this.request(`/datasets${qs ? `?${qs}` : ""}`);
 	}
@@ -7634,6 +7671,8 @@ var MastraClient = class extends BaseResource {
 		if (params?.comparisonId !== void 0) searchParams.set("comparisonId", params.comparisonId);
 		if (params?.variantId !== void 0) searchParams.set("variantId", params.variantId);
 		if (params?.trialIndex !== void 0) searchParams.set("trialIndex", String(params.trialIndex));
+		if (params?.targetType !== void 0) searchParams.set("targetType", params.targetType);
+		if (params?.targetId !== void 0) searchParams.set("targetId", params.targetId);
 		const qs = searchParams.toString();
 		return this.request(`/experiments${qs ? `?${qs}` : ""}`);
 	}
@@ -7654,6 +7693,8 @@ var MastraClient = class extends BaseResource {
 		if (params?.comparisonId !== void 0) searchParams.set("comparisonId", params.comparisonId);
 		if (params?.variantId !== void 0) searchParams.set("variantId", params.variantId);
 		if (params?.trialIndex !== void 0) searchParams.set("trialIndex", String(params.trialIndex));
+		if (params?.targetType !== void 0) searchParams.set("targetType", params.targetType);
+		if (params?.targetId !== void 0) searchParams.set("targetId", params.targetId);
 		const qs = searchParams.toString();
 		return this.request(`/datasets/${encodeURIComponent(datasetId)}/experiments${qs ? `?${qs}` : ""}`);
 	}