@tanstack/ai-client 0.31.1 → 0.32.1

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)
 	};
 }
 /**
@@ -559,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) } : {};
 		})()
 	};
 }
@@ -657,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;
@@ -740,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;
@@ -921,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;
@@ -958,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;