@tanstack/ai-client 0.26.0 → 0.28.0

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

@@ -94,6 +94,8 @@ export interface RunAgentInputContext {
     }>;
     /** Arbitrary user-controlled passthrough data. */
     forwardedProps?: Record<string, unknown>;
+    /** Extra request headers for this run (e.g. BYOK keys). POST only. */
+    headers?: Record<string, string>;
 }
 export interface ConnectConnectionAdapter {
     /**

package/dist/esm/connection-adapters.js

@@ -1,6 +1,7 @@
 import { createResponseStreamTextDecoder, getResponseStreamReader } from "./response-stream.js";
 import { parseSseDataLine } from "./sse-utils.js";
 import { EventType, getChunkRunId, restoreInboundChunk, tanstackMetadata, uiMessagesToWire, withTanstackMetadata } from "@tanstack/ai/client";
+import { ByokMissingError, isByokMissingBody } from "@tanstack/ai/byok";
 //#region src/connection-adapters.ts
 /**
 * Associates connect-wrapped chunks with the run they were produced under.
@@ -287,8 +288,25 @@ async function* linesToNdjsonEvents(lines) {
 		else yield { chunk: restoreInboundUsage(parsed) };
 	}
 }
-function assertResponseOk(response) {
-	if (!response.ok) throw new Error(`HTTP error! status: ${response.status} ${response.statusText}`);
+async function assertResponseOk(response) {
+	if (response.ok) return;
+	if (response.status === 401) {
+		const body = await response.clone().json().catch(() => null);
+		if (isByokMissingBody(body)) throw new ByokMissingError(body.error.provider);
+	}
+	throw new Error(`HTTP error! status: ${response.status} ${response.statusText}`);
+}
+function errorFromXhrStatus(xhr) {
+	if (xhr.status === 401) {
+		let parsed = null;
+		try {
+			parsed = JSON.parse(xhr.responseText);
+		} catch {
+			parsed = null;
+		}
+		if (isByokMissingBody(parsed)) return new ByokMissingError(parsed.error.provider);
+	}
+	return /* @__PURE__ */ new Error(`XHR error! status: ${xhr.status} ${xhr.statusText}`);
 }
 /**
 * GET the hydration endpoint for a thread and parse its JSON `{ messages,
@@ -306,7 +324,7 @@ async function fetchThreadHydration(fetchClient, url, headers, credentials, thre
 		},
 		credentials
 	});
-	assertResponseOk(response);
+	await assertResponseOk(response);
 	const data = await response.json();
 	const activeRun = data.activeRun && typeof data.activeRun.runId === "string" ? { runId: data.activeRun.runId } : null;
 	const interrupts = data.interrupts && typeof data.interrupts.runId === "string" && Array.isArray(data.interrupts.pending) && data.interrupts.pending.length > 0 ? {
@@ -335,7 +353,7 @@ async function fetchGenerationHydration(fetchClient, url, headers, credentials,
 		},
 		credentials
 	});
-	assertResponseOk(response);
+	await assertResponseOk(response);
 	const raw = await response.json();
 	if (raw === null) return {
 		resumeSnapshot: null,
@@ -351,12 +369,12 @@ async function fetchGenerationHydration(fetchClient, url, headers, credentials,
 }
 /** Yield SSE stream events (chunk + offset) from a fetch Response body. */
 async function* responseToSSEEvents(response, abortSignal, fallbackIds) {
-	assertResponseOk(response);
+	await assertResponseOk(response);
 	yield* linesToSSEEvents(readStreamLines(getResponseStreamReader(response), abortSignal), fallbackIds);
 }
 /** Yield NDJSON stream events (chunk + offset) from a fetch Response body. */
 async function* responseToNdjsonEvents(response, abortSignal) {
-	assertResponseOk(response);
+	await assertResponseOk(response);
 	yield* linesToNdjsonEvents(readStreamLines(getResponseStreamReader(response), abortSignal));
 }
 async function* responseToSSEChunks(response, abortSignal) {
@@ -588,6 +606,7 @@ function fetchServerSentEvents(url, options = {}) {
 			const requestHeaders = {
 				"Content-Type": "application/json",
 				...mergeHeaders(resolvedOptions.headers),
+				...mergeHeaders(runContext?.headers),
 				...runIdHeader(runContext?.runId)
 			};
 			const requestBody = buildRunAgentInputBody(messages, data, runContext, resolvedOptions);
@@ -673,6 +692,7 @@ function fetchHttpStream(url, options = {}) {
 			const requestHeaders = {
 				"Content-Type": "application/json",
 				...mergeHeaders(resolvedOptions.headers),
+				...mergeHeaders(runContext?.headers),
 				...runIdHeader(runContext?.runId)
 			};
 			const requestBody = buildRunAgentInputBody(messages, data, runContext, resolvedOptions);
@@ -739,7 +759,7 @@ function readXhrLines(xhr, abortSignal) {
 	};
 	const enqueueDelta = () => {
 		if (xhr.status !== 0 && (xhr.status < 200 || xhr.status >= 300)) {
-			error = /* @__PURE__ */ new Error(`XHR error! status: ${xhr.status} ${xhr.statusText}`);
+			error = errorFromXhrStatus(xhr);
 			done = true;
 			return;
 		}
@@ -756,7 +776,7 @@ function readXhrLines(xhr, abortSignal) {
 	};
 	const finish = () => {
 		enqueueDelta();
-		if (xhr.status !== 0 && (xhr.status < 200 || xhr.status >= 300)) error = /* @__PURE__ */ new Error(`XHR error! status: ${xhr.status} ${xhr.statusText}`);
+		if (xhr.status !== 0 && (xhr.status < 200 || xhr.status >= 300)) error = errorFromXhrStatus(xhr);
 		else if (buffer.trim() && !aborted) error = new StreamTruncatedError();
 		done = true;
 		wake();
@@ -813,6 +833,7 @@ function createConfiguredXhrRequest(url, options, messages, data, runContext, me
 	const requestHeaders = {
 		"Content-Type": "application/json",
 		...mergeHeaders(options.headers),
+		...mergeHeaders(method === "POST" ? runContext?.headers : void 0),
 		...method === "POST" ? runIdHeader(runContext?.runId) : {},
 		...extraHeaders
 	};
@@ -1211,7 +1232,10 @@ function fetcherToConnectionAdapter(fetcher) {
 			runId: runContext.runId,
 			...runContext.parentRunId !== void 0 ? { parentRunId: runContext.parentRunId } : {},
 			...runContext.resume !== void 0 ? { resume: runContext.resume } : {}
-		}, { signal: abortSignal });
+		}, {
+			signal: abortSignal,
+			headers: runContext.headers
+		});
 		if (result instanceof Response) yield* responseToSSEChunks(result, abortSignal);
 		else yield* abortableIterable(result, abortSignal);
 	} };