@mastra/client-js 1.47.0-alpha.4 → 1.47.0-alpha.6

package/dist/index.d.ts

@@ -11,5 +11,6 @@ export type { ObservabilityCollector } from './observability/types.js';
 export type { QueryTraceThreadsInput, QueryTraceThreadsResult, QueryTracesInput } from './resources/observability.js';
 export type { UIMessageWithMetadata } from '@mastra/core/agent';
 export type { GetMetricTimeSeriesResponse } from './types.js';
+export type { GetTraceQueryFieldsArgs, GetTraceQueryFieldsResponse, GetTraceQueryValuesArgs, GetTraceQueryValuesResponse, TraceQueryCanonicalFieldDescriptor, TraceQueryObservedFieldDescriptor, TraceQueryOperator, TraceQueryPredicateScope, 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,GACX,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAG9D,YAAY,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AACpE,YAAY,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AACnH,YAAY,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAChE,YAAY,EAAE,2BAA2B,EAAE,MAAM,SAAS,CAAC;AAC3D,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,GACX,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAG9D,YAAY,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AACpE,YAAY,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AACnH,YAAY,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAChE,YAAY,EAAE,2BAA2B,EAAE,MAAM,SAAS,CAAC;AAC3D,YAAY,EACV,uBAAuB,EACvB,2BAA2B,EAC3B,uBAAuB,EACvB,2BAA2B,EAC3B,kCAAkC,EAClC,iCAAiC,EACjC,kBAAkB,EAClB,wBAAwB,EACxB,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

@@ -341,6 +341,33 @@ function buildTenancyQuery(tenancy) {
 	const queryString = searchParams.toString();
 	return queryString ? `?${queryString}` : "";
 }
+/**
+* Merges multiple optional abort signals into one. Returns `undefined` when no
+* signals are provided, the single signal when only one is provided, and a
+* combined signal otherwise. Uses `AbortSignal.any` when available and falls
+* back to a manual listener merge for runtimes that lack it.
+*/
+function mergeAbortSignals(...signals) {
+	const defined = signals.filter((signal) => signal !== void 0);
+	if (defined.length === 0) return void 0;
+	if (defined.length === 1) return defined[0];
+	if (typeof AbortSignal.any === "function") return AbortSignal.any(defined);
+	const controller = new AbortController();
+	const alreadyAborted = defined.find((signal) => signal.aborted);
+	if (alreadyAborted) {
+		controller.abort(alreadyAborted.reason);
+		return controller.signal;
+	}
+	const cleanup = () => {
+		for (const signal of defined) signal.removeEventListener("abort", onAbort);
+	};
+	function onAbort() {
+		cleanup();
+		controller.abort(this.reason);
+	}
+	for (const signal of defined) signal.addEventListener("abort", onAbort);
+	return controller.signal;
+}
 //#endregion
 //#region src/utils/client-tool-model-output.ts
 /**
@@ -535,43 +562,61 @@ var BaseResource = class {
 	*/
 	async request(path, options = {}) {
 		let lastError = null;
-		const { baseUrl, retries = 3, backoffMs = 100, maxBackoffMs = 1e3, headers = {}, credentials, fetch: customFetch } = this.options;
+		const { baseUrl, retries: defaultRetries = 3, backoffMs = 100, maxBackoffMs = 1e3, headers = {}, credentials, fetch: customFetch } = this.options;
+		const { retries: requestRetries, ...fetchOptions } = options;
+		const retries = requestRetries ?? defaultRetries;
+		if (!Number.isSafeInteger(retries) || retries < 0) throw new RangeError("retries must be a non-negative safe integer");
+		const signal = mergeAbortSignals(this.options.abortSignal, fetchOptions.signal);
 		const fetchFn = customFetch || fetch;
 		let delay = backoffMs;
 		const fullPath = `${this.apiPrefix}${path}`;
-		for (let attempt = 0; attempt <= retries; attempt++) try {
-			const response = await fetchFn(`${baseUrl.replace(/\/$/, "")}${fullPath}`, {
-				...options,
-				headers: {
-					...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
-				},
-				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) {
-				const errorBody = await response.text();
-				let parsedBody;
-				let errorMessage = `HTTP error! status: ${response.status}`;
-				try {
-					parsedBody = JSON.parse(errorBody);
-					errorMessage += ` - ${JSON.stringify(parsedBody)}`;
-				} catch {
-					if (errorBody) errorMessage += ` - ${errorBody}`;
+		for (let attempt = 0; attempt <= retries; attempt++) {
+			if (signal?.aborted) throw signal.reason ?? new DOMException("This operation was aborted", "AbortError");
+			try {
+				const response = await fetchFn(`${baseUrl.replace(/\/$/, "")}${fullPath}`, {
+					...fetchOptions,
+					headers: {
+						...fetchOptions.body && !(fetchOptions.body instanceof FormData) && (fetchOptions.method === "POST" || fetchOptions.method === "PUT" || fetchOptions.method === "PATCH" || fetchOptions.method === "DELETE") ? { "content-type": "application/json" } : {},
+						...headers,
+						...fetchOptions.headers
+					},
+					signal,
+					credentials: fetchOptions.credentials ?? credentials,
+					body: fetchOptions.body instanceof FormData ? fetchOptions.body : fetchOptions.body ? JSON.stringify(fetchOptions.body) : void 0
+				});
+				if (!response.ok) {
+					const errorBody = await response.text();
+					let parsedBody;
+					let errorMessage = `HTTP error! status: ${response.status}`;
+					try {
+						parsedBody = JSON.parse(errorBody);
+						errorMessage += ` - ${JSON.stringify(parsedBody)}`;
+					} catch {
+						if (errorBody) errorMessage += ` - ${errorBody}`;
+					}
+					throw new MastraClientError(response.status, response.statusText, errorMessage, parsedBody);
 				}
-				throw new MastraClientError(response.status, response.statusText, errorMessage, parsedBody);
+				if (options.stream) return response;
+				return await response.json();
+			} catch (error) {
+				lastError = error;
+				if (signal?.aborted || error?.name === "AbortError") throw error;
+				const status = error.status;
+				if (status !== void 0 && status >= 400 && status < 500) throw error;
+				if (attempt === retries) break;
+				await new Promise((resolve, reject) => {
+					const timer = setTimeout(() => {
+						signal?.removeEventListener("abort", onAbort);
+						resolve();
+					}, delay);
+					const onAbort = () => {
+						clearTimeout(timer);
+						reject(signal?.reason ?? new DOMException("This operation was aborted", "AbortError"));
+					};
+					signal?.addEventListener("abort", onAbort, { once: true });
+				});
+				delay = Math.min(delay * 2, maxBackoffMs);
 			}
-			if (options.stream) return response;
-			return await response.json();
-		} catch (error) {
-			lastError = error;
-			const status = error.status;
-			if (status !== void 0 && status >= 400 && status < 500) throw error;
-			if (attempt === retries) break;
-			await new Promise((resolve) => setTimeout(resolve, delay));
-			delay = Math.min(delay * 2, maxBackoffMs);
 		}
 		throw lastError || /* @__PURE__ */ new Error("Request failed");
 	}
@@ -1301,8 +1346,10 @@ var Agent = class extends BaseResource {
 		return this.request(`/stored/agents/${encodeURIComponent(this.agentId)}/versions/compare?${queryParams.toString()}${contextString ? `&${contextString.slice(1)}` : ""}`);
 	}
 	async generateLegacy(params) {
+		const { abortSignal: perCallSignal, ...bodyParams } = params;
+		const abortSignal = mergeAbortSignals(this.options.abortSignal, perCallSignal);
 		const processedParams = {
-			...params,
+			...bodyParams,
 			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),
@@ -1311,9 +1358,11 @@ var Agent = class extends BaseResource {
 		const { resourceId, threadId, requestContext } = processedParams;
 		const response = await this.request(`/agents/${this.agentId}/generate-legacy`, {
 			method: "POST",
-			body: processedParams
+			body: processedParams,
+			signal: abortSignal
 		});
 		if (response.finishReason === "tool-calls") {
+			if (abortSignal?.aborted) throw abortSignal.reason ?? new DOMException("This operation was aborted", "AbortError");
 			const toolCalls = response.toolCalls;
 			if (!toolCalls || !Array.isArray(toolCalls)) return response;
 			for (const toolCall of toolCalls) {
@@ -1362,8 +1411,10 @@ var Agent = class extends BaseResource {
 			messages
 		};
 		const resolvedClientTools = params.clientToolsResolver?.() ?? params.clientTools;
+		const abortSignal = mergeAbortSignals(this.options.abortSignal, params.abortSignal);
 		const processedParams = {
 			...params,
+			abortSignal: void 0,
 			requestContext: parseClientRequestContext(params.requestContext),
 			clientTools: processClientTools(resolvedClientTools),
 			structuredOutput: params.structuredOutput ? {
@@ -1377,20 +1428,24 @@ var Agent = class extends BaseResource {
 		const threadId = typeof thread === "string" ? thread : thread?.id;
 		const response = await this.request(`/agents/${this.agentId}/generate`, {
 			method: "POST",
-			body: processedParams
-		});
-		if (response.finishReason === "tool-calls") return executeToolCallAndRespond({
-			response,
-			params: {
-				...params,
-				clientTools: resolvedClientTools
-			},
-			agentId: this.agentId,
-			resourceId,
-			threadId,
-			requestContext,
-			respondFn: this.generate.bind(this)
+			body: processedParams,
+			signal: abortSignal
 		});
+		if (response.finishReason === "tool-calls") {
+			if (abortSignal?.aborted) throw abortSignal.reason ?? new DOMException("This operation was aborted", "AbortError");
+			return executeToolCallAndRespond({
+				response,
+				params: {
+					...params,
+					clientTools: resolvedClientTools
+				},
+				agentId: this.agentId,
+				resourceId,
+				threadId,
+				requestContext,
+				respondFn: this.generate.bind(this)
+			});
+		}
 		return response;
 	}
 	async processChatResponse({ stream, update, onToolCall, onFinish, getCurrentDate = () => /* @__PURE__ */ new Date(), lastMessage }) {
@@ -1626,7 +1681,7 @@ var Agent = class extends BaseResource {
 				throw new Error(error);
 			}
 		});
-		onFinish?.({
+		await onFinish?.({
 			message,
 			finishReason,
 			usage
@@ -1645,8 +1700,22 @@ var Agent = class extends BaseResource {
 			requestContext: parseClientRequestContext(params.requestContext),
 			clientTools: processClientTools(params.clientToolsResolver?.() ?? params.clientTools)
 		};
-		const { readable, writable } = new TransformStream();
-		const response = await this.processStreamResponseLegacy(processedParams, writable);
+		const { readable: innerReadable, writable } = new TransformStream();
+		const abortController = new AbortController();
+		const signal = mergeAbortSignals(abortController.signal, params.abortSignal, this.options.abortSignal);
+		const innerReader = innerReadable.getReader();
+		const readable = new ReadableStream({
+			async pull(controller) {
+				const { done, value } = await innerReader.read();
+				if (done) controller.close();
+				else controller.enqueue(value);
+			},
+			cancel(reason) {
+				abortController.abort(reason);
+				return innerReader.cancel(reason).catch(() => {});
+			}
+		});
+		const response = await this.processStreamResponseLegacy(processedParams, writable, signal);
 		const streamResponse = new Response(readable, {
 			status: response.status,
 			statusText: response.statusText,
@@ -1887,29 +1956,63 @@ var Agent = class extends BaseResource {
 				}
 			}
 		});
-		onFinish?.({
+		await onFinish?.({
 			message,
 			finishReason,
 			usage
 		});
 	}
-	async processStreamResponse(processedParams, controller, route = "stream") {
+	/**
+	* 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.
+	*/
+	createCancellableStream(externalSignal) {
+		const abortController = new AbortController();
+		let readableController;
+		return {
+			readable: new ReadableStream({
+				start(controller) {
+					readableController = controller;
+				},
+				cancel(reason) {
+					abortController.abort(reason);
+				}
+			}),
+			controller: readableController,
+			signal: mergeAbortSignals(abortController.signal, externalSignal, this.options.abortSignal)
+		};
+	}
+	async processStreamResponse(processedParams, controller, route = "stream", signal) {
 		const { memory } = processedParams ?? {};
 		const { resource, thread } = memory ?? {};
 		const threadId = processedParams.threadId ?? (typeof thread === "string" ? thread : thread?.id);
 		const resourceId = processedParams.resourceId ?? resource;
-		let requestBody = processedParams;
+		const { abortSignal: _abortSignal, ...serializableParams } = processedParams;
+		let requestBody = serializableParams;
 		if (route === "resume-stream") {
-			const { messages: _messages, ...resumeStreamBody } = processedParams;
+			const { messages: _messages, ...resumeStreamBody } = serializableParams;
 			requestBody = resumeStreamBody;
 		}
 		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
+			stream: true,
+			signal
 		});
 		if (!response.body) throw new Error("No response body");
+		const isAborted = () => Boolean(signal?.aborted);
+		const isAbortError = (error) => isAborted() || error?.name === "AbortError";
+		const safeClose = () => {
+			try {
+				controller.close();
+			} catch {}
+		};
+		const safeEnqueue = (chunk) => {
+			if (isAborted()) return;
+			controller.enqueue(chunk);
+		};
 		try {
 			let messages = [];
 			let streamRunId = processedParams.runId;
@@ -1921,7 +2024,7 @@ var Agent = class extends BaseResource {
 				const lines = pendingText.split("\n\n");
 				pendingText = isFinal ? "" : lines.pop() ?? "";
 				const readableLines = lines.filter((line) => line.trim() !== "[DONE]" && line.trim() !== "data: [DONE]").join("\n\n");
-				if (readableLines) controller.enqueue(new TextEncoder().encode(`${readableLines}\n\n`));
+				if (readableLines) safeEnqueue(new TextEncoder().encode(`${readableLines}\n\n`));
 			};
 			const pipePromise = streamForController.pipeTo(new WritableStream({
 				async write(chunk) {
@@ -1929,26 +2032,32 @@ var Agent = class extends BaseResource {
 						enqueueReadableText(decoder.decode(chunk, { stream: true }));
 					} catch (error) {
 						console.error("Error enqueueing to controller:", error);
-						controller.enqueue(chunk);
+						safeEnqueue(chunk);
 					}
 				},
 				close() {
 					enqueueReadableText(decoder.decode(), true);
 				}
-			})).catch((error) => {
-				console.error("Error piping to controller:", error);
+			}), { signal }).catch((error) => {
+				if (!isAbortError(error)) console.error("Error piping to controller:", error);
+				safeClose();
+			});
+			const onAbort = () => {
 				try {
-					controller.close();
+					controller.error(signal?.reason);
 				} catch {}
-			});
+			};
+			signal?.addEventListener("abort", onAbort, { once: true });
+			const processingStream = signal ? streamForProcessing.pipeThrough(new TransformStream(), { signal }) : streamForProcessing;
 			this.processChatResponse_vNext({
-				stream: streamForProcessing,
+				stream: processingStream,
 				update: ({ message }) => {
 					const existingIndex = messages.findIndex((m) => m.id === message.id);
 					if (existingIndex !== -1) messages[existingIndex] = message;
 					else messages.push(message);
 				},
 				onFinish: async ({ finishReason, message }) => {
+					if (isAborted()) return;
 					if (finishReason === "tool-calls") {
 						const toolInvocationsById = /* @__PURE__ */ new Map();
 						for (const part of message?.parts ?? []) {
@@ -1975,6 +2084,7 @@ var Agent = class extends BaseResource {
 							const syntheticChunks = [];
 							const toolResultContents = [];
 							for (const toolCall of executableToolCalls) {
+								if (isAborted()) return;
 								const clientTool = processedParams.clientTools?.[toolCall.toolName];
 								const runId = streamRunId ?? toolCall.toolCallId;
 								let result;
@@ -2042,6 +2152,7 @@ var Agent = class extends BaseResource {
 							try {
 								await pipePromise;
 							} catch {}
+							if (isAborted()) return;
 							for (const synthetic of syntheticChunks) try {
 								const errorForSerialization = synthetic.type === "tool-error" ? synthetic.payload.error : void 0;
 								const serializedError = errorForSerialization instanceof Error ? {
@@ -2057,7 +2168,7 @@ var Agent = class extends BaseResource {
 									}
 								} : synthetic;
 								const sseLine = `data: ${JSON.stringify(payloadForWire)}\n\n`;
-								controller.enqueue(new TextEncoder().encode(sseLine));
+								safeEnqueue(new TextEncoder().encode(sseLine));
 							} catch (enqueueErr) {
 								console.error("Failed to enqueue synthetic tool-result chunk:", enqueueErr);
 							}
@@ -2071,22 +2182,23 @@ var Agent = class extends BaseResource {
 								toolResultMessage
 							];
 							const recursionRoute = route === "resume-stream" ? "stream" : route === "resume-stream-until-idle" ? "stream-until-idle" : route;
+							if (isAborted()) return;
 							try {
 								await this.processStreamResponse({
 									...processedParams,
 									messages: updatedMessages,
 									clientTools: processClientTools(processedParams.clientToolsResolver?.() ?? processedParams.clientTools)
-								}, controller, recursionRoute);
+								}, controller, recursionRoute, signal);
 							} catch (error) {
-								console.error("Error processing recursive stream response:", error);
+								if (!isAbortError(error)) console.error("Error processing recursive stream response:", error);
 							}
 						} else {
 							await pipePromise;
-							controller.close();
+							safeClose();
 						}
 					} else {
 						await pipePromise;
-						controller.close();
+						safeClose();
 					}
 				},
 				onStreamChunk: (chunk) => {
@@ -2094,14 +2206,16 @@ var Agent = class extends BaseResource {
 				},
 				lastMessage: void 0
 			}).catch(async (error) => {
-				console.error("Error processing stream response:", error);
+				if (!isAbortError(error)) console.error("Error processing stream response:", error);
 				try {
 					await pipePromise;
-					controller.close();
 				} catch {}
+				safeClose();
+			}).finally(() => {
+				signal?.removeEventListener("abort", onAbort);
 			});
 		} catch (error) {
-			console.error("Error processing stream response:", error);
+			if (!isAbortError(error)) console.error("Error processing stream response:", error);
 		}
 		return response;
 	}
@@ -2198,11 +2312,8 @@ var Agent = class extends BaseResource {
 			clientTools: processClientTools(params.clientToolsResolver?.() ?? params.clientTools),
 			structuredOutput
 		};
-		let readableController;
-		const readable = new ReadableStream({ start(controller) {
-			readableController = controller;
-		} });
-		const response = await this.processStreamResponse(processedParams, readableController);
+		const { readable, controller: readableController, signal } = this.createCancellableStream(params.abortSignal);
+		const response = await this.processStreamResponse(processedParams, readableController, "stream", signal);
 		const streamResponse = new Response(readable, {
 			status: response.status,
 			statusText: response.statusText,
@@ -2232,11 +2343,8 @@ var Agent = class extends BaseResource {
 			clientTools: processClientTools(params.clientToolsResolver?.() ?? params.clientTools),
 			structuredOutput
 		};
-		let readableController;
-		const readable = new ReadableStream({ start(controller) {
-			readableController = controller;
-		} });
-		const response = await this.processStreamResponse(processedParams, readableController, "stream-until-idle");
+		const { readable, controller: readableController, signal } = this.createCancellableStream(params.abortSignal);
+		const response = await this.processStreamResponse(processedParams, readableController, "stream-until-idle", signal);
 		const streamResponse = new Response(readable, {
 			status: response.status,
 			statusText: response.statusText,
@@ -2276,11 +2384,8 @@ var Agent = class extends BaseResource {
 			...rest,
 			requestContext: parseClientRequestContext(requestContext)
 		};
-		let readableController;
-		const readable = new ReadableStream({ start(controller) {
-			readableController = controller;
-		} });
-		const response = await this.processStreamResponse(processedParams, readableController, "approve-tool-call");
+		const { readable, controller: readableController, signal } = this.createCancellableStream(params.abortSignal);
+		const response = await this.processStreamResponse(processedParams, readableController, "approve-tool-call", signal);
 		const streamResponse = new Response(readable, {
 			status: response.status,
 			statusText: response.statusText,
@@ -2310,11 +2415,8 @@ var Agent = class extends BaseResource {
 			...rest,
 			requestContext: parseClientRequestContext(requestContext)
 		};
-		let readableController;
-		const readable = new ReadableStream({ start(controller) {
-			readableController = controller;
-		} });
-		const response = await this.processStreamResponse(processedParams, readableController, "decline-tool-call");
+		const { readable, controller: readableController, signal } = this.createCancellableStream(params.abortSignal);
+		const response = await this.processStreamResponse(processedParams, readableController, "decline-tool-call", signal);
 		const streamResponse = new Response(readable, {
 			status: response.status,
 			statusText: response.statusText,
@@ -2384,11 +2486,8 @@ var Agent = class extends BaseResource {
 				schema: standardSchemaToJSONSchema(toStandardSchema(options.structuredOutput.schema))
 			} : void 0
 		};
-		let readableController;
-		const readable = new ReadableStream({ start(controller) {
-			readableController = controller;
-		} });
-		const response = await this.processStreamResponse(processedParams, readableController, "resume-stream");
+		const { readable, controller: readableController, signal } = this.createCancellableStream(options.abortSignal);
+		const response = await this.processStreamResponse(processedParams, readableController, "resume-stream", signal);
 		const streamResponse = new Response(readable, {
 			status: response.status,
 			statusText: response.statusText,
@@ -2447,11 +2546,8 @@ var Agent = class extends BaseResource {
 				schema: standardSchemaToJSONSchema(toStandardSchema(options.structuredOutput.schema))
 			} : void 0
 		};
-		let readableController;
-		const readable = new ReadableStream({ start(controller) {
-			readableController = controller;
-		} });
-		const response = await this.processStreamResponse(processedParams, readableController, "resume-stream-until-idle");
+		const { readable, controller: readableController, signal } = this.createCancellableStream(options.abortSignal);
+		const response = await this.processStreamResponse(processedParams, readableController, "resume-stream-until-idle", signal);
 		const streamResponse = new Response(readable, {
 			status: response.status,
 			statusText: response.statusText,
@@ -2496,32 +2592,41 @@ var Agent = class extends BaseResource {
 	/**
 	* Processes the stream response and handles tool calls
 	*/
-	async processStreamResponseLegacy(processedParams, writable) {
+	async processStreamResponseLegacy(processedParams, writable, signal) {
 		const { memory } = processedParams ?? {};
 		const { resource, thread } = memory ?? {};
 		const threadId = processedParams.threadId ?? (typeof thread === "string" ? thread : thread?.id);
 		const resourceId = processedParams.resourceId ?? resource;
+		const { abortSignal: _abortSignal, ...bodyParams } = processedParams ?? {};
+		const isAborted = () => signal?.aborted === true;
+		const isAbortError = (error) => isAborted() || error?.name === "AbortError";
 		const response = await this.request(this.routeOverrides?.stream ?? `/agents/${this.agentId}/stream-legacy`, {
 			method: "POST",
-			body: processedParams,
-			stream: true
+			body: bodyParams,
+			stream: true,
+			signal
 		});
 		if (!response.body) throw new Error("No response body");
 		try {
 			let toolCalls = [];
 			let messages = [];
 			const [streamForWritable, streamForProcessing] = response.body.tee();
-			streamForWritable.pipeTo(writable, { preventClose: true }).catch((error) => {
-				console.error("Error piping to writable stream:", error);
+			streamForWritable.pipeTo(writable, {
+				preventClose: true,
+				signal
+			}).catch((error) => {
+				if (!isAbortError(error)) console.error("Error piping to writable stream:", error);
 			});
+			const processingStream = signal ? streamForProcessing.pipeThrough(new TransformStream(), { signal }) : streamForProcessing;
 			this.processChatResponse({
-				stream: streamForProcessing,
+				stream: processingStream,
 				update: ({ message }) => {
 					const existingIndex = messages.findIndex((m) => m.id === message.id);
 					if (existingIndex !== -1) messages[existingIndex] = message;
 					else messages.push(message);
 				},
 				onFinish: async ({ finishReason, message }) => {
+					if (isAborted()) return;
 					if (finishReason === "tool-calls") {
 						const toolCall = [...message?.parts ?? []].reverse().find((part) => part.type === "tool-invocation")?.toolInvocation;
 						if (toolCall) {
@@ -2553,6 +2658,7 @@ var Agent = class extends BaseResource {
 										}
 									}
 								});
+								if (isAborted()) return;
 								const writer = writable.getWriter();
 								try {
 									await writer.write(new TextEncoder().encode("a:" + JSON.stringify({
@@ -2582,21 +2688,21 @@ var Agent = class extends BaseResource {
 									...processedParams,
 									clientTools: processClientTools(processedParams.clientToolsResolver?.() ?? processedParams.clientTools),
 									messages: updatedMessages
-								}, writable).catch((error) => {
-									console.error("Error processing stream response:", error);
+								}, writable, signal).catch((error) => {
+									if (!isAbortError(error)) console.error("Error processing stream response:", error);
 								});
 							}
 						}
 					} else setTimeout(() => {
-						writable.close();
+						writable.close().catch(() => {});
 					}, 0);
 				},
 				lastMessage: void 0
 			}).catch((error) => {
-				console.error("Error processing stream response:", error);
+				if (!isAbortError(error)) console.error("Error processing stream response:", error);
 			});
 		} catch (error) {
-			console.error("Error processing stream response:", error);
+			if (!isAbortError(error)) console.error("Error processing stream response:", error);
 		}
 		return response;
 	}
@@ -4389,6 +4495,24 @@ var Observability = class extends BaseResource {
 			body: params
 		});
 	}
+	/** Returns canonical and observed fields available to the advanced trace-query grammar. */
+	getTraceQueryFields(params, options) {
+		return this.request("/observability/traces/query/fields", {
+			method: "POST",
+			body: params,
+			retries: 0,
+			signal: options?.signal
+		});
+	}
+	/** Returns bounded string suggestions for one eligible trace-query field. */
+	getTraceQueryValues(params, options) {
+		return this.request("/observability/traces/query/values", {
+			method: "POST",
+			body: params,
+			retries: 0,
+			signal: options?.signal
+		});
+	}
 	/**
 	* Queries thread identities using eligible-trace and cross-trace predicates.
 	*
@@ -6967,6 +7091,14 @@ var MastraClient = class extends BaseResource {
 	queryTraces(params) {
 		return this.observability.queryTraces(params);
 	}
+	/** Returns canonical and observed fields available to the advanced trace-query grammar. */
+	getTraceQueryFields(params, options) {
+		return this.observability.getTraceQueryFields(params, options);
+	}
+	/** Returns bounded string suggestions for one eligible trace-query field. */
+	getTraceQueryValues(params, options) {
+		return this.observability.getTraceQueryValues(params, options);
+	}
 	/** Queries thread identities using eligible-trace and cross-trace predicates. */
 	queryTraceThreads(params) {
 		return this.observability.queryTraceThreads(params);