@mastra/client-js 1.56.0 → 1.57.0-alpha.1

package/dist/index.d.ts

@@ -11,6 +11,6 @@ export type { ObservabilityCollector } from './observability/types.js';
 export type { QuerySpansInput, QuerySpansResult, QueryTraceThreadsInput, QueryTraceThreadsResult, QueryTracesInput, QueryTracesDeltaInput, QueryTracesKeysetInput, QueryTracesPaginatedInput, } from './resources/observability.js';
 export type { UIMessageWithMetadata } from '@mastra/core/agent';
 export type { GetMetricTimeSeriesResponse } from './types.js';
-export type { AggregateTracesArgs, AggregateTracesResponse, TraceAggregateDiscovery, GetTraceQueryFieldsArgs, GetTraceQueryFieldsResponse, GetTraceQueryValuesArgs, GetTraceQueryValuesResponse, TraceQueryCanonicalFieldDescriptor, TraceQueryGroupResponse, TraceQueryKeysetTraceResponse, TraceQueryObservedFieldDescriptor, TraceQueryOperator, TraceQueryPredicateScope, TraceQueryResponse, TraceQueryTraceResponse, TraceQueryValueKind, } from './resources/observability-route-types.js';
+export type { AggregateTracesArgs, AggregateTracesResponse, AggregateSpansArgs, AggregateSpansResponse, TraceAggregateDiscovery, GetTraceQueryFieldsArgs, GetTraceQueryFieldsResponse, GetTraceQueryValuesArgs, GetTraceQueryValuesResponse, TraceQueryCanonicalFieldDescriptor, TraceQueryGroupResponse, TraceQueryKeysetTraceResponse, TraceQueryObservedFieldDescriptor, TraceQueryOperator, TraceQueryPredicateScope, TraceQueryResponse, TraceQueryTraceResponse, TraceQueryValueKind, } from './resources/observability-route-types.js';
 export type { Body, Client, ClientMethod, ClientPath, ClientRequest, ClientResponse, ClientResponseKind, ClientRoute, PathParams, QueryParams, RouteKey, RouteRequest, RouteResponse, RouteResponseType, RouteTypes, } from './route-types.generated.js';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,kBAAkB,CAAC;AACjC,cAAc,UAAU,CAAC;AACzB,cAAc,SAAS,CAAC;AACxB,cAAc,SAAS,CAAC;AACxB,YAAY,EACV,mBAAmB,EACnB,uBAAuB,EACvB,mBAAmB,EACnB,sBAAsB,EACtB,uBAAuB,EACvB,oBAAoB,GACrB,MAAM,sBAAsB,CAAC;AAC9B,YAAY,EACV,kCAAkC,EAClC,wBAAwB,EACxB,mBAAmB,EACnB,+BAA+B,GAChC,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,0BAA0B,EAAE,2BAA2B,EAAE,MAAM,8BAA8B,CAAC;AACvG,YAAY,EACV,eAAe,EACf,sBAAsB,EACtB,iBAAiB,EACjB,oBAAoB,EACpB,yBAAyB,EACzB,yBAAyB,EACzB,6BAA6B,EAC7B,sCAAsC,EACtC,2BAA2B,EAC3B,UAAU,EACV,yBAAyB,EACzB,+BAA+B,GAChC,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAG9D,YAAY,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AACpE,YAAY,EACV,eAAe,EACf,gBAAgB,EAChB,sBAAsB,EACtB,uBAAuB,EACvB,gBAAgB,EAChB,qBAAqB,EACrB,sBAAsB,EACtB,yBAAyB,GAC1B,MAAM,2BAA2B,CAAC;AACnC,YAAY,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAChE,YAAY,EAAE,2BAA2B,EAAE,MAAM,SAAS,CAAC;AAC3D,YAAY,EACV,mBAAmB,EACnB,uBAAuB,EACvB,uBAAuB,EACvB,uBAAuB,EACvB,2BAA2B,EAC3B,uBAAuB,EACvB,2BAA2B,EAC3B,kCAAkC,EAClC,uBAAuB,EACvB,6BAA6B,EAC7B,iCAAiC,EACjC,kBAAkB,EAClB,wBAAwB,EACxB,kBAAkB,EAClB,uBAAuB,EACvB,mBAAmB,GACpB,MAAM,uCAAuC,CAAC;AAC/C,YAAY,EACV,IAAI,EACJ,MAAM,EACN,YAAY,EACZ,UAAU,EACV,aAAa,EACb,cAAc,EACd,kBAAkB,EAClB,WAAW,EACX,UAAU,EACV,WAAW,EACX,QAAQ,EACR,YAAY,EACZ,aAAa,EACb,iBAAiB,EACjB,UAAU,GACX,MAAM,4BAA4B,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,kBAAkB,CAAC;AACjC,cAAc,UAAU,CAAC;AACzB,cAAc,SAAS,CAAC;AACxB,cAAc,SAAS,CAAC;AACxB,YAAY,EACV,mBAAmB,EACnB,uBAAuB,EACvB,mBAAmB,EACnB,sBAAsB,EACtB,uBAAuB,EACvB,oBAAoB,GACrB,MAAM,sBAAsB,CAAC;AAC9B,YAAY,EACV,kCAAkC,EAClC,wBAAwB,EACxB,mBAAmB,EACnB,+BAA+B,GAChC,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,0BAA0B,EAAE,2BAA2B,EAAE,MAAM,8BAA8B,CAAC;AACvG,YAAY,EACV,eAAe,EACf,sBAAsB,EACtB,iBAAiB,EACjB,oBAAoB,EACpB,yBAAyB,EACzB,yBAAyB,EACzB,6BAA6B,EAC7B,sCAAsC,EACtC,2BAA2B,EAC3B,UAAU,EACV,yBAAyB,EACzB,+BAA+B,GAChC,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAG9D,YAAY,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AACpE,YAAY,EACV,eAAe,EACf,gBAAgB,EAChB,sBAAsB,EACtB,uBAAuB,EACvB,gBAAgB,EAChB,qBAAqB,EACrB,sBAAsB,EACtB,yBAAyB,GAC1B,MAAM,2BAA2B,CAAC;AACnC,YAAY,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAChE,YAAY,EAAE,2BAA2B,EAAE,MAAM,SAAS,CAAC;AAC3D,YAAY,EACV,mBAAmB,EACnB,uBAAuB,EACvB,kBAAkB,EAClB,sBAAsB,EACtB,uBAAuB,EACvB,uBAAuB,EACvB,2BAA2B,EAC3B,uBAAuB,EACvB,2BAA2B,EAC3B,kCAAkC,EAClC,uBAAuB,EACvB,6BAA6B,EAC7B,iCAAiC,EACjC,kBAAkB,EAClB,wBAAwB,EACxB,kBAAkB,EAClB,uBAAuB,EACvB,mBAAmB,GACpB,MAAM,uCAAuC,CAAC;AAC/C,YAAY,EACV,IAAI,EACJ,MAAM,EACN,YAAY,EACZ,UAAU,EACV,aAAa,EACb,cAAc,EACd,kBAAkB,EAClB,WAAW,EACX,UAAU,EACV,WAAW,EACX,QAAQ,EACR,YAAY,EACZ,aAAa,EACb,iBAAiB,EACjB,UAAU,GACX,MAAM,4BAA4B,CAAC"}

package/dist/index.js

@@ -1056,11 +1056,51 @@ function processClientTools(clientTools) {
 	}));
 }
 //#endregion
+//#region src/utils/sse-event-splitter.ts
+/** Split SSE events without rescanning or copying an unfinished event on each read. */
+function createSseEventSplitter() {
+	let parts = [];
+	let pendingNewline = false;
+	return (text, isFinal = false) => {
+		const events = [];
+		let start = 0;
+		if (text) {
+			if (pendingNewline) {
+				if (text.startsWith("\n")) {
+					events.push(parts.join(""));
+					parts = [];
+					start = 1;
+				} else parts.push("\n");
+				pendingNewline = false;
+			}
+			let separator;
+			while ((separator = text.indexOf("\n\n", start)) !== -1) {
+				parts.push(text.slice(start, separator));
+				events.push(parts.join(""));
+				parts = [];
+				start = separator + 2;
+			}
+			if (start < text.length) {
+				pendingNewline = text.endsWith("\n");
+				const end = pendingNewline ? text.length - 1 : text.length;
+				if (start < end) parts.push(text.slice(start, end));
+			}
+		}
+		if (isFinal) {
+			if (pendingNewline) parts.push("\n");
+			events.push(parts.join(""));
+			parts = [];
+			pendingNewline = false;
+		}
+		return events;
+	};
+}
+//#endregion
 //#region src/utils/process-mastra-stream.ts
 async function sharedProcessMastraStream({ stream, onChunk, signal }) {
 	const reader = stream.getReader();
 	const decoder = new TextDecoder();
-	let buffer = "";
+	const splitEvents = createSseEventSplitter();
 	const abort = () => void reader.cancel();
 	if (signal?.aborted) abort();
 	else signal?.addEventListener("abort", abort, { once: true });
@@ -1068,9 +1108,7 @@ async function sharedProcessMastraStream({ stream, onChunk, signal }) {
 		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() || "";
+			const lines = splitEvents(decoder.decode(value, { stream: true }));
 			for (const line of lines) if (line.startsWith("data: ")) {
 				const data = line.slice(6);
 				if (data === "[DONE]") return;
@@ -1870,10 +1908,18 @@ var Agent = class extends BaseResource {
 		};
 		return streamResponse;
 	}
-	async abortThread(params) {
+	/** Stops execution of a run, including runs without a memory thread. */
+	async abortRun(runId, options) {
+		return this.request(`/agents/${this.agentId}/runs/${encodeURIComponent(runId)}/abort`, {
+			method: "POST",
+			signal: options?.signal
+		});
+	}
+	async abortThread(params, options) {
 		const { resourceId, threadId, clearPendingSignals, expectedRunId } = params;
 		return this.request(`/agents/${this.agentId}/threads/abort`, {
 			method: "POST",
+			signal: options?.signal,
 			body: {
 				resourceId,
 				threadId,
@@ -2607,10 +2653,11 @@ var Agent = class extends BaseResource {
 	}
 	/**
 	* Creates the consumer-facing stream for the streaming methods. Cancelling the
-	* returned stream (or aborting `externalSignal`) aborts the underlying request
-	* and stops any client-tool continuations.
+	* returned stream or the client-wide signal detaches the HTTP observer and
+	* stops any client-tool continuations. Per-call execution cancellation is
+	* handled separately by processStreamResponse.
 	*/
-	createCancellableStream(externalSignal) {
+	createCancellableStream() {
 		const abortController = new AbortController();
 		let readableController;
 		return {
@@ -2623,10 +2670,63 @@ var Agent = class extends BaseResource {
 				}
 			}),
 			controller: readableController,
-			signal: mergeAbortSignals(abortController.signal, externalSignal, this.options.abortSignal)
+			signal: mergeAbortSignals(abortController.signal, this.options.abortSignal)
 		};
 	}
 	async processStreamResponse(processedParams, controller, route = "stream", signal) {
+		const suppliedRunId = processedParams.runId;
+		processedParams = {
+			...processedParams,
+			runId: processedParams.runId ?? v4()
+		};
+		const callerSignal = processedParams.abortSignal;
+		callerSignal?.throwIfAborted();
+		const transportAbort = new AbortController();
+		const cancellationAbort = new AbortController();
+		const detachSignal = signal;
+		signal = mergeAbortSignals(detachSignal, transportAbort.signal);
+		let cancellationTimer;
+		const stopCancellation = () => cancellationAbort.abort();
+		const cancelExecution = () => {
+			cancellationTimer = setTimeout(() => cancellationAbort.abort(/* @__PURE__ */ new Error("Agent run cancellation was not acknowledged")), 1e4);
+			(async () => {
+				try {
+					while (true) {
+						cancellationAbort.signal.throwIfAborted();
+						if ((await this.abortRun(processedParams.runId, { signal: cancellationAbort.signal })).aborted) return;
+						await new Promise((resolve, reject) => {
+							const onAbort = () => {
+								clearTimeout(timer);
+								reject(cancellationAbort.signal.reason);
+							};
+							const timer = setTimeout(() => {
+								cancellationAbort.signal.removeEventListener("abort", onAbort);
+								resolve();
+							}, 100);
+							cancellationAbort.signal.addEventListener("abort", onAbort, { once: true });
+							if (cancellationAbort.signal.aborted) onAbort();
+						});
+					}
+				} catch (error) {
+					if (!cancellationAbort.signal.aborted && !(error instanceof Error && "status" in error && (error.status === 404 || error.status === 501))) try {
+						controller.error(error);
+					} catch {}
+				} finally {
+					clearTimeout(cancellationTimer);
+					transportAbort.abort(callerSignal?.reason);
+				}
+			})();
+		};
+		const cleanupCancellation = () => {
+			callerSignal?.removeEventListener("abort", cancelExecution);
+			detachSignal?.removeEventListener("abort", stopCancellation);
+			clearTimeout(cancellationTimer);
+			stopCancellation();
+		};
+		detachSignal?.addEventListener("abort", stopCancellation, { once: true });
+		if (detachSignal?.aborted) stopCancellation();
+		callerSignal?.addEventListener("abort", cancelExecution, { once: true });
+		if (callerSignal?.aborted) cancelExecution();
 		const { memory } = processedParams ?? {};
 		const { resource, thread } = memory ?? {};
 		const threadId = processedParams.threadId ?? (typeof thread === "string" ? thread : thread?.id);
@@ -2643,9 +2743,15 @@ var Agent = class extends BaseResource {
 			body: requestBody,
 			stream: true,
 			signal
+		}).catch((error) => {
+			cleanupCancellation();
+			throw error;
 		});
-		if (!response.body) throw new Error("No response body");
-		const isAborted = () => Boolean(signal?.aborted);
+		if (!response.body) {
+			cleanupCancellation();
+			throw new Error("No response body");
+		}
+		const isAborted = () => Boolean(signal?.aborted || callerSignal?.aborted);
 		const isAbortError = (error) => isAborted() || error?.name === "AbortError";
 		const safeClose = () => {
 			try {
@@ -2658,18 +2764,16 @@ var Agent = class extends BaseResource {
 		};
 		try {
 			let messages = [];
-			let streamRunId = processedParams.runId;
+			let streamRunId = suppliedRunId;
 			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 = "";
+			const splitEvents = createSseEventSplitter();
 			const enqueueReadableText = (text, isFinal = false) => {
-				pendingText += text;
-				const lines = pendingText.split("\n\n");
-				pendingText = isFinal ? "" : lines.pop() ?? "";
+				const lines = splitEvents(text, isFinal);
 				for (const line of lines) {
 					if (!line.startsWith("data: ") || !line.includes("\"type\":\"finish\"")) continue;
 					try {
@@ -2775,6 +2879,7 @@ var Agent = class extends BaseResource {
 										payload: {
 											toolCallId: toolCall.toolCallId,
 											toolName: toolCall.toolName,
+											args: toolCall.args,
 											result,
 											isError: false,
 											providerExecuted: false
@@ -2875,10 +2980,12 @@ var Agent = class extends BaseResource {
 				} catch {}
 				safeClose();
 			}).finally(() => {
+				cleanupCancellation();
 				signal?.removeEventListener("abort", onAbort);
 				releaseOpenStream();
 			});
 		} catch (error) {
+			cleanupCancellation();
 			if (!isAbortError(error)) console.error("Error processing stream response:", error);
 		}
 		return response;
@@ -2976,7 +3083,7 @@ var Agent = class extends BaseResource {
 			clientTools: processClientTools(params.clientToolsResolver?.() ?? params.clientTools),
 			structuredOutput
 		};
-		const { readable, controller: readableController, signal } = this.createCancellableStream(params.abortSignal);
+		const { readable, controller: readableController, signal } = this.createCancellableStream();
 		const response = await this.processStreamResponse(processedParams, readableController, "stream", signal);
 		const streamResponse = new Response(readable, {
 			status: response.status,
@@ -3007,7 +3114,7 @@ var Agent = class extends BaseResource {
 			clientTools: processClientTools(params.clientToolsResolver?.() ?? params.clientTools),
 			structuredOutput
 		};
-		const { readable, controller: readableController, signal } = this.createCancellableStream(params.abortSignal);
+		const { readable, controller: readableController, signal } = this.createCancellableStream();
 		const response = await this.processStreamResponse(processedParams, readableController, "stream-until-idle", signal);
 		const streamResponse = new Response(readable, {
 			status: response.status,
@@ -3051,7 +3158,7 @@ var Agent = class extends BaseResource {
 			requestContext: parseClientRequestContext(requestContext),
 			clientTools: processClientTools(executableClientTools)
 		};
-		const { readable, controller: readableController, signal } = this.createCancellableStream(params.abortSignal);
+		const { readable, controller: readableController, signal } = this.createCancellableStream();
 		const response = await this.processStreamResponse(processedParams, readableController, "approve-tool-call", signal).catch((error) => {
 			releaseHold({ force: true });
 			throw error;
@@ -3088,7 +3195,7 @@ var Agent = class extends BaseResource {
 			requestContext: parseClientRequestContext(requestContext),
 			clientTools: processClientTools(executableClientTools)
 		};
-		const { readable, controller: readableController, signal } = this.createCancellableStream(params.abortSignal);
+		const { readable, controller: readableController, signal } = this.createCancellableStream();
 		const response = await this.processStreamResponse(processedParams, readableController, "decline-tool-call", signal).catch((error) => {
 			releaseHold({ force: true });
 			throw error;
@@ -3164,7 +3271,7 @@ var Agent = class extends BaseResource {
 				schema: standardSchemaToJSONSchema(toStandardSchema(options.structuredOutput.schema))
 			} : void 0
 		};
-		const { readable, controller: readableController, signal } = this.createCancellableStream(options.abortSignal);
+		const { readable, controller: readableController, signal } = this.createCancellableStream();
 		const response = await this.processStreamResponse(processedParams, readableController, "resume-stream", signal).catch((error) => {
 			releaseHold({ force: true });
 			throw error;
@@ -3233,7 +3340,7 @@ var Agent = class extends BaseResource {
 				schema: standardSchemaToJSONSchema(toStandardSchema(options.structuredOutput.schema))
 			} : void 0
 		};
-		const { readable, controller: readableController, signal } = this.createCancellableStream(options.abortSignal);
+		const { readable, controller: readableController, signal } = this.createCancellableStream();
 		const response = await this.processStreamResponse(processedParams, readableController, "resume-stream-until-idle", signal).catch((error) => {
 			releaseHold({ force: true });
 			throw error;
@@ -5266,6 +5373,17 @@ var Observability = class extends BaseResource {
 			body: params
 		});
 	}
+	/**
+	* Groups and measures individual completed spans selected by a time range and optional span-query predicate.
+	* Token and cost measures count usage recorded on each span itself. Check `capabilities.spanAggregate` from
+	* `getObservabilityCapabilities()` first; servers without it return 404 or 501.
+	*/
+	aggregateSpans(params) {
+		return this.request("/observability/spans/aggregate", {
+			method: "POST",
+			body: params
+		});
+	}
 	/** Returns canonical and observed fields available to the advanced trace-query grammar. */
 	getTraceQueryFields(params, options) {
 		return this.request("/observability/traces/query/fields", {
@@ -8130,6 +8248,14 @@ var MastraClient = class extends BaseResource {
 	aggregateTraces(params) {
 		return this.observability.aggregateTraces(params);
 	}
+	/**
+	* Groups and measures individual completed spans selected by a time range and optional span-query predicate.
+	* Token and cost measures count usage recorded on each span itself. Check `capabilities.spanAggregate` from
+	* `getObservabilityCapabilities()` first; servers without it return 404 or 501.
+	*/
+	aggregateSpans(params) {
+		return this.observability.aggregateSpans(params);
+	}
 	/** Returns canonical and observed fields available to the advanced trace-query grammar. */
 	getTraceQueryFields(params, options) {
 		return this.observability.getTraceQueryFields(params, options);