@mastra/client-js 1.56.0-alpha.9 → 1.56.0

package/dist/index.js

@@ -59,15 +59,16 @@ function attributesToOtlp(attrs) {
 			key,
 			value: { boolValue: raw }
 		});
-		else if (typeof raw === "number") if (Number.isInteger(raw)) out.push({
-			key,
-			value: { intValue: raw }
-		});
-		else out.push({
-			key,
-			value: { doubleValue: raw }
-		});
-		else try {
+		else if (typeof raw === "number") {
+			if (Number.isInteger(raw)) out.push({
+				key,
+				value: { intValue: raw }
+			});
+			else out.push({
+				key,
+				value: { doubleValue: raw }
+			});
+		} else try {
 			out.push({
 				key,
 				value: { stringValue: JSON.stringify(raw) }
@@ -662,7 +663,6 @@ function fixJson(input) {
 				stack.pop();
 				stack.push(swapState);
 				stack.push("INSIDE_ARRAY_START");
-				break;
 		}
 	}
 	function processAfterObjectValue(char, i) {
@@ -674,7 +674,6 @@ function fixJson(input) {
 			case "}":
 				lastValidIndex = i;
 				stack.pop();
-				break;
 		}
 	}
 	function processAfterArrayValue(char, i) {
@@ -686,7 +685,6 @@ function fixJson(input) {
 			case "]":
 				lastValidIndex = i;
 				stack.pop();
-				break;
 		}
 	}
 	for (let i = 0; i < input.length; i++) {
@@ -704,7 +702,6 @@ function fixJson(input) {
 					case "}":
 						lastValidIndex = i;
 						stack.pop();
-						break;
 				}
 				break;
 			case "INSIDE_OBJECT_AFTER_COMMA":
@@ -712,7 +709,6 @@ function fixJson(input) {
 					case "\"":
 						stack.pop();
 						stack.push("INSIDE_OBJECT_KEY");
-						break;
 				}
 				break;
 			case "INSIDE_OBJECT_KEY":
@@ -720,7 +716,6 @@ function fixJson(input) {
 					case "\"":
 						stack.pop();
 						stack.push("INSIDE_OBJECT_AFTER_KEY");
-						break;
 				}
 				break;
 			case "INSIDE_OBJECT_AFTER_KEY":
@@ -728,7 +723,6 @@ function fixJson(input) {
 					case ":":
 						stack.pop();
 						stack.push("INSIDE_OBJECT_BEFORE_VALUE");
-						break;
 				}
 				break;
 			case "INSIDE_OBJECT_BEFORE_VALUE":
@@ -758,7 +752,6 @@ function fixJson(input) {
 					default:
 						lastValidIndex = i;
 						processValueStart(char, i, "INSIDE_ARRAY_AFTER_VALUE");
-						break;
 				}
 				break;
 			case "INSIDE_ARRAY_AFTER_VALUE":
@@ -771,9 +764,7 @@ function fixJson(input) {
 						lastValidIndex = i;
 						stack.pop();
 						break;
-					default:
-						lastValidIndex = i;
-						break;
+					default: lastValidIndex = i;
 				}
 				break;
 			case "INSIDE_ARRAY_AFTER_COMMA":
@@ -814,9 +805,7 @@ function fixJson(input) {
 						stack.pop();
 						if (stack[stack.length - 1] === "INSIDE_ARRAY_AFTER_VALUE") processAfterArrayValue(char, i);
 						break;
-					default:
-						stack.pop();
-						break;
+					default: stack.pop();
 				}
 				break;
 			case "INSIDE_LITERAL": {
@@ -1225,8 +1214,16 @@ var BaseResource = class {
 };
 //#endregion
 //#region src/resources/agent.ts
-const SIGNAL_RUNTIME_OPTIONS_TTL_MS = 300 * 1e3;
+const SIGNAL_RUNTIME_OPTIONS_TTL_MS = 3e5;
 const signalRuntimeOptionsByRunId = /* @__PURE__ */ new Map();
+/**
+* Client tools a resume request can execute, keyed by run. The stream that stayed open from
+* before the suspension must not run those tools again, but it must run any resumed client tool
+* the resume request cannot execute (a server-side approve sends no client tools). Held until the
+* last stream of the run closes, which is after the resume request returns.
+*/
+const resumingClientToolNamesByRunKey = /* @__PURE__ */ new Map();
+const openStreamCountByRunKey = /* @__PURE__ */ new Map();
 const latestSignalRuntimeOptionsByThread = /* @__PURE__ */ new Map();
 const createSignalRuntimeOptionsEntry = (store, key, streamOptions) => {
 	const timeout = setTimeout(() => store.delete(key), SIGNAL_RUNTIME_OPTIONS_TTL_MS);
@@ -1419,6 +1416,49 @@ var Agent = class extends BaseResource {
 	getSignalRuntimeRunKey(runId) {
 		return `${this.options.baseUrl}|${this.apiPrefix}|${this.agentId}|${runId}`;
 	}
+	/**
+	* Records that `runId` is being resumed by a request that can execute these client tools, so the
+	* stream that is still open from before the suspension skips them. Released when the last stream
+	* of the run closes, which is after this request returns.
+	*/
+	holdResumingClientTools(runId, clientTools) {
+		const names = Object.entries(clientTools ?? {}).filter(([, tool]) => typeof tool.execute === "function").map(([name]) => name);
+		if (names.length === 0) return () => {};
+		const key = this.getSignalRuntimeRunKey(runId);
+		const held = resumingClientToolNamesByRunKey.get(key) ?? /* @__PURE__ */ new Map();
+		for (const name of names) held.set(name, (held.get(name) ?? 0) + 1);
+		resumingClientToolNamesByRunKey.set(key, held);
+		let released = false;
+		return ({ force = false } = {}) => {
+			if (released || !force && openStreamCountByRunKey.has(key)) return;
+			released = true;
+			const current = resumingClientToolNamesByRunKey.get(key);
+			if (!current) return;
+			for (const name of names) {
+				const count = (current.get(name) ?? 1) - 1;
+				if (count > 0) current.set(name, count);
+				else current.delete(name);
+			}
+			if (current.size === 0) resumingClientToolNamesByRunKey.delete(key);
+		};
+	}
+	/** Drops the held tool names when the last stream of the run closes. */
+	trackOpenStream(runId) {
+		const key = this.getSignalRuntimeRunKey(runId);
+		openStreamCountByRunKey.set(key, (openStreamCountByRunKey.get(key) ?? 0) + 1);
+		let released = false;
+		return () => {
+			if (released) return;
+			released = true;
+			const remaining = (openStreamCountByRunKey.get(key) ?? 1) - 1;
+			if (remaining > 0) {
+				openStreamCountByRunKey.set(key, remaining);
+				return;
+			}
+			openStreamCountByRunKey.delete(key);
+			resumingClientToolNamesByRunKey.delete(key);
+		};
+	}
 	getSignalRuntimeThreadKey({ resourceId, threadId }) {
 		if (!threadId) return void 0;
 		return `${this.options.baseUrl}|${this.apiPrefix}|${this.agentId}|${resourceId ?? ""}|${threadId}`;
@@ -2556,7 +2596,6 @@ var Agent = class extends BaseResource {
 					case "finish":
 						finishReason = chunk.payload?.stepResult?.reason ?? finishReason;
 						if (chunk.payload?.usage != null) usage = chunk.payload.usage;
-						break;
 				}
 			}
 		});
@@ -2621,6 +2660,9 @@ var Agent = class extends BaseResource {
 			let messages = [];
 			let streamRunId = processedParams.runId;
 			let finishedHistory;
+			let suspended = false;
+			const resumedToolCallIds = /* @__PURE__ */ new Set();
+			const releaseOpenStream = streamRunId ? this.trackOpenStream(streamRunId) : () => {};
 			const [streamForController, streamForProcessing] = response.body.tee();
 			const decoder = new TextDecoder();
 			let pendingText = "";
@@ -2689,6 +2731,9 @@ var Agent = class extends BaseResource {
 							});
 						}
 						const executableToolCalls = [...toolInvocationsById.values()].filter((toolCall) => {
+							if (resumedToolCallIds.has(toolCall.toolCallId)) {
+								if ((streamRunId ? resumingClientToolNamesByRunKey.get(this.getSignalRuntimeRunKey(streamRunId)) : void 0)?.has(toolCall.toolName)) return false;
+							}
 							const clientTool = processedParams.clientTools?.[toolCall.toolName];
 							return Boolean(clientTool?.execute);
 						});
@@ -2815,6 +2860,12 @@ var Agent = class extends BaseResource {
 				},
 				onStreamChunk: (chunk) => {
 					if (!streamRunId && typeof chunk.runId === "string") streamRunId = chunk.runId;
+					if (chunk.type === "tool-call-approval" || chunk.type === "tool-call-suspended") suspended = true;
+					else if (suspended && chunk.type === "tool-call" && typeof chunk.payload?.toolCallId === "string") {
+						const toolName = chunk.payload.toolName;
+						const held = streamRunId ? resumingClientToolNamesByRunKey.get(this.getSignalRuntimeRunKey(streamRunId)) : void 0;
+						if (typeof toolName === "string" && held?.has(toolName)) resumedToolCallIds.add(chunk.payload.toolCallId);
+					}
 				},
 				lastMessage: void 0
 			}).catch(async (error) => {
@@ -2825,6 +2876,7 @@ var Agent = class extends BaseResource {
 				safeClose();
 			}).finally(() => {
 				signal?.removeEventListener("abort", onAbort);
+				releaseOpenStream();
 			});
 		} catch (error) {
 			if (!isAbortError(error)) console.error("Error processing stream response:", error);
@@ -2992,13 +3044,18 @@ var Agent = class extends BaseResource {
 	}
 	async approveToolCall(params) {
 		const { requestContext, ...rest } = params;
+		const executableClientTools = params.clientToolsResolver?.() ?? params.clientTools;
+		const releaseHold = this.holdResumingClientTools(rest.runId, executableClientTools);
 		const processedParams = {
 			...rest,
 			requestContext: parseClientRequestContext(requestContext),
-			clientTools: processClientTools(params.clientToolsResolver?.() ?? params.clientTools)
+			clientTools: processClientTools(executableClientTools)
 		};
 		const { readable, controller: readableController, signal } = this.createCancellableStream(params.abortSignal);
-		const response = await this.processStreamResponse(processedParams, readableController, "approve-tool-call", signal);
+		const response = await this.processStreamResponse(processedParams, readableController, "approve-tool-call", signal).catch((error) => {
+			releaseHold({ force: true });
+			throw error;
+		});
 		const streamResponse = new Response(readable, {
 			status: response.status,
 			statusText: response.statusText,
@@ -3024,13 +3081,18 @@ var Agent = class extends BaseResource {
 	}
 	async declineToolCall(params) {
 		const { requestContext, ...rest } = params;
+		const executableClientTools = params.clientToolsResolver?.() ?? params.clientTools;
+		const releaseHold = this.holdResumingClientTools(rest.runId, executableClientTools);
 		const processedParams = {
 			...rest,
 			requestContext: parseClientRequestContext(requestContext),
-			clientTools: processClientTools(params.clientToolsResolver?.() ?? params.clientTools)
+			clientTools: processClientTools(executableClientTools)
 		};
 		const { readable, controller: readableController, signal } = this.createCancellableStream(params.abortSignal);
-		const response = await this.processStreamResponse(processedParams, readableController, "decline-tool-call", signal);
+		const response = await this.processStreamResponse(processedParams, readableController, "decline-tool-call", signal).catch((error) => {
+			releaseHold({ force: true });
+			throw error;
+		});
 		const streamResponse = new Response(readable, {
 			status: response.status,
 			statusText: response.statusText,
@@ -3090,18 +3152,23 @@ var Agent = class extends BaseResource {
 	* Used to continue execution after a suspension point (e.g., workflow suspend within an agent).
 	*/
 	async resumeStream(resumeData, options) {
+		const executableClientTools = options.clientToolsResolver?.() ?? options.clientTools;
+		const releaseHold = this.holdResumingClientTools(options.runId, executableClientTools);
 		const processedParams = {
 			...options,
 			resumeData,
 			requestContext: parseClientRequestContext(options.requestContext),
-			clientTools: processClientTools(options.clientToolsResolver?.() ?? options.clientTools),
+			clientTools: processClientTools(executableClientTools),
 			structuredOutput: options.structuredOutput ? {
 				...options.structuredOutput,
 				schema: standardSchemaToJSONSchema(toStandardSchema(options.structuredOutput.schema))
 			} : void 0
 		};
 		const { readable, controller: readableController, signal } = this.createCancellableStream(options.abortSignal);
-		const response = await this.processStreamResponse(processedParams, readableController, "resume-stream", signal);
+		const response = await this.processStreamResponse(processedParams, readableController, "resume-stream", signal).catch((error) => {
+			releaseHold({ force: true });
+			throw error;
+		});
 		const streamResponse = new Response(readable, {
 			status: response.status,
 			statusText: response.statusText,
@@ -3154,18 +3221,23 @@ var Agent = class extends BaseResource {
 	* Used to continue execution after a suspension point (e.g., workflow suspend within an agent).
 	*/
 	async resumeStreamUntilIdle(resumeData, options) {
+		const executableClientTools = options.clientToolsResolver?.() ?? options.clientTools;
+		const releaseHold = this.holdResumingClientTools(options.runId, executableClientTools);
 		const processedParams = {
 			...options,
 			resumeData,
 			requestContext: parseClientRequestContext(options.requestContext),
-			clientTools: processClientTools(options.clientToolsResolver?.() ?? options.clientTools),
+			clientTools: processClientTools(executableClientTools),
 			structuredOutput: options.structuredOutput ? {
 				...options.structuredOutput,
 				schema: standardSchemaToJSONSchema(toStandardSchema(options.structuredOutput.schema))
 			} : void 0
 		};
 		const { readable, controller: readableController, signal } = this.createCancellableStream(options.abortSignal);
-		const response = await this.processStreamResponse(processedParams, readableController, "resume-stream-until-idle", signal);
+		const response = await this.processStreamResponse(processedParams, readableController, "resume-stream-until-idle", signal).catch((error) => {
+			releaseHold({ force: true });
+			throw error;
+		});
 		const streamResponse = new Response(readable, {
 			status: response.status,
 			statusText: response.statusText,
@@ -3185,20 +3257,32 @@ var Agent = class extends BaseResource {
 	*/
 	async approveToolCallGenerate(params) {
 		const { requestContext, clientTools, clientToolsResolver, memory, ...rest } = params;
-		const response = await this.request(`/agents/${this.agentId}/approve-tool-call-generate`, {
-			method: "POST",
-			body: {
-				...rest,
-				requestContext: parseClientRequestContext(requestContext),
-				clientTools: processClientTools(clientToolsResolver?.() ?? clientTools)
-			}
-		});
-		return this.continueResumedGenerate(response, {
-			requestContext,
-			clientTools,
-			clientToolsResolver,
-			memory
-		});
+		const executableClientTools = clientToolsResolver?.() ?? clientTools;
+		const releaseHold = this.holdResumingClientTools(rest.runId, executableClientTools);
+		let response;
+		try {
+			response = await this.request(`/agents/${this.agentId}/approve-tool-call-generate`, {
+				method: "POST",
+				body: {
+					...rest,
+					requestContext: parseClientRequestContext(requestContext),
+					clientTools: processClientTools(executableClientTools)
+				}
+			});
+		} catch (error) {
+			releaseHold({ force: true });
+			throw error;
+		}
+		try {
+			return await this.continueResumedGenerate(response, {
+				requestContext,
+				clientTools: executableClientTools,
+				clientToolsResolver,
+				memory
+			});
+		} finally {
+			releaseHold();
+		}
 	}
 	/**
 	* Declines a pending tool call and returns the complete response (non-streaming).
@@ -3206,20 +3290,32 @@ var Agent = class extends BaseResource {
 	*/
 	async declineToolCallGenerate(params) {
 		const { requestContext, clientTools, clientToolsResolver, memory, ...rest } = params;
-		const response = await this.request(`/agents/${this.agentId}/decline-tool-call-generate`, {
-			method: "POST",
-			body: {
-				...rest,
-				requestContext: parseClientRequestContext(requestContext),
-				clientTools: processClientTools(clientToolsResolver?.() ?? clientTools)
-			}
-		});
-		return this.continueResumedGenerate(response, {
-			requestContext,
-			clientTools,
-			clientToolsResolver,
-			memory
-		});
+		const executableClientTools = clientToolsResolver?.() ?? clientTools;
+		const releaseHold = this.holdResumingClientTools(rest.runId, executableClientTools);
+		let response;
+		try {
+			response = await this.request(`/agents/${this.agentId}/decline-tool-call-generate`, {
+				method: "POST",
+				body: {
+					...rest,
+					requestContext: parseClientRequestContext(requestContext),
+					clientTools: processClientTools(executableClientTools)
+				}
+			});
+		} catch (error) {
+			releaseHold({ force: true });
+			throw error;
+		}
+		try {
+			return await this.continueResumedGenerate(response, {
+				requestContext,
+				clientTools: executableClientTools,
+				clientToolsResolver,
+				memory
+			});
+		} finally {
+			releaseHold();
+		}
 	}
 	/**
 	* Runs executable client tools the model called after an approval decision and continues on
@@ -4444,7 +4540,8 @@ async function verifyAgentCardSignatureIfPresent(agentCard, options) {
 			index
 		});
 		if (!verificationKey) throw new Error("No verification key was provided for Agent Card signature verification");
-		await compactVerify(compactJws, await importVerificationKey(verificationKey, protectedHeader.alg), { algorithms: allowedAlgorithms });
+		const importedKey = await importVerificationKey(verificationKey, protectedHeader.alg);
+		await compactVerify(compactJws, importedKey, { algorithms: allowedAlgorithms });
 		return agentCard;
 	} catch (error) {
 		verificationErrors.push(error instanceof Error ? error.message : "Unknown verification failure");
@@ -7696,8 +7793,10 @@ var MastraClient = class extends BaseResource {
 		if (logLevel) searchParams.set("logLevel", logLevel);
 		if (page !== void 0) searchParams.set("page", String(page));
 		if (perPage !== void 0) searchParams.set("perPage", String(perPage));
-		if (_filters) if (Array.isArray(_filters)) for (const filter of _filters) searchParams.append("filters", filter);
-		else searchParams.set("filters", _filters);
+		if (_filters) {
+			if (Array.isArray(_filters)) for (const filter of _filters) searchParams.append("filters", filter);
+			else searchParams.set("filters", _filters);
+		}
 		if (searchParams.size) return this.request(`/logs?${searchParams}`);
 		else return this.request(`/logs`);
 	}
@@ -7717,8 +7816,10 @@ var MastraClient = class extends BaseResource {
 		if (logLevel) searchParams.set("logLevel", logLevel);
 		if (page !== void 0) searchParams.set("page", String(page));
 		if (perPage !== void 0) searchParams.set("perPage", String(perPage));
-		if (_filters) if (Array.isArray(_filters)) for (const filter of _filters) searchParams.append("filters", filter);
-		else searchParams.set("filters", _filters);
+		if (_filters) {
+			if (Array.isArray(_filters)) for (const filter of _filters) searchParams.append("filters", filter);
+			else searchParams.set("filters", _filters);
+		}
 		if (searchParams.size) return this.request(`/logs/${runId}?${searchParams}`);
 		else return this.request(`/logs/${runId}`);
 	}
@@ -7795,6 +7896,77 @@ var MastraClient = class extends BaseResource {
 		});
 	}
 	/**
+	* Lists the event types an MCP server declares (experimental).
+	* `testEmit` is true when the server serves {@link emitMcpServerEvent} (under `mastra dev`).
+	* @param serverId - The ID of the MCP server.
+	*/
+	getMcpServerEvents(serverId) {
+		return this.request(`/mcp/${encodeURIComponent(serverId)}/events`);
+	}
+	/**
+	* Reads occurrences of an MCP server event type, like the `events/poll` MCP request (experimental).
+	* Pass the returned `cursor` to the next call to resume.
+	* @param serverId - The ID of the MCP server.
+	* @param eventName - The event type name.
+	* @param body - Event arguments, resume cursor and page size.
+	*/
+	pollMcpServerEvent(serverId, eventName, body = {}) {
+		return this.request(`/mcp/${encodeURIComponent(serverId)}/events/${encodeURIComponent(eventName)}/poll`, {
+			method: "POST",
+			body
+		});
+	}
+	/**
+	* Streams occurrences of a push-capable MCP server event type (experimental).
+	* Yields an `active` frame first, then `event`, `heartbeat` and `error` frames, and ends
+	* when the server ends the stream. Abort `options.signal` or break out of the loop to stop.
+	*
+	* @example
+	* ```typescript
+	* const controller = new AbortController();
+	* for await (const frame of client.streamMcpServerEvent('tickets', 'ticket.created', {}, { signal: controller.signal })) {
+	*   if (frame.type === 'event') console.log(frame.occurrence.data);
+	* }
+	* ```
+	*/
+	async *streamMcpServerEvent(serverId, eventName, body = {}, options = {}) {
+		const response = await this.request(`/mcp/${encodeURIComponent(serverId)}/events/${encodeURIComponent(eventName)}/stream`, {
+			method: "POST",
+			body,
+			stream: true,
+			signal: options.signal,
+			retries: 0
+		});
+		if (!response.body) throw new Error("Response body is null");
+		const reader = response.body.pipeThrough(createSseJsonTransform()).getReader();
+		const cancel = () => void reader.cancel().catch(() => {});
+		options.signal?.addEventListener("abort", cancel, { once: true });
+		try {
+			while (true) {
+				const { value, done } = await reader.read();
+				if (done) return;
+				yield value;
+			}
+		} finally {
+			options.signal?.removeEventListener("abort", cancel);
+			await reader.cancel().catch(() => {});
+		}
+	}
+	/**
+	* Emits a test occurrence of an emit-fed MCP server event type (experimental).
+	* Only served under `mastra dev`; other servers respond 404. The occurrence's
+	* `eventId` starts with `mastra-test-`.
+	* @param serverId - The ID of the MCP server.
+	* @param eventName - The event type name.
+	* @param data - The event payload; validated against the event's payload schema.
+	*/
+	emitMcpServerEvent(serverId, eventName, data) {
+		return this.request(`/mcp/${encodeURIComponent(serverId)}/events/${encodeURIComponent(eventName)}/emit`, {
+			method: "POST",
+			body: { data }
+		});
+	}
+	/**
 	* Gets an A2A client for interacting with an agent via the A2A protocol
 	* @param agentId - ID of the agent to interact with
 	* @returns A2A client instance