@tanstack/ai-client 0.31.0 → 0.32.0

package/dist/esm/connection-adapters.d.ts

@@ -130,7 +130,7 @@ export interface ConnectConnectionAdapter {
      * `fetchServerSentEvents` / `fetchHttpStream`, and on `stream()` /
      * `rpcStream()` when supplied via {@link StreamConnectionHandlers}.
      */
-    hydrate?: (threadId: string) => Promise<ChatHydrationResult>;
+    hydrate?: (threadId: string, options?: ChatHydrateOptions) => Promise<ChatHydrationResult>;
 }
 /**
  * Server-resolved hydration for a generation thread. `resumeSnapshot` is the
@@ -166,11 +166,22 @@ export interface GenerationHydrationResult {
         runId: string;
     } | null;
 }
+/**
+ * Optional paging for a chat hydrate GET.
+ * Omit both fields to load the full thread.
+ */
+export interface ChatHydrateOptions {
+    limit?: number;
+    before?: string;
+}
 /**
  * Server-resolved hydration for a thread. `messages` is the stored transcript;
  * `activeRun` is a cursor to a run still generating for the thread (or `null`).
  * Keyed on the STABLE thread id — the client never handles a run id, so a turn
  * that spans several runs (interrupt/tool continuations) reconnects correctly.
+ *
+ * `page` is additive. Old servers omit it; treat that as a full list
+ * (`truncated: false`).
  */
 export interface ChatHydrationResult {
     messages: Array<UIMessage>;
@@ -186,6 +197,16 @@ export interface ChatHydrationResult {
         runId: string;
         pending: Array<ChatPendingInterrupt>;
     } | null;
+    /**
+     * Whether this window is a slice of a longer thread.
+     * `cursor` is set when `truncated` is true and is sent back as `before`.
+     */
+    page?: {
+        truncated: false;
+    } | {
+        truncated: true;
+        cursor: string;
+    };
 }
 /**
  * A {@link ConnectConnectionAdapter} that also supports joining an existing run
@@ -205,7 +226,7 @@ export interface ResumableConnectConnectionAdapter extends ConnectConnectionAdap
      * JSON GET (`?threadId`), so it is transport-agnostic regardless of how the
      * delivery stream is served.
      */
-    hydrate?: (threadId: string) => Promise<ChatHydrationResult>;
+    hydrate?: (threadId: string, options?: ChatHydrateOptions) => Promise<ChatHydrationResult>;
 }
 export interface SubscribeConnectionAdapter {
     /**
@@ -228,7 +249,7 @@ export interface SubscribeConnectionAdapter {
      * cursor). Present only when the underlying connection supports it. The client
      * calls it on mount to re-hydrate without any app-side loader or prop.
      */
-    hydrate?: (threadId: string) => Promise<ChatHydrationResult>;
+    hydrate?: (threadId: string, options?: ChatHydrateOptions) => Promise<ChatHydrationResult>;
 }
 /**
  * Connection adapter union.
@@ -394,7 +415,7 @@ export interface StreamConnectionHandlers {
      * Server-driven chat hydration for `persistence: true`: the stored
      * transcript for `threadId` plus a cursor to an in-flight run.
      */
-    hydrate?: (threadId: string) => Promise<ChatHydrationResult>;
+    hydrate?: (threadId: string, options?: ChatHydrateOptions) => Promise<ChatHydrationResult>;
     /**
      * Server-driven generation hydration for `persistence: true`: the last
      * generation's resume snapshot for `threadId` plus a cursor to a run still

package/dist/esm/connection-adapters.js

@@ -309,15 +309,33 @@ function errorFromXhrStatus(xhr) {
 	}
 	return /* @__PURE__ */ new Error(`XHR error! status: ${xhr.status} ${xhr.statusText}`);
 }
+function chatHydrationSearchParams(threadId, options) {
+	const values = { threadId };
+	if (options?.limit !== void 0) values.limit = String(options.limit);
+	if (options?.before !== void 0) values.before = options.before;
+	return values;
+}
+function readChatHydrationPage(value) {
+	if (value === null || typeof value !== "object") return { truncated: false };
+	if (!("truncated" in value) || value.truncated !== true) return { truncated: false };
+	if (!("cursor" in value) || typeof value.cursor !== "string" || value.cursor === "") return { truncated: false };
+	return {
+		truncated: true,
+		cursor: value.cursor
+	};
+}
 /**
 * GET the hydration endpoint for a thread and parse its JSON `{ messages,
 * activeRun }` body. This is the transport-agnostic reconnect probe: keyed on
 * the STABLE thread id, it returns the stored transcript and — if a run is still
 * generating — a cursor the caller tails via `joinRun`. Shared by every fetch/
 * XHR adapter so the client never has to know which transport is in use.
+*
+* Pass `limit` / `before` to page. Old servers that omit `page` are treated as
+* a full list (`truncated: false`).
 */
-async function fetchThreadHydration(fetchClient, url, headers, credentials, threadId) {
-	const response = await fetchClient(withSearchParams(url, { threadId }), {
+async function fetchThreadHydration(fetchClient, url, headers, credentials, threadId, options) {
+	const response = await fetchClient(withSearchParams(url, chatHydrationSearchParams(threadId, options)), {
 		method: "GET",
 		headers: {
 			Accept: "application/json",
@@ -335,7 +353,8 @@ async function fetchThreadHydration(fetchClient, url, headers, credentials, thre
 	return {
 		messages: Array.isArray(data.messages) ? normalizeMessagesDates(data.messages) : [],
 		activeRun,
-		interrupts
+		interrupts,
+		page: readChatHydrationPage(data.page)
 	};
 }
 /**
@@ -474,6 +493,7 @@ function normalizeConnectionAdapter(connection) {
 	if (!hasConnect) throw new Error("Connection adapter must provide either connect or both subscribe and send");
 	let activeBuffer = [];
 	let activeWaiters = [];
+	let activeSubscriber;
 	function push(chunk, runId) {
 		if (runId) chunkRunIds.set(chunk, runId);
 		const waiter = activeWaiters.shift();
@@ -481,17 +501,14 @@ function normalizeConnectionAdapter(connection) {
 		else activeBuffer.push(chunk);
 	}
 	async function waitUntilSubscriberIdle(abortSignal) {
-		const idle = () => activeBuffer.length === 0 && (activeWaiters.length > 0 || abortSignal?.aborted);
+		const idle = () => activeSubscriber !== activeWaiters || activeBuffer.length === 0 && (activeWaiters.length > 0 || abortSignal?.aborted);
 		for (let i = 0; i < 16 && !abortSignal?.aborted; i++) {
 			if (idle()) return;
 			await Promise.resolve();
 		}
-		let macrotaskWaits = 0;
 		while (!abortSignal?.aborted) {
 			if (idle()) return;
 			await new Promise((resolve) => setTimeout(resolve, 0));
-			macrotaskWaits++;
-			if (activeWaiters.length === 0 && macrotaskWaits >= 32) return;
 		}
 	}
 	return {
@@ -501,19 +518,24 @@ function normalizeConnectionAdapter(connection) {
 			activeBuffer = myBuffer;
 			activeWaiters = myWaiters;
 			return (async function* () {
-				while (!abortSignal?.aborted) {
-					let chunk;
-					const buffered = myBuffer.shift();
-					if (buffered !== void 0) chunk = buffered;
-					else chunk = await new Promise((resolve) => {
-						const onAbort = () => resolve(null);
-						myWaiters.push((c) => {
-							abortSignal?.removeEventListener("abort", onAbort);
-							resolve(c);
+				activeSubscriber = myWaiters;
+				try {
+					while (!abortSignal?.aborted) {
+						let chunk;
+						const buffered = myBuffer.shift();
+						if (buffered !== void 0) chunk = buffered;
+						else chunk = await new Promise((resolve) => {
+							const onAbort = () => resolve(null);
+							myWaiters.push((c) => {
+								abortSignal?.removeEventListener("abort", onAbort);
+								resolve(c);
+							});
+							abortSignal?.addEventListener("abort", onAbort, { once: true });
 						});
-						abortSignal?.addEventListener("abort", onAbort, { once: true });
-					});
-					if (chunk !== null) yield chunk;
+						if (chunk !== null) yield chunk;
+					}
+				} finally {
+					if (activeSubscriber === myWaiters) activeSubscriber = void 0;
 				}
 			})();
 		},
@@ -556,7 +578,7 @@ function normalizeConnectionAdapter(connection) {
 		...typeof connection.joinRun === "function" ? { joinRun: (runId, abortSignal) => connection.joinRun(runId, abortSignal) } : {},
 		...(() => {
 			const hydrate = connection.hydrate;
-			return typeof hydrate === "function" ? { hydrate: (threadId) => hydrate(threadId) } : {};
+			return typeof hydrate === "function" ? { hydrate: (threadId, options) => hydrate(threadId, options) } : {};
 		})()
 	};
 }
@@ -654,10 +676,10 @@ function fetchServerSentEvents(url, options = {}) {
 				credentials: resolvedOptions.credentials || "same-origin"
 			}, (response, sseSignal) => responseToSSEEvents(response, sseSignal, { runId })), signal, resolvedOptions.reconnect);
 		},
-		async hydrate(threadId) {
+		async hydrate(threadId, hydrateOptions) {
 			const resolvedUrl = typeof url === "function" ? url() : url;
 			const resolvedOptions = typeof options === "function" ? await options() : options;
-			return fetchThreadHydration(resolvedOptions.fetchClient ?? fetch, resolvedUrl, mergeHeaders(resolvedOptions.headers), resolvedOptions.credentials || "same-origin", threadId);
+			return fetchThreadHydration(resolvedOptions.fetchClient ?? fetch, resolvedUrl, mergeHeaders(resolvedOptions.headers), resolvedOptions.credentials || "same-origin", threadId, hydrateOptions);
 		},
 		async hydrateGeneration(threadId) {
 			const resolvedUrl = typeof url === "function" ? url() : url;
@@ -737,10 +759,10 @@ function fetchHttpStream(url, options = {}) {
 				credentials: resolvedOptions.credentials || "same-origin"
 			}, responseToNdjsonEvents), signal, resolvedOptions.reconnect);
 		},
-		async hydrate(threadId) {
+		async hydrate(threadId, hydrateOptions) {
 			const resolvedUrl = typeof url === "function" ? url() : url;
 			const resolvedOptions = typeof options === "function" ? await options() : options;
-			return fetchThreadHydration(resolvedOptions.fetchClient ?? fetch, resolvedUrl, mergeHeaders(resolvedOptions.headers), resolvedOptions.credentials || "same-origin", threadId);
+			return fetchThreadHydration(resolvedOptions.fetchClient ?? fetch, resolvedUrl, mergeHeaders(resolvedOptions.headers), resolvedOptions.credentials || "same-origin", threadId, hydrateOptions);
 		},
 		async hydrateGeneration(threadId) {
 			const resolvedUrl = typeof url === "function" ? url() : url;
@@ -918,10 +940,10 @@ function xhrServerSentEvents(url, options = {}) {
 				runId
 			}), resolvedOptions, "GET", [], void 0, void 0, (lines) => linesToSSEEvents(lines, { runId })), signal, resolvedOptions.reconnect);
 		},
-		async hydrate(threadId) {
+		async hydrate(threadId, hydrateOptions) {
 			const resolvedUrl = typeof url === "function" ? url() : url;
 			const resolvedOptions = await resolveXhrConnectionOptions(options);
-			return fetchThreadHydration(fetch, resolvedUrl, mergeHeaders(resolvedOptions.headers), resolvedOptions.withCredentials ? "include" : "same-origin", threadId);
+			return fetchThreadHydration(fetch, resolvedUrl, mergeHeaders(resolvedOptions.headers), resolvedOptions.withCredentials ? "include" : "same-origin", threadId, hydrateOptions);
 		},
 		async hydrateGeneration(threadId) {
 			const resolvedUrl = typeof url === "function" ? url() : url;
@@ -955,10 +977,10 @@ function xhrHttpStream(url, options = {}) {
 				runId
 			}), resolvedOptions, "GET", [], void 0, void 0, linesToNdjsonEvents), signal, resolvedOptions.reconnect);
 		},
-		async hydrate(threadId) {
+		async hydrate(threadId, hydrateOptions) {
 			const resolvedUrl = typeof url === "function" ? url() : url;
 			const resolvedOptions = await resolveXhrConnectionOptions(options);
-			return fetchThreadHydration(fetch, resolvedUrl, mergeHeaders(resolvedOptions.headers), resolvedOptions.withCredentials ? "include" : "same-origin", threadId);
+			return fetchThreadHydration(fetch, resolvedUrl, mergeHeaders(resolvedOptions.headers), resolvedOptions.withCredentials ? "include" : "same-origin", threadId, hydrateOptions);
 		},
 		async hydrateGeneration(threadId) {
 			const resolvedUrl = typeof url === "function" ? url() : url;