@mastra/client-js 1.56.0-alpha.2 → 1.56.0-alpha.4

package/dist/index.js

@@ -376,6 +376,16 @@ function isTerminalHttpError(error) {
 	return typeof status === "number" && (status >= 400 && status < 500 || status === 501);
 }
 /**
+* Whether an error is the server's trace query execution timeout (504 with code
+* `TRACE_QUERY_EXECUTION_TIMEOUT`). Re-sending the same query hits the same
+* timeout budget, so retrying only repeats an expensive database query.
+* Plain 504s from proxies or gateways do not carry this code and stay retryable.
+*/
+function isTraceQueryTimeoutError(error) {
+	const { status, body } = error ?? {};
+	return status === 504 && body?.code === "TRACE_QUERY_EXECUTION_TIMEOUT";
+}
+/**
 * Whether a failed resubscribe should end a long-lived subscription instead of
 * retrying. Like {@link isTerminalHttpError}, but 408 Request Timeout and
 * 429 Too Many Requests stay retryable because they mean "try again later".
@@ -1194,7 +1204,7 @@ var BaseResource = class {
 			} catch (error) {
 				lastError = error;
 				if (signal?.aborted || error?.name === "AbortError") throw error;
-				if (isTerminalHttpError(error)) throw error;
+				if (isTerminalHttpError(error) || isTraceQueryTimeoutError(error)) throw error;
 				if (attempt === retries) break;
 				await new Promise((resolve, reject) => {
 					const timer = setTimeout(() => {
@@ -5126,10 +5136,10 @@ var Observability = class extends BaseResource {
 		});
 	}
 	/**
-	* Queries thread identities using eligible-trace and cross-trace predicates.
+	* Queries full-history thread rows using a trace or thread activity window.
 	*
-	* @param params - Thread query with its eligible trace selection
-	* @returns Matching thread identities
+	* @param params - Thread query with its qualification window, filters and pagination
+	* @returns Matching thread rows. Legacy requests retain their identity-only response.
 	*/
 	queryTraceThreads(params) {
 		return this.request("/observability/threads/query", {
@@ -6781,10 +6791,12 @@ var AgentControllerSession = class extends BaseResource {
 			delayMs: finiteOr(options.reconnect.delayMs, 1e3),
 			maxDelayMs: finiteOr(options.reconnect.maxDelayMs, 3e4)
 		} : null;
+		const idleTimeoutMs = finiteOr(options.idleTimeoutMs, 6e4);
 		/** Exponential backoff for the Nth (1-based) retry of an outage streak. */
 		const backoffDelay = (attempt) => reconnectOptions ? Math.min(reconnectOptions.delayMs * 2 ** (attempt - 1), reconnectOptions.maxDelayMs) : 0;
 		let cancelled = false;
 		let currentReader = null;
+		let currentAbort = null;
 		let reconnectTimer;
 		let delayResolve;
 		const settleDelay = () => {
@@ -6800,8 +6812,39 @@ var AgentControllerSession = class extends BaseResource {
 			delayResolve = resolve;
 			reconnectTimer = setTimeout(settleDelay, ms);
 		});
+		const idleTimeoutError = () => {
+			const error = /* @__PURE__ */ new Error(`Agent controller session stream received no data for ${idleTimeoutMs}ms`);
+			error.name = "AgentControllerStreamIdleError";
+			return error;
+		};
 		const requestStream = async () => {
-			const response = await this.request(this.url(`${this.base()}/stream`), { stream: true });
+			const abort = new AbortController();
+			currentAbort = abort;
+			let openTimer;
+			const request = this.request(this.url(`${this.base()}/stream`), {
+				stream: true,
+				signal: abort.signal
+			});
+			request.then((response) => {
+				if (abort.signal.aborted) response.body?.cancel().catch(() => {});
+			}).catch(() => {});
+			let response;
+			try {
+				if (idleTimeoutMs > 0) response = await Promise.race([request, new Promise((_resolve, reject) => {
+					openTimer = setTimeout(() => {
+						const error = idleTimeoutError();
+						abort.abort(error);
+						reject(error);
+					}, idleTimeoutMs);
+				})]);
+				else response = await request;
+			} finally {
+				clearTimeout(openTimer);
+			}
+			if (abort.signal.aborted) {
+				response.body?.cancel().catch(() => {});
+				throw abort.signal.reason;
+			}
 			if (!response.body) throw new Error("No response body for agent controller session stream");
 			return response;
 		};
@@ -6827,12 +6870,40 @@ var AgentControllerSession = class extends BaseResource {
 		const pump = async (response) => {
 			const reader = response.body.getReader();
 			currentReader = reader;
+			const abort = currentAbort;
 			const decoder = new TextDecoder();
 			let buffer = "";
+			let idleTimer;
+			let timedOut = false;
+			const armIdleTimer = () => {
+				if (idleTimeoutMs <= 0) return;
+				clearTimeout(idleTimer);
+				idleTimer = setTimeout(() => {
+					timedOut = true;
+					reader.cancel(idleTimeoutError()).catch(() => {});
+					abort?.abort(idleTimeoutError());
+				}, idleTimeoutMs);
+			};
 			try {
+				armIdleTimer();
 				while (!cancelled) {
-					const { done, value } = await reader.read();
+					let chunk;
+					try {
+						chunk = await reader.read();
+					} catch (error) {
+						if (timedOut && !cancelled) return {
+							kind: "transport_error",
+							error: idleTimeoutError()
+						};
+						throw error;
+					}
+					if (timedOut && !cancelled) return {
+						kind: "transport_error",
+						error: idleTimeoutError()
+					};
+					const { done, value } = chunk;
 					if (done) return cancelled ? { kind: "cancelled" } : { kind: "done" };
+					armIdleTimer();
 					if (cancelled) return { kind: "cancelled" };
 					buffer += decoder.decode(value, { stream: true });
 					let separator;
@@ -6866,7 +6937,9 @@ var AgentControllerSession = class extends BaseResource {
 					error
 				};
 			} finally {
+				clearTimeout(idleTimer);
 				if (currentReader === reader) currentReader = null;
+				if (currentAbort === abort) currentAbort = null;
 				reader.cancel().catch(() => {});
 			}
 		};
@@ -6943,6 +7016,7 @@ var AgentControllerSession = class extends BaseResource {
 			cancelled = true;
 			settleDelay();
 			currentReader?.cancel().catch(() => {});
+			currentAbort?.abort();
 		} };
 	}
 	/**
@@ -6982,6 +7056,7 @@ var AgentControllerSession = class extends BaseResource {
 			body: {
 				toolCallId,
 				approved,
+				...options?.alwaysAllowCategory ? { alwaysAllowCategory: true } : {},
 				...requestContext ? { requestContext } : {}
 			}
 		});
@@ -7848,7 +7923,7 @@ var MastraClient = class extends BaseResource {
 	getTraceQueryValues(params, options) {
 		return this.observability.getTraceQueryValues(params, options);
 	}
-	/** Queries thread identities using eligible-trace and cross-trace predicates. */
+	/** Queries full-history thread rows using a trace or thread activity window. */
 	queryTraceThreads(params) {
 		return this.observability.queryTraceThreads(params);
 	}