@tanstack/ai 0.47.3 → 0.49.1

package/dist/esm/activities/summarize/chat-stream-summarize.d.ts

@@ -1,3 +1,4 @@
+import { AdapterYieldChunk } from '../../utilities/adapter-yield-chunk.js';
 import { BaseSummarizeAdapter } from './adapter.js';
 import { StreamChunk, SummarizationOptions, SummarizationResult, TextOptions } from '../../types.js';
 /**
@@ -12,7 +13,7 @@ import { StreamChunk, SummarizationOptions, SummarizationResult, TextOptions } f
  * `SummarizationOptions<TProviderOptions>` on the wrapper itself.
  */
 export interface ChatStreamCapable {
-    chatStream: (options: TextOptions<any>) => AsyncIterable<StreamChunk>;
+    chatStream: (options: TextOptions<any>) => AsyncIterable<AdapterYieldChunk>;
 }
 /**
  * Extract the per-model `modelOptions` type a text adapter accepts. Used by

package/dist/esm/activities/summarize/chat-stream-summarize.js

@@ -1,8 +1,29 @@
 import { toRunErrorPayload } from "../error-payload.js";
+import { rebuildTokenUsage } from "../../utilities/ag-ui-usage.js";
+import { tanstackMetadata } from "../../utilities/merge-metadata.js";
+import { normalizeStreamChunk } from "../../utilities/normalize-stream-chunk.js";
 import { BaseSummarizeAdapter } from "./adapter.js";
 import { MAX_TOKENS_KEYS } from "../../utilities/sampling-keys.js";
 import { EventType } from "@ag-ui/core";
 //#region src/activities/summarize/chat-stream-summarize.ts
+function consumeSpecSummarizeChunk(chunk, state) {
+	if (chunk.type === EventType.TEXT_MESSAGE_CONTENT) {
+		if (chunk.delta) state.summary += chunk.delta;
+		return;
+	}
+	const tanstack = tanstackMetadata(chunk);
+	if ((chunk.type === EventType.RUN_STARTED || chunk.type === EventType.RUN_FINISHED || chunk.type === EventType.TEXT_MESSAGE_START) && typeof tanstack?.model === "string") state.model = tanstack.model;
+	if (chunk.type === EventType.RUN_FINISHED) {
+		const rebuilt = rebuildTokenUsage(chunk.usage, tanstack?.usage);
+		if (rebuilt) state.usage = rebuilt;
+	}
+}
+function throwRunError(chunk) {
+	const message = typeof chunk.message === "string" && chunk.message.length > 0 ? chunk.message : "Summarization failed";
+	const err = new Error(message);
+	if (typeof chunk.code === "string") err.code = chunk.code;
+	throw err;
+}
 /**
 * Provider-native max-output-tokens key per summarize-adapter `name`. summarize
 * is provider-agnostic and forwards `modelOptions` opaquely to the wrapped text
@@ -118,35 +139,24 @@ var ChatStreamSummarizeAdapter = class extends BaseSummarizeAdapter {
 	}
 	async summarize(options) {
 		const systemPrompt = this.buildSummarizationPrompt(options);
-		let summary = "";
 		const id = this.generateId();
-		let model = options.model;
-		let usage = {
-			promptTokens: 0,
-			completionTokens: 0,
-			totalTokens: 0
+		const state = {
+			summary: "",
+			model: options.model,
+			usage: {
+				promptTokens: 0,
+				completionTokens: 0,
+				totalTokens: 0
+			}
 		};
 		options.logger.request(`activity=summarize provider=${this.name} model=${options.model} text-length=${options.text.length} maxLength=${options.maxLength ?? "unset"}`, {
 			provider: this.name,
 			model: options.model
 		});
 		try {
-			for await (const chunk of this.textAdapter.chatStream(this.buildTextOptions(options, systemPrompt))) {
-				if (chunk.type === "TEXT_MESSAGE_CONTENT") {
-					if (chunk.content) summary = chunk.content;
-					else if (chunk.delta) summary += chunk.delta;
-					model = chunk.model || model;
-				}
-				if (chunk.type === "RUN_FINISHED") {
-					if (chunk.usage) usage = chunk.usage;
-				}
-				if (chunk.type === "RUN_ERROR") {
-					const message = (chunk.error && typeof chunk.error.message === "string" ? chunk.error.message : null) ?? "Summarization failed";
-					const code = chunk.error && typeof chunk.error.code === "string" ? chunk.error.code : void 0;
-					const err = new Error(message);
-					if (code) err.code = code;
-					throw err;
-				}
+			for await (const raw of this.textAdapter.chatStream(this.buildTextOptions(options, systemPrompt))) for (const chunk of normalizeStreamChunk(raw)) {
+				if (chunk.type === EventType.RUN_ERROR) throwRunError(chunk);
+				consumeSpecSummarizeChunk(chunk, state);
 			}
 		} catch (error) {
 			options.logger.errors(`${this.name}.summarize fatal`, {
@@ -157,9 +167,9 @@ var ChatStreamSummarizeAdapter = class extends BaseSummarizeAdapter {
 		}
 		return {
 			id,
-			model,
-			summary,
-			usage
+			model: state.model,
+			summary: state.summary,
+			usage: state.usage
 		};
 	}
 	async *summarizeStream(options) {
@@ -169,36 +179,29 @@ var ChatStreamSummarizeAdapter = class extends BaseSummarizeAdapter {
 			model: options.model
 		});
 		const id = this.generateId();
-		let summary = "";
-		let model = options.model;
-		let usage = {
-			promptTokens: 0,
-			completionTokens: 0,
-			totalTokens: 0
+		const state = {
+			summary: "",
+			model: options.model,
+			usage: {
+				promptTokens: 0,
+				completionTokens: 0,
+				totalTokens: 0
+			}
 		};
 		try {
-			for await (const chunk of this.textAdapter.chatStream(this.buildTextOptions(options, systemPrompt))) {
-				if (chunk.type === "TEXT_MESSAGE_CONTENT") {
-					if (chunk.content) summary = chunk.content;
-					else if (chunk.delta) summary += chunk.delta;
-					if (chunk.model) model = chunk.model;
-				}
-				if (chunk.type === "RUN_FINISHED") {
-					if (chunk.usage) usage = chunk.usage;
-					if (chunk.model) model = chunk.model;
-					yield {
-						type: EventType.CUSTOM,
-						name: "generation:result",
-						value: {
-							id,
-							model,
-							summary,
-							usage
-						},
-						model,
-						timestamp: Date.now()
-					};
-				}
+			for await (const raw of this.textAdapter.chatStream(this.buildTextOptions(options, systemPrompt))) for (const chunk of normalizeStreamChunk(raw)) {
+				consumeSpecSummarizeChunk(chunk, state);
+				if (chunk.type === EventType.RUN_FINISHED) yield {
+					type: EventType.CUSTOM,
+					name: "generation:result",
+					value: {
+						id,
+						model: state.model,
+						summary: state.summary,
+						usage: state.usage
+					},
+					timestamp: Date.now()
+				};
 				yield chunk;
 			}
 		} catch (error) {

package/dist/esm/activities/summarize/chat-stream-summarize.js.map

@@ -1 +1 @@
-{"version":3,"file":"chat-stream-summarize.js","names":[],"sources":["../../../../src/activities/summarize/chat-stream-summarize.ts"],"sourcesContent":["import { EventType } from '@ag-ui/core'\nimport { toRunErrorPayload } from '../error-payload'\nimport { MAX_TOKENS_KEYS } from '../../utilities/sampling-keys'\nimport { BaseSummarizeAdapter } from './adapter'\nimport type {\n  StreamChunk,\n  SummarizationOptions,\n  SummarizationResult,\n  TextOptions,\n} from '../../types'\n\n/**\n * Minimal contract for a text adapter that supports `chatStream`. Lets\n * `ChatStreamSummarizeAdapter` work with any text adapter without coupling\n * to a specific implementation.\n *\n * The provider-options shape is intentionally `any` here — the wrapper only\n * forwards `modelOptions` straight through, so a text adapter with a richer\n * per-model options type (e.g. `ResolveProviderOptions<TModel>`) is still\n * acceptable. Summarize-level type safety is enforced via\n * `SummarizationOptions<TProviderOptions>` on the wrapper itself.\n */\nexport interface ChatStreamCapable {\n  chatStream: (options: TextOptions<any>) => AsyncIterable<StreamChunk>\n}\n\n/**\n * Provider-native max-output-tokens key per summarize-adapter `name`. summarize\n * is provider-agnostic and forwards `modelOptions` opaquely to the wrapped text\n * adapter, so `maxLength` must be written under the exact key the underlying\n * provider reads — no adapter reads a generic `maxTokens`. Ollama is the one\n * exception: it nests sampling under `options`, so it has no entry here and is\n * handled as a special nested case in `applyMaxLength`/`applyDefaultTemperature`.\n *\n * Keep in sync with each adapter's wire mapping:\n * - OpenAI (Responses): `max_output_tokens`\n * - Anthropic / Grok: `max_tokens`\n * - Groq: `max_completion_tokens`\n * - Gemini: `maxOutputTokens`\n * - OpenRouter: `maxCompletionTokens`\n * - LLM Gateway: `max_tokens`\n * - Ollama: nested `options.num_predict` (no entry — see `applyMaxLength`)\n */\nconst MAX_TOKENS_KEY_BY_ADAPTER: Record<string, string> = {\n  openai: 'max_output_tokens',\n  anthropic: 'max_tokens',\n  grok: 'max_tokens',\n  groq: 'max_completion_tokens',\n  gemini: 'maxOutputTokens',\n  openrouter: 'maxCompletionTokens',\n  // LLM Gateway exposes an OpenAI-compatible Chat Completions surface whose\n  // only output cap is `max_tokens` — it does not read `max_completion_tokens`.\n  llmgateway: 'max_tokens',\n}\n\n/**\n * Every flat key any supported provider uses to cap output tokens (plus the\n * generic `maxTokens` spelling no adapter reads). Used to detect a\n * caller-supplied token limit so the summarize default never overrides an\n * explicit caller value. Shared with the OTel middleware via\n * `MAX_TOKENS_KEYS` so the two spelling sets cannot drift.\n */\nconst KNOWN_MAX_TOKENS_KEYS = MAX_TOKENS_KEYS\n\n/**\n * Whether `applyMaxLength` knows how to place a token limit for this adapter\n * `name` (either the nested Ollama shape or a flat provider-native key).\n * Used to surface a warning when `maxLength` would otherwise be silently\n * dropped for an unrecognised adapter name.\n */\nfunction isKnownMaxTokensAdapter(adapterName: string): boolean {\n  return (\n    adapterName === 'ollama' ||\n    MAX_TOKENS_KEY_BY_ADAPTER[adapterName] !== undefined\n  )\n}\n\n/**\n * Apply the low-temperature summarize default to a working copy of the\n * caller's `modelOptions`, placed where the wrapped provider actually reads\n * it (nested under `options` for Ollama, flat otherwise). The caller always\n * wins: if they already set `temperature` in that location, it is untouched.\n */\nfunction applyDefaultTemperature(\n  adapterName: string,\n  temperature: number,\n  modelOptions: Record<string, unknown>,\n): Record<string, unknown> {\n  const merged: Record<string, unknown> = { ...modelOptions }\n\n  if (adapterName === 'ollama') {\n    const existing =\n      merged.options && typeof merged.options === 'object'\n        ? (merged.options as Record<string, unknown>)\n        : undefined\n    if (existing && 'temperature' in existing) return merged\n    merged.options = { temperature, ...existing }\n    return merged\n  }\n\n  if ('temperature' in merged) return merged\n  merged.temperature = temperature\n  return merged\n}\n\n/**\n * Resolve `maxLength` to the provider-native max-output-tokens key for the\n * given summarize-adapter `name` (this wrapper's OWN `name`, not the wrapped\n * text adapter's) and merge it into a working copy of the caller's\n * `modelOptions`. The caller always wins: if they already set any recognised\n * token-limit key (flat or, for Ollama, nested `options.num_predict`), the\n * default is left untouched. Unknown/unrecognised adapter names fall back to\n * NOT setting a token key (the prompt hint still asks the model to stay under\n * `maxLength`) rather than writing a dead key no provider reads.\n *\n * Caveat (intentional): \"caller wins\" keys off ANY recognised spelling in\n * `KNOWN_MAX_TOKENS_KEYS`, but only the adapter's native key is read on the\n * wire. So a caller who sets a NON-native spelling for this provider — e.g.\n * `maxTokens`, or Anthropic's `max_tokens` against an OpenAI adapter — suppresses\n * the summarize default WITHOUT getting their own value applied either: neither\n * cap reaches the wire. This favours never clobbering a migration leftover over\n * guaranteeing a cap; the prompt-level hint still asks the model to stay under\n * `maxLength`. Rename the key to the provider-native spelling to forward it.\n */\nfunction applyMaxLength(\n  adapterName: string,\n  maxLength: number,\n  modelOptions: Record<string, unknown>,\n): Record<string, unknown> {\n  const merged: Record<string, unknown> = { ...modelOptions }\n\n  if (adapterName === 'ollama') {\n    // Honor a caller-set limit in either shape: a recognised flat key (e.g.\n    // left over from a migration) or the nested `options.num_predict`.\n    const callerSetFlatLimit = KNOWN_MAX_TOKENS_KEYS.some(\n      (k) => typeof merged[k] === 'number',\n    )\n    const existing =\n      merged.options && typeof merged.options === 'object'\n        ? (merged.options as Record<string, unknown>)\n        : undefined\n    if (\n      callerSetFlatLimit ||\n      (existing && typeof existing.num_predict === 'number')\n    ) {\n      return merged\n    }\n    merged.options = { num_predict: maxLength, ...existing }\n    return merged\n  }\n\n  const key = MAX_TOKENS_KEY_BY_ADAPTER[adapterName]\n  if (key === undefined) return merged\n\n  const callerSetLimit = KNOWN_MAX_TOKENS_KEYS.some(\n    (k) => typeof merged[k] === 'number',\n  )\n  if (callerSetLimit) return merged\n\n  merged[key] = maxLength\n  return merged\n}\n\n/**\n * Extract the per-model `modelOptions` type a text adapter accepts. Used by\n * provider summarize factories so their `modelOptions` IntelliSense matches\n * what the underlying text adapter actually understands.\n */\nexport type InferTextProviderOptions<TAdapter> = TAdapter extends {\n  '~types': { providerOptions: infer P }\n}\n  ? P extends object\n    ? P\n    : object\n  : object\n\n/**\n * Summarize adapter that wraps any `ChatStreamCapable` text adapter and\n * prompts it for summarization. Not tied to any wire format.\n */\nexport class ChatStreamSummarizeAdapter<\n  TModel extends string,\n  TProviderOptions extends object = Record<string, unknown>,\n> extends BaseSummarizeAdapter<TModel, TProviderOptions> {\n  readonly name: string\n\n  private readonly textAdapter: ChatStreamCapable\n\n  constructor(\n    textAdapter: ChatStreamCapable,\n    model: TModel,\n    name: string = 'chat-stream-summarize',\n  ) {\n    super({}, model)\n    this.name = name\n    this.textAdapter = textAdapter\n  }\n\n  async summarize(\n    options: SummarizationOptions<TProviderOptions>,\n  ): Promise<SummarizationResult> {\n    const systemPrompt = this.buildSummarizationPrompt(options)\n\n    let summary = ''\n    const id = this.generateId()\n    let model = options.model\n    let usage = { promptTokens: 0, completionTokens: 0, totalTokens: 0 }\n\n    options.logger.request(\n      `activity=summarize provider=${this.name} model=${options.model} text-length=${options.text.length} maxLength=${options.maxLength ?? 'unset'}`,\n      { provider: this.name, model: options.model },\n    )\n\n    try {\n      for await (const chunk of this.textAdapter.chatStream(\n        this.buildTextOptions(options, systemPrompt),\n      )) {\n        if (chunk.type === 'TEXT_MESSAGE_CONTENT') {\n          if (chunk.content) {\n            summary = chunk.content\n          } else if (chunk.delta) {\n            // Append delta only when present — a content-less chunk with no\n            // delta would otherwise concat literal `'undefined'`.\n            summary += chunk.delta\n          }\n          model = chunk.model || model\n        }\n        if (chunk.type === 'RUN_FINISHED') {\n          if (chunk.usage) {\n            usage = chunk.usage\n          }\n        }\n        // Surface failures: the underlying chatStream emits RUN_ERROR instead\n        // of throwing, so without this branch summarize() would return an\n        // empty summary and pretend a failed run succeeded.\n        if (chunk.type === 'RUN_ERROR') {\n          const message =\n            (chunk.error && typeof chunk.error.message === 'string'\n              ? chunk.error.message\n              : null) ?? 'Summarization failed'\n          const code =\n            chunk.error && typeof chunk.error.code === 'string'\n              ? chunk.error.code\n              : undefined\n          const err = new Error(message)\n          if (code) {\n            ;(err as Error & { code?: string }).code = code\n          }\n          throw err\n        }\n      }\n    } catch (error: unknown) {\n      // Narrow before logging: raw SDK errors can carry request metadata\n      // (including auth headers) which we must never surface to user loggers.\n      options.logger.errors(`${this.name}.summarize fatal`, {\n        error: toRunErrorPayload(error, `${this.name}.summarize failed`),\n        source: `${this.name}.summarize`,\n      })\n      throw error\n    }\n\n    return { id, model, summary, usage }\n  }\n\n  override async *summarizeStream(\n    options: SummarizationOptions<TProviderOptions>,\n  ): AsyncIterable<StreamChunk> {\n    const systemPrompt = this.buildSummarizationPrompt(options)\n\n    options.logger.request(\n      `activity=summarizeStream provider=${this.name} model=${options.model} text-length=${options.text.length} maxLength=${options.maxLength ?? 'unset'}`,\n      { provider: this.name, model: options.model },\n    )\n\n    const id = this.generateId()\n    let summary = ''\n    let model = options.model\n    let usage: SummarizationResult['usage'] = {\n      promptTokens: 0,\n      completionTokens: 0,\n      totalTokens: 0,\n    }\n\n    try {\n      for await (const chunk of this.textAdapter.chatStream(\n        this.buildTextOptions(options, systemPrompt),\n      )) {\n        // Accumulate the same way `summarize()` does so consumers see deltas\n        // AND the terminal `generation:result` event below carries the same\n        // final summary that non-streaming returns.\n        if (chunk.type === 'TEXT_MESSAGE_CONTENT') {\n          if (chunk.content) {\n            summary = chunk.content\n          } else if (chunk.delta) {\n            summary += chunk.delta\n          }\n          if (chunk.model) model = chunk.model\n        }\n\n        // Emit the GenerationClient-shaped result event just before the\n        // terminal RUN_FINISHED so subscribers (useSummarize) populate\n        // `result` before flipping `status` to success.\n        if (chunk.type === 'RUN_FINISHED') {\n          if (chunk.usage) usage = chunk.usage\n          if (chunk.model) model = chunk.model\n          yield {\n            type: EventType.CUSTOM,\n            name: 'generation:result',\n            value: { id, model, summary, usage } satisfies SummarizationResult,\n            model,\n            timestamp: Date.now(),\n          }\n        }\n\n        yield chunk\n      }\n    } catch (error: unknown) {\n      options.logger.errors(`${this.name}.summarizeStream fatal`, {\n        error: toRunErrorPayload(error, `${this.name}.summarizeStream failed`),\n        source: `${this.name}.summarizeStream`,\n      })\n      throw error\n    }\n  }\n\n  /**\n   * Build the TextOptions passed to the underlying chatStream. Provider\n   * `modelOptions` from the summarize call are forwarded as-is so knobs like\n   * Anthropic cache headers, Gemini safety settings, or Ollama tuning params\n   * still reach the wire layer.\n   */\n  protected buildTextOptions(\n    options: SummarizationOptions<TProviderOptions>,\n    systemPrompt: string,\n  ): TextOptions<TProviderOptions> {\n    // Sampling knobs now live in provider-native `modelOptions`. Apply the\n    // low-temperature default where the wrapped provider actually reads it\n    // (nested under `options` for Ollama, flat otherwise) so callers can still\n    // override it. Resolving the placement from this summarize adapter's OWN\n    // `name` keeps the default off the wire correctly per provider — a flat\n    // `temperature` would be silently dropped by Ollama while still showing up\n    // in OTel.\n    let working: Record<string, unknown> = {\n      ...(options.modelOptions as Record<string, unknown> | undefined),\n    }\n    working = applyDefaultTemperature(this.name, 0.3, working)\n    // `maxLength` must reach the wire under the provider-native token key (it\n    // differs per provider, and no adapter reads a generic `maxTokens`).\n    // Resolve it from this summarize adapter's `name` (the constructor arg,\n    // not the wrapped text adapter's name), never overriding a caller-supplied\n    // token limit.\n    if (options.maxLength !== undefined) {\n      if (!isKnownMaxTokensAdapter(this.name)) {\n        options.logger.warn(\n          `summarize: maxLength=${options.maxLength} could not be mapped to a provider token key for adapter name \"${this.name}\" — it was dropped from modelOptions (the prompt still asks the model to stay under it). Construct ChatStreamSummarizeAdapter with a recognised provider name to forward the cap.`,\n          { provider: this.name },\n        )\n      }\n      working = applyMaxLength(this.name, options.maxLength, working)\n    }\n    const modelOptions = working as TProviderOptions\n\n    return {\n      model: options.model,\n      messages: [{ role: 'user', content: options.text }],\n      systemPrompts: [systemPrompt],\n      modelOptions,\n      logger: options.logger,\n      // Forward the run identity so the wrapped chat stamps it onto RUN_STARTED\n      // (chat uses `runId` as its `runIdOverride`). Conditional spreads keep the\n      // fields absent when unset, under `exactOptionalPropertyTypes`.\n      ...(options.runId !== undefined ? { runId: options.runId } : {}),\n      ...(options.threadId !== undefined ? { threadId: options.threadId } : {}),\n    }\n  }\n\n  protected buildSummarizationPrompt(\n    options: SummarizationOptions<TProviderOptions>,\n  ): string {\n    let prompt = 'You are a professional summarizer. '\n\n    switch (options.style) {\n      case 'bullet-points':\n        prompt += 'Provide a summary in bullet point format. '\n        break\n      case 'paragraph':\n        prompt += 'Provide a summary in paragraph format. '\n        break\n      case 'concise':\n        prompt += 'Provide a very concise summary in 1-2 sentences. '\n        break\n      case undefined:\n        prompt += 'Provide a clear and concise summary. '\n        break\n      default:\n        prompt += 'Provide a clear and concise summary. '\n    }\n\n    if (options.focus && options.focus.length > 0) {\n      prompt += `Focus on the following aspects: ${options.focus.join(', ')}. `\n    }\n\n    if (options.maxLength) {\n      prompt += `Keep the summary under ${options.maxLength} tokens. `\n    }\n\n    return prompt\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AA2CA,IAAM,4BAAoD;CACxD,QAAQ;CACR,WAAW;CACX,MAAM;CACN,MAAM;CACN,QAAQ;CACR,YAAY;CAGZ,YAAY;AACd;;;;;;;;AASA,IAAM,wBAAwB;;;;;;;AAQ9B,SAAS,wBAAwB,aAA8B;CAC7D,OACE,gBAAgB,YAChB,0BAA0B,iBAAiB,KAAA;AAE/C;;;;;;;AAQA,SAAS,wBACP,aACA,aACA,cACyB;CACzB,MAAM,SAAkC,EAAE,GAAG,aAAa;CAE1D,IAAI,gBAAgB,UAAU;EAC5B,MAAM,WACJ,OAAO,WAAW,OAAO,OAAO,YAAY,WACvC,OAAO,UACR,KAAA;EACN,IAAI,YAAY,iBAAiB,UAAU,OAAO;EAClD,OAAO,UAAU;GAAE;GAAa,GAAG;EAAS;EAC5C,OAAO;CACT;CAEA,IAAI,iBAAiB,QAAQ,OAAO;CACpC,OAAO,cAAc;CACrB,OAAO;AACT;;;;;;;;;;;;;;;;;;;;AAqBA,SAAS,eACP,aACA,WACA,cACyB;CACzB,MAAM,SAAkC,EAAE,GAAG,aAAa;CAE1D,IAAI,gBAAgB,UAAU;EAG5B,MAAM,qBAAqB,sBAAsB,MAC9C,MAAM,OAAO,OAAO,OAAO,QAC9B;EACA,MAAM,WACJ,OAAO,WAAW,OAAO,OAAO,YAAY,WACvC,OAAO,UACR,KAAA;EACN,IACE,sBACC,YAAY,OAAO,SAAS,gBAAgB,UAE7C,OAAO;EAET,OAAO,UAAU;GAAE,aAAa;GAAW,GAAG;EAAS;EACvD,OAAO;CACT;CAEA,MAAM,MAAM,0BAA0B;CACtC,IAAI,QAAQ,KAAA,GAAW,OAAO;CAK9B,IAHuB,sBAAsB,MAC1C,MAAM,OAAO,OAAO,OAAO,QAE1B,GAAgB,OAAO;CAE3B,OAAO,OAAO;CACd,OAAO;AACT;;;;;AAmBA,IAAa,6BAAb,cAGU,qBAA+C;CACvD;CAEA;CAEA,YACE,aACA,OACA,OAAe,yBACf;EACA,MAAM,CAAC,GAAG,KAAK;EACf,KAAK,OAAO;EACZ,KAAK,cAAc;CACrB;CAEA,MAAM,UACJ,SAC8B;EAC9B,MAAM,eAAe,KAAK,yBAAyB,OAAO;EAE1D,IAAI,UAAU;EACd,MAAM,KAAK,KAAK,WAAW;EAC3B,IAAI,QAAQ,QAAQ;EACpB,IAAI,QAAQ;GAAE,cAAc;GAAG,kBAAkB;GAAG,aAAa;EAAE;EAEnE,QAAQ,OAAO,QACb,+BAA+B,KAAK,KAAK,SAAS,QAAQ,MAAM,eAAe,QAAQ,KAAK,OAAO,aAAa,QAAQ,aAAa,WACrI;GAAE,UAAU,KAAK;GAAM,OAAO,QAAQ;EAAM,CAC9C;EAEA,IAAI;GACF,WAAW,MAAM,SAAS,KAAK,YAAY,WACzC,KAAK,iBAAiB,SAAS,YAAY,CAC7C,GAAG;IACD,IAAI,MAAM,SAAS,wBAAwB;KACzC,IAAI,MAAM,SACR,UAAU,MAAM;UACX,IAAI,MAAM,OAGf,WAAW,MAAM;KAEnB,QAAQ,MAAM,SAAS;IACzB;IACA,IAAI,MAAM,SAAS,gBACb;SAAA,MAAM,OACR,QAAQ,MAAM;IAAA;IAMlB,IAAI,MAAM,SAAS,aAAa;KAC9B,MAAM,WACH,MAAM,SAAS,OAAO,MAAM,MAAM,YAAY,WAC3C,MAAM,MAAM,UACZ,SAAS;KACf,MAAM,OACJ,MAAM,SAAS,OAAO,MAAM,MAAM,SAAS,WACvC,MAAM,MAAM,OACZ,KAAA;KACN,MAAM,MAAM,IAAI,MAAM,OAAO;KAC7B,IAAI,MACD,IAAmC,OAAO;KAE7C,MAAM;IACR;GACF;EACF,SAAS,OAAgB;GAGvB,QAAQ,OAAO,OAAO,GAAG,KAAK,KAAK,mBAAmB;IACpD,OAAO,kBAAkB,OAAO,GAAG,KAAK,KAAK,kBAAkB;IAC/D,QAAQ,GAAG,KAAK,KAAK;GACvB,CAAC;GACD,MAAM;EACR;EAEA,OAAO;GAAE;GAAI;GAAO;GAAS;EAAM;CACrC;CAEA,OAAgB,gBACd,SAC4B;EAC5B,MAAM,eAAe,KAAK,yBAAyB,OAAO;EAE1D,QAAQ,OAAO,QACb,qCAAqC,KAAK,KAAK,SAAS,QAAQ,MAAM,eAAe,QAAQ,KAAK,OAAO,aAAa,QAAQ,aAAa,WAC3I;GAAE,UAAU,KAAK;GAAM,OAAO,QAAQ;EAAM,CAC9C;EAEA,MAAM,KAAK,KAAK,WAAW;EAC3B,IAAI,UAAU;EACd,IAAI,QAAQ,QAAQ;EACpB,IAAI,QAAsC;GACxC,cAAc;GACd,kBAAkB;GAClB,aAAa;EACf;EAEA,IAAI;GACF,WAAW,MAAM,SAAS,KAAK,YAAY,WACzC,KAAK,iBAAiB,SAAS,YAAY,CAC7C,GAAG;IAID,IAAI,MAAM,SAAS,wBAAwB;KACzC,IAAI,MAAM,SACR,UAAU,MAAM;UACX,IAAI,MAAM,OACf,WAAW,MAAM;KAEnB,IAAI,MAAM,OAAO,QAAQ,MAAM;IACjC;IAKA,IAAI,MAAM,SAAS,gBAAgB;KACjC,IAAI,MAAM,OAAO,QAAQ,MAAM;KAC/B,IAAI,MAAM,OAAO,QAAQ,MAAM;KAC/B,MAAM;MACJ,MAAM,UAAU;MAChB,MAAM;MACN,OAAO;OAAE;OAAI;OAAO;OAAS;MAAM;MACnC;MACA,WAAW,KAAK,IAAI;KACtB;IACF;IAEA,MAAM;GACR;EACF,SAAS,OAAgB;GACvB,QAAQ,OAAO,OAAO,GAAG,KAAK,KAAK,yBAAyB;IAC1D,OAAO,kBAAkB,OAAO,GAAG,KAAK,KAAK,wBAAwB;IACrE,QAAQ,GAAG,KAAK,KAAK;GACvB,CAAC;GACD,MAAM;EACR;CACF;;;;;;;CAQA,iBACE,SACA,cAC+B;EAQ/B,IAAI,UAAmC,EACrC,GAAI,QAAQ,aACd;EACA,UAAU,wBAAwB,KAAK,MAAM,IAAK,OAAO;EAMzD,IAAI,QAAQ,cAAc,KAAA,GAAW;GACnC,IAAI,CAAC,wBAAwB,KAAK,IAAI,GACpC,QAAQ,OAAO,KACb,wBAAwB,QAAQ,UAAU,iEAAiE,KAAK,KAAK,oLACrH,EAAE,UAAU,KAAK,KAAK,CACxB;GAEF,UAAU,eAAe,KAAK,MAAM,QAAQ,WAAW,OAAO;EAChE;EACA,MAAM,eAAe;EAErB,OAAO;GACL,OAAO,QAAQ;GACf,UAAU,CAAC;IAAE,MAAM;IAAQ,SAAS,QAAQ;GAAK,CAAC;GAClD,eAAe,CAAC,YAAY;GAC5B;GACA,QAAQ,QAAQ;GAIhB,GAAI,QAAQ,UAAU,KAAA,IAAY,EAAE,OAAO,QAAQ,MAAM,IAAI,CAAC;GAC9D,GAAI,QAAQ,aAAa,KAAA,IAAY,EAAE,UAAU,QAAQ,SAAS,IAAI,CAAC;EACzE;CACF;CAEA,yBACE,SACQ;EACR,IAAI,SAAS;EAEb,QAAQ,QAAQ,OAAhB;GACE,KAAK;IACH,UAAU;IACV;GACF,KAAK;IACH,UAAU;IACV;GACF,KAAK;IACH,UAAU;IACV;GACF,KAAK,KAAA;IACH,UAAU;IACV;GACF,SACE,UAAU;EACd;EAEA,IAAI,QAAQ,SAAS,QAAQ,MAAM,SAAS,GAC1C,UAAU,mCAAmC,QAAQ,MAAM,KAAK,IAAI,EAAE;EAGxE,IAAI,QAAQ,WACV,UAAU,0BAA0B,QAAQ,UAAU;EAGxD,OAAO;CACT;AACF"}
+{"version":3,"file":"chat-stream-summarize.js","names":[],"sources":["../../../../src/activities/summarize/chat-stream-summarize.ts"],"sourcesContent":["import { EventType } from '@ag-ui/core'\nimport { toRunErrorPayload } from '../error-payload'\nimport { MAX_TOKENS_KEYS } from '../../utilities/sampling-keys'\nimport { rebuildTokenUsage } from '../../utilities/ag-ui-usage'\nimport type { AdapterYieldChunk } from '../../utilities/adapter-yield-chunk'\nimport { tanstackMetadata } from '../../utilities/merge-metadata'\nimport { normalizeStreamChunk } from '../../utilities/normalize-stream-chunk'\nimport { BaseSummarizeAdapter } from './adapter'\nimport type {\n  StreamChunk,\n  SummarizationOptions,\n  SummarizationResult,\n  TextOptions,\n  TokenUsage,\n} from '../../types'\n\nfunction consumeSpecSummarizeChunk(\n  chunk: StreamChunk,\n  state: { summary: string; model: string; usage: TokenUsage },\n): void {\n  if (chunk.type === EventType.TEXT_MESSAGE_CONTENT) {\n    if (chunk.delta) state.summary += chunk.delta\n    return\n  }\n\n  const tanstack = tanstackMetadata(chunk)\n  if (\n    (chunk.type === EventType.RUN_STARTED ||\n      chunk.type === EventType.RUN_FINISHED ||\n      chunk.type === EventType.TEXT_MESSAGE_START) &&\n    typeof tanstack?.model === 'string'\n  ) {\n    state.model = tanstack.model\n  }\n\n  if (chunk.type === EventType.RUN_FINISHED) {\n    const rebuilt = rebuildTokenUsage(chunk.usage, tanstack?.usage)\n    if (rebuilt) state.usage = rebuilt\n  }\n}\n\nfunction throwRunError(\n  chunk: Extract<StreamChunk, { type: 'RUN_ERROR' }>,\n): never {\n  const message =\n    typeof chunk.message === 'string' && chunk.message.length > 0\n      ? chunk.message\n      : 'Summarization failed'\n  const err = new Error(message)\n  if (typeof chunk.code === 'string') {\n    ;(err as Error & { code?: string }).code = chunk.code\n  }\n  throw err\n}\n\n/**\n * Minimal contract for a text adapter that supports `chatStream`. Lets\n * `ChatStreamSummarizeAdapter` work with any text adapter without coupling\n * to a specific implementation.\n *\n * The provider-options shape is intentionally `any` here — the wrapper only\n * forwards `modelOptions` straight through, so a text adapter with a richer\n * per-model options type (e.g. `ResolveProviderOptions<TModel>`) is still\n * acceptable. Summarize-level type safety is enforced via\n * `SummarizationOptions<TProviderOptions>` on the wrapper itself.\n */\nexport interface ChatStreamCapable {\n  chatStream: (options: TextOptions<any>) => AsyncIterable<AdapterYieldChunk>\n}\n\n/**\n * Provider-native max-output-tokens key per summarize-adapter `name`. summarize\n * is provider-agnostic and forwards `modelOptions` opaquely to the wrapped text\n * adapter, so `maxLength` must be written under the exact key the underlying\n * provider reads — no adapter reads a generic `maxTokens`. Ollama is the one\n * exception: it nests sampling under `options`, so it has no entry here and is\n * handled as a special nested case in `applyMaxLength`/`applyDefaultTemperature`.\n *\n * Keep in sync with each adapter's wire mapping:\n * - OpenAI (Responses): `max_output_tokens`\n * - Anthropic / Grok: `max_tokens`\n * - Groq: `max_completion_tokens`\n * - Gemini: `maxOutputTokens`\n * - OpenRouter: `maxCompletionTokens`\n * - LLM Gateway: `max_tokens`\n * - Ollama: nested `options.num_predict` (no entry — see `applyMaxLength`)\n */\nconst MAX_TOKENS_KEY_BY_ADAPTER: Record<string, string> = {\n  openai: 'max_output_tokens',\n  anthropic: 'max_tokens',\n  grok: 'max_tokens',\n  groq: 'max_completion_tokens',\n  gemini: 'maxOutputTokens',\n  openrouter: 'maxCompletionTokens',\n  // LLM Gateway exposes an OpenAI-compatible Chat Completions surface whose\n  // only output cap is `max_tokens` — it does not read `max_completion_tokens`.\n  llmgateway: 'max_tokens',\n}\n\n/**\n * Every flat key any supported provider uses to cap output tokens (plus the\n * generic `maxTokens` spelling no adapter reads). Used to detect a\n * caller-supplied token limit so the summarize default never overrides an\n * explicit caller value. Shared with the OTel middleware via\n * `MAX_TOKENS_KEYS` so the two spelling sets cannot drift.\n */\nconst KNOWN_MAX_TOKENS_KEYS = MAX_TOKENS_KEYS\n\n/**\n * Whether `applyMaxLength` knows how to place a token limit for this adapter\n * `name` (either the nested Ollama shape or a flat provider-native key).\n * Used to surface a warning when `maxLength` would otherwise be silently\n * dropped for an unrecognised adapter name.\n */\nfunction isKnownMaxTokensAdapter(adapterName: string): boolean {\n  return (\n    adapterName === 'ollama' ||\n    MAX_TOKENS_KEY_BY_ADAPTER[adapterName] !== undefined\n  )\n}\n\n/**\n * Apply the low-temperature summarize default to a working copy of the\n * caller's `modelOptions`, placed where the wrapped provider actually reads\n * it (nested under `options` for Ollama, flat otherwise). The caller always\n * wins: if they already set `temperature` in that location, it is untouched.\n */\nfunction applyDefaultTemperature(\n  adapterName: string,\n  temperature: number,\n  modelOptions: Record<string, unknown>,\n): Record<string, unknown> {\n  const merged: Record<string, unknown> = { ...modelOptions }\n\n  if (adapterName === 'ollama') {\n    const existing =\n      merged.options && typeof merged.options === 'object'\n        ? (merged.options as Record<string, unknown>)\n        : undefined\n    if (existing && 'temperature' in existing) return merged\n    merged.options = { temperature, ...existing }\n    return merged\n  }\n\n  if ('temperature' in merged) return merged\n  merged.temperature = temperature\n  return merged\n}\n\n/**\n * Resolve `maxLength` to the provider-native max-output-tokens key for the\n * given summarize-adapter `name` (this wrapper's OWN `name`, not the wrapped\n * text adapter's) and merge it into a working copy of the caller's\n * `modelOptions`. The caller always wins: if they already set any recognised\n * token-limit key (flat or, for Ollama, nested `options.num_predict`), the\n * default is left untouched. Unknown/unrecognised adapter names fall back to\n * NOT setting a token key (the prompt hint still asks the model to stay under\n * `maxLength`) rather than writing a dead key no provider reads.\n *\n * Caveat (intentional): \"caller wins\" keys off ANY recognised spelling in\n * `KNOWN_MAX_TOKENS_KEYS`, but only the adapter's native key is read on the\n * wire. So a caller who sets a NON-native spelling for this provider — e.g.\n * `maxTokens`, or Anthropic's `max_tokens` against an OpenAI adapter — suppresses\n * the summarize default WITHOUT getting their own value applied either: neither\n * cap reaches the wire. This favours never clobbering a migration leftover over\n * guaranteeing a cap; the prompt-level hint still asks the model to stay under\n * `maxLength`. Rename the key to the provider-native spelling to forward it.\n */\nfunction applyMaxLength(\n  adapterName: string,\n  maxLength: number,\n  modelOptions: Record<string, unknown>,\n): Record<string, unknown> {\n  const merged: Record<string, unknown> = { ...modelOptions }\n\n  if (adapterName === 'ollama') {\n    // Honor a caller-set limit in either shape: a recognised flat key (e.g.\n    // left over from a migration) or the nested `options.num_predict`.\n    const callerSetFlatLimit = KNOWN_MAX_TOKENS_KEYS.some(\n      (k) => typeof merged[k] === 'number',\n    )\n    const existing =\n      merged.options && typeof merged.options === 'object'\n        ? (merged.options as Record<string, unknown>)\n        : undefined\n    if (\n      callerSetFlatLimit ||\n      (existing && typeof existing.num_predict === 'number')\n    ) {\n      return merged\n    }\n    merged.options = { num_predict: maxLength, ...existing }\n    return merged\n  }\n\n  const key = MAX_TOKENS_KEY_BY_ADAPTER[adapterName]\n  if (key === undefined) return merged\n\n  const callerSetLimit = KNOWN_MAX_TOKENS_KEYS.some(\n    (k) => typeof merged[k] === 'number',\n  )\n  if (callerSetLimit) return merged\n\n  merged[key] = maxLength\n  return merged\n}\n\n/**\n * Extract the per-model `modelOptions` type a text adapter accepts. Used by\n * provider summarize factories so their `modelOptions` IntelliSense matches\n * what the underlying text adapter actually understands.\n */\nexport type InferTextProviderOptions<TAdapter> = TAdapter extends {\n  '~types': { providerOptions: infer P }\n}\n  ? P extends object\n    ? P\n    : object\n  : object\n\n/**\n * Summarize adapter that wraps any `ChatStreamCapable` text adapter and\n * prompts it for summarization. Not tied to any wire format.\n */\nexport class ChatStreamSummarizeAdapter<\n  TModel extends string,\n  TProviderOptions extends object = Record<string, unknown>,\n> extends BaseSummarizeAdapter<TModel, TProviderOptions> {\n  readonly name: string\n\n  private readonly textAdapter: ChatStreamCapable\n\n  constructor(\n    textAdapter: ChatStreamCapable,\n    model: TModel,\n    name: string = 'chat-stream-summarize',\n  ) {\n    super({}, model)\n    this.name = name\n    this.textAdapter = textAdapter\n  }\n\n  async summarize(\n    options: SummarizationOptions<TProviderOptions>,\n  ): Promise<SummarizationResult> {\n    const systemPrompt = this.buildSummarizationPrompt(options)\n\n    const id = this.generateId()\n    const state = {\n      summary: '',\n      model: options.model,\n      usage: { promptTokens: 0, completionTokens: 0, totalTokens: 0 },\n    }\n\n    options.logger.request(\n      `activity=summarize provider=${this.name} model=${options.model} text-length=${options.text.length} maxLength=${options.maxLength ?? 'unset'}`,\n      { provider: this.name, model: options.model },\n    )\n\n    try {\n      for await (const raw of this.textAdapter.chatStream(\n        this.buildTextOptions(options, systemPrompt),\n      )) {\n        for (const chunk of normalizeStreamChunk(raw as AdapterYieldChunk)) {\n          // Surface failures: the underlying chatStream emits RUN_ERROR instead\n          // of throwing, so without this branch summarize() would return an\n          // empty summary and pretend a failed run succeeded.\n          if (chunk.type === EventType.RUN_ERROR) throwRunError(chunk)\n          consumeSpecSummarizeChunk(chunk, state)\n        }\n      }\n    } catch (error: unknown) {\n      // Narrow before logging: raw SDK errors can carry request metadata\n      // (including auth headers) which we must never surface to user loggers.\n      options.logger.errors(`${this.name}.summarize fatal`, {\n        error: toRunErrorPayload(error, `${this.name}.summarize failed`),\n        source: `${this.name}.summarize`,\n      })\n      throw error\n    }\n\n    return {\n      id,\n      model: state.model,\n      summary: state.summary,\n      usage: state.usage,\n    }\n  }\n\n  override async *summarizeStream(\n    options: SummarizationOptions<TProviderOptions>,\n  ): AsyncIterable<StreamChunk> {\n    const systemPrompt = this.buildSummarizationPrompt(options)\n\n    options.logger.request(\n      `activity=summarizeStream provider=${this.name} model=${options.model} text-length=${options.text.length} maxLength=${options.maxLength ?? 'unset'}`,\n      { provider: this.name, model: options.model },\n    )\n\n    const id = this.generateId()\n    const state = {\n      summary: '',\n      model: options.model,\n      usage: {\n        promptTokens: 0,\n        completionTokens: 0,\n        totalTokens: 0,\n      } satisfies SummarizationResult['usage'],\n    }\n\n    try {\n      for await (const raw of this.textAdapter.chatStream(\n        this.buildTextOptions(options, systemPrompt),\n      )) {\n        for (const chunk of normalizeStreamChunk(raw as AdapterYieldChunk)) {\n          // Accumulate the same way `summarize()` does so consumers see deltas\n          // AND the terminal `generation:result` event below carries the same\n          // final summary that non-streaming returns.\n          consumeSpecSummarizeChunk(chunk, state)\n\n          // Emit the GenerationClient-shaped result event just before the\n          // terminal RUN_FINISHED so subscribers (useSummarize) populate\n          // `result` before flipping `status` to success.\n          if (chunk.type === EventType.RUN_FINISHED) {\n            yield {\n              type: EventType.CUSTOM,\n              name: 'generation:result',\n              value: {\n                id,\n                model: state.model,\n                summary: state.summary,\n                usage: state.usage,\n              } satisfies SummarizationResult,\n              timestamp: Date.now(),\n            }\n          }\n\n          yield chunk\n        }\n      }\n    } catch (error: unknown) {\n      options.logger.errors(`${this.name}.summarizeStream fatal`, {\n        error: toRunErrorPayload(error, `${this.name}.summarizeStream failed`),\n        source: `${this.name}.summarizeStream`,\n      })\n      throw error\n    }\n  }\n\n  /**\n   * Build the TextOptions passed to the underlying chatStream. Provider\n   * `modelOptions` from the summarize call are forwarded as-is so knobs like\n   * Anthropic cache headers, Gemini safety settings, or Ollama tuning params\n   * still reach the wire layer.\n   */\n  protected buildTextOptions(\n    options: SummarizationOptions<TProviderOptions>,\n    systemPrompt: string,\n  ): TextOptions<TProviderOptions> {\n    // Sampling knobs now live in provider-native `modelOptions`. Apply the\n    // low-temperature default where the wrapped provider actually reads it\n    // (nested under `options` for Ollama, flat otherwise) so callers can still\n    // override it. Resolving the placement from this summarize adapter's OWN\n    // `name` keeps the default off the wire correctly per provider — a flat\n    // `temperature` would be silently dropped by Ollama while still showing up\n    // in OTel.\n    let working: Record<string, unknown> = {\n      ...(options.modelOptions as Record<string, unknown> | undefined),\n    }\n    working = applyDefaultTemperature(this.name, 0.3, working)\n    // `maxLength` must reach the wire under the provider-native token key (it\n    // differs per provider, and no adapter reads a generic `maxTokens`).\n    // Resolve it from this summarize adapter's `name` (the constructor arg,\n    // not the wrapped text adapter's name), never overriding a caller-supplied\n    // token limit.\n    if (options.maxLength !== undefined) {\n      if (!isKnownMaxTokensAdapter(this.name)) {\n        options.logger.warn(\n          `summarize: maxLength=${options.maxLength} could not be mapped to a provider token key for adapter name \"${this.name}\" — it was dropped from modelOptions (the prompt still asks the model to stay under it). Construct ChatStreamSummarizeAdapter with a recognised provider name to forward the cap.`,\n          { provider: this.name },\n        )\n      }\n      working = applyMaxLength(this.name, options.maxLength, working)\n    }\n    const modelOptions = working as TProviderOptions\n\n    return {\n      model: options.model,\n      messages: [{ role: 'user', content: options.text }],\n      systemPrompts: [systemPrompt],\n      modelOptions,\n      logger: options.logger,\n      // Forward the run identity so the wrapped chat stamps it onto RUN_STARTED\n      // (chat uses `runId` as its `runIdOverride`). Conditional spreads keep the\n      // fields absent when unset, under `exactOptionalPropertyTypes`.\n      ...(options.runId !== undefined ? { runId: options.runId } : {}),\n      ...(options.threadId !== undefined ? { threadId: options.threadId } : {}),\n    }\n  }\n\n  protected buildSummarizationPrompt(\n    options: SummarizationOptions<TProviderOptions>,\n  ): string {\n    let prompt = 'You are a professional summarizer. '\n\n    switch (options.style) {\n      case 'bullet-points':\n        prompt += 'Provide a summary in bullet point format. '\n        break\n      case 'paragraph':\n        prompt += 'Provide a summary in paragraph format. '\n        break\n      case 'concise':\n        prompt += 'Provide a very concise summary in 1-2 sentences. '\n        break\n      case undefined:\n        prompt += 'Provide a clear and concise summary. '\n        break\n      default:\n        prompt += 'Provide a clear and concise summary. '\n    }\n\n    if (options.focus && options.focus.length > 0) {\n      prompt += `Focus on the following aspects: ${options.focus.join(', ')}. `\n    }\n\n    if (options.maxLength) {\n      prompt += `Keep the summary under ${options.maxLength} tokens. `\n    }\n\n    return prompt\n  }\n}\n"],"mappings":";;;;;;;;AAgBA,SAAS,0BACP,OACA,OACM;CACN,IAAI,MAAM,SAAS,UAAU,sBAAsB;EACjD,IAAI,MAAM,OAAO,MAAM,WAAW,MAAM;EACxC;CACF;CAEA,MAAM,WAAW,iBAAiB,KAAK;CACvC,KACG,MAAM,SAAS,UAAU,eACxB,MAAM,SAAS,UAAU,gBACzB,MAAM,SAAS,UAAU,uBAC3B,OAAO,UAAU,UAAU,UAE3B,MAAM,QAAQ,SAAS;CAGzB,IAAI,MAAM,SAAS,UAAU,cAAc;EACzC,MAAM,UAAU,kBAAkB,MAAM,OAAO,UAAU,KAAK;EAC9D,IAAI,SAAS,MAAM,QAAQ;CAC7B;AACF;AAEA,SAAS,cACP,OACO;CACP,MAAM,UACJ,OAAO,MAAM,YAAY,YAAY,MAAM,QAAQ,SAAS,IACxD,MAAM,UACN;CACN,MAAM,MAAM,IAAI,MAAM,OAAO;CAC7B,IAAI,OAAO,MAAM,SAAS,UACvB,IAAmC,OAAO,MAAM;CAEnD,MAAM;AACR;;;;;;;;;;;;;;;;;;AAkCA,IAAM,4BAAoD;CACxD,QAAQ;CACR,WAAW;CACX,MAAM;CACN,MAAM;CACN,QAAQ;CACR,YAAY;CAGZ,YAAY;AACd;;;;;;;;AASA,IAAM,wBAAwB;;;;;;;AAQ9B,SAAS,wBAAwB,aAA8B;CAC7D,OACE,gBAAgB,YAChB,0BAA0B,iBAAiB,KAAA;AAE/C;;;;;;;AAQA,SAAS,wBACP,aACA,aACA,cACyB;CACzB,MAAM,SAAkC,EAAE,GAAG,aAAa;CAE1D,IAAI,gBAAgB,UAAU;EAC5B,MAAM,WACJ,OAAO,WAAW,OAAO,OAAO,YAAY,WACvC,OAAO,UACR,KAAA;EACN,IAAI,YAAY,iBAAiB,UAAU,OAAO;EAClD,OAAO,UAAU;GAAE;GAAa,GAAG;EAAS;EAC5C,OAAO;CACT;CAEA,IAAI,iBAAiB,QAAQ,OAAO;CACpC,OAAO,cAAc;CACrB,OAAO;AACT;;;;;;;;;;;;;;;;;;;;AAqBA,SAAS,eACP,aACA,WACA,cACyB;CACzB,MAAM,SAAkC,EAAE,GAAG,aAAa;CAE1D,IAAI,gBAAgB,UAAU;EAG5B,MAAM,qBAAqB,sBAAsB,MAC9C,MAAM,OAAO,OAAO,OAAO,QAC9B;EACA,MAAM,WACJ,OAAO,WAAW,OAAO,OAAO,YAAY,WACvC,OAAO,UACR,KAAA;EACN,IACE,sBACC,YAAY,OAAO,SAAS,gBAAgB,UAE7C,OAAO;EAET,OAAO,UAAU;GAAE,aAAa;GAAW,GAAG;EAAS;EACvD,OAAO;CACT;CAEA,MAAM,MAAM,0BAA0B;CACtC,IAAI,QAAQ,KAAA,GAAW,OAAO;CAK9B,IAHuB,sBAAsB,MAC1C,MAAM,OAAO,OAAO,OAAO,QAE1B,GAAgB,OAAO;CAE3B,OAAO,OAAO;CACd,OAAO;AACT;;;;;AAmBA,IAAa,6BAAb,cAGU,qBAA+C;CACvD;CAEA;CAEA,YACE,aACA,OACA,OAAe,yBACf;EACA,MAAM,CAAC,GAAG,KAAK;EACf,KAAK,OAAO;EACZ,KAAK,cAAc;CACrB;CAEA,MAAM,UACJ,SAC8B;EAC9B,MAAM,eAAe,KAAK,yBAAyB,OAAO;EAE1D,MAAM,KAAK,KAAK,WAAW;EAC3B,MAAM,QAAQ;GACZ,SAAS;GACT,OAAO,QAAQ;GACf,OAAO;IAAE,cAAc;IAAG,kBAAkB;IAAG,aAAa;GAAE;EAChE;EAEA,QAAQ,OAAO,QACb,+BAA+B,KAAK,KAAK,SAAS,QAAQ,MAAM,eAAe,QAAQ,KAAK,OAAO,aAAa,QAAQ,aAAa,WACrI;GAAE,UAAU,KAAK;GAAM,OAAO,QAAQ;EAAM,CAC9C;EAEA,IAAI;GACF,WAAW,MAAM,OAAO,KAAK,YAAY,WACvC,KAAK,iBAAiB,SAAS,YAAY,CAC7C,GACE,KAAK,MAAM,SAAS,qBAAqB,GAAwB,GAAG;IAIlE,IAAI,MAAM,SAAS,UAAU,WAAW,cAAc,KAAK;IAC3D,0BAA0B,OAAO,KAAK;GACxC;EAEJ,SAAS,OAAgB;GAGvB,QAAQ,OAAO,OAAO,GAAG,KAAK,KAAK,mBAAmB;IACpD,OAAO,kBAAkB,OAAO,GAAG,KAAK,KAAK,kBAAkB;IAC/D,QAAQ,GAAG,KAAK,KAAK;GACvB,CAAC;GACD,MAAM;EACR;EAEA,OAAO;GACL;GACA,OAAO,MAAM;GACb,SAAS,MAAM;GACf,OAAO,MAAM;EACf;CACF;CAEA,OAAgB,gBACd,SAC4B;EAC5B,MAAM,eAAe,KAAK,yBAAyB,OAAO;EAE1D,QAAQ,OAAO,QACb,qCAAqC,KAAK,KAAK,SAAS,QAAQ,MAAM,eAAe,QAAQ,KAAK,OAAO,aAAa,QAAQ,aAAa,WAC3I;GAAE,UAAU,KAAK;GAAM,OAAO,QAAQ;EAAM,CAC9C;EAEA,MAAM,KAAK,KAAK,WAAW;EAC3B,MAAM,QAAQ;GACZ,SAAS;GACT,OAAO,QAAQ;GACf,OAAO;IACL,cAAc;IACd,kBAAkB;IAClB,aAAa;GACf;EACF;EAEA,IAAI;GACF,WAAW,MAAM,OAAO,KAAK,YAAY,WACvC,KAAK,iBAAiB,SAAS,YAAY,CAC7C,GACE,KAAK,MAAM,SAAS,qBAAqB,GAAwB,GAAG;IAIlE,0BAA0B,OAAO,KAAK;IAKtC,IAAI,MAAM,SAAS,UAAU,cAC3B,MAAM;KACJ,MAAM,UAAU;KAChB,MAAM;KACN,OAAO;MACL;MACA,OAAO,MAAM;MACb,SAAS,MAAM;MACf,OAAO,MAAM;KACf;KACA,WAAW,KAAK,IAAI;IACtB;IAGF,MAAM;GACR;EAEJ,SAAS,OAAgB;GACvB,QAAQ,OAAO,OAAO,GAAG,KAAK,KAAK,yBAAyB;IAC1D,OAAO,kBAAkB,OAAO,GAAG,KAAK,KAAK,wBAAwB;IACrE,QAAQ,GAAG,KAAK,KAAK;GACvB,CAAC;GACD,MAAM;EACR;CACF;;;;;;;CAQA,iBACE,SACA,cAC+B;EAQ/B,IAAI,UAAmC,EACrC,GAAI,QAAQ,aACd;EACA,UAAU,wBAAwB,KAAK,MAAM,IAAK,OAAO;EAMzD,IAAI,QAAQ,cAAc,KAAA,GAAW;GACnC,IAAI,CAAC,wBAAwB,KAAK,IAAI,GACpC,QAAQ,OAAO,KACb,wBAAwB,QAAQ,UAAU,iEAAiE,KAAK,KAAK,oLACrH,EAAE,UAAU,KAAK,KAAK,CACxB;GAEF,UAAU,eAAe,KAAK,MAAM,QAAQ,WAAW,OAAO;EAChE;EACA,MAAM,eAAe;EAErB,OAAO;GACL,OAAO,QAAQ;GACf,UAAU,CAAC;IAAE,MAAM;IAAQ,SAAS,QAAQ;GAAK,CAAC;GAClD,eAAe,CAAC,YAAY;GAC5B;GACA,QAAQ,QAAQ;GAIhB,GAAI,QAAQ,UAAU,KAAA,IAAY,EAAE,OAAO,QAAQ,MAAM,IAAI,CAAC;GAC9D,GAAI,QAAQ,aAAa,KAAA,IAAY,EAAE,UAAU,QAAQ,SAAS,IAAI,CAAC;EACzE;CACF;CAEA,yBACE,SACQ;EACR,IAAI,SAAS;EAEb,QAAQ,QAAQ,OAAhB;GACE,KAAK;IACH,UAAU;IACV;GACF,KAAK;IACH,UAAU;IACV;GACF,KAAK;IACH,UAAU;IACV;GACF,KAAK,KAAA;IACH,UAAU;IACV;GACF,SACE,UAAU;EACd;EAEA,IAAI,QAAQ,SAAS,QAAQ,MAAM,SAAS,GAC1C,UAAU,mCAAmC,QAAQ,MAAM,KAAK,IAAI,EAAE;EAGxE,IAAI,QAAQ,WACV,UAAU,0BAA0B,QAAQ,UAAU;EAGxD,OAAO;CACT;AACF"}

package/dist/esm/adapter-internals.d.ts

@@ -15,3 +15,5 @@ export type { PendingInterruptResumeRecord } from './interrupt-resume.js';
 export { assertUniqueToolNames, DuplicateToolNameError, } from './activities/chat/tools/unique-tool-names.js';
 export { appendOutputSchemaInstruction, parseJsonFromAssistantText, } from './utilities/structured-output-text.js';
 export { structuredOutputCompleteChunk, structuredOutputStartChunk, } from './utilities/structured-output-events.js';
+export { tanstackMetadata } from './utilities/merge-metadata.js';
+export { isSpecTopLevelKey } from './utilities/spec-event-keys.js';

package/dist/esm/adapter-internals.js

@@ -1,4 +1,6 @@
 import { toRunErrorPayload, toRunErrorRawEvent } from "./activities/error-payload.js";
+import { tanstackMetadata } from "./utilities/merge-metadata.js";
+import { isSpecTopLevelKey } from "./utilities/spec-event-keys.js";
 import { InternalLogger } from "./logger/internal-logger.js";
 import { resolveDebugOption } from "./logger/resolve.js";
 import { readInterruptBinding, validateInterruptResumeBatch } from "./interrupt-resume.js";
@@ -10,4 +12,4 @@ import { GenericInterruptDefinitionRegistryCapability, getGenericInterruptDefini
 import { PendingTurnCapability, getPendingTurn, providePendingTurn } from "./activities/chat/middleware/pending-turn.js";
 import { appendOutputSchemaInstruction, parseJsonFromAssistantText } from "./utilities/structured-output-text.js";
 import { structuredOutputCompleteChunk, structuredOutputStartChunk } from "./utilities/structured-output-events.js";
-export { DuplicateToolNameError, GenericInterruptDefinitionRegistryCapability, InternalLogger, PendingTurnCapability, RunDisconnectCapability, SandboxRuntimeCapability, appendOutputSchemaInstruction, assertUniqueToolNames, createInterruptBinding, getGenericInterruptDefinitionRegistry, getInterruptRequestInput, getPendingTurn, getRunDisconnect, getSandboxRuntime, parseJsonFromAssistantText, provideGenericInterruptDefinitionRegistry, providePendingTurn, provideRunDisconnect, provideSandboxRuntime, readInterruptBinding, rehydrateInterruptRequest, resolveDebugOption, structuredOutputCompleteChunk, structuredOutputStartChunk, toRunErrorPayload, toRunErrorRawEvent, validateInterruptResumeBatch };
+export { DuplicateToolNameError, GenericInterruptDefinitionRegistryCapability, InternalLogger, PendingTurnCapability, RunDisconnectCapability, SandboxRuntimeCapability, appendOutputSchemaInstruction, assertUniqueToolNames, createInterruptBinding, getGenericInterruptDefinitionRegistry, getInterruptRequestInput, getPendingTurn, getRunDisconnect, getSandboxRuntime, isSpecTopLevelKey, parseJsonFromAssistantText, provideGenericInterruptDefinitionRegistry, providePendingTurn, provideRunDisconnect, provideSandboxRuntime, readInterruptBinding, rehydrateInterruptRequest, resolveDebugOption, structuredOutputCompleteChunk, structuredOutputStartChunk, tanstackMetadata, toRunErrorPayload, toRunErrorRawEvent, validateInterruptResumeBatch };

package/dist/esm/byok/define-provider.d.ts

@@ -0,0 +1,23 @@
+/**
+ * A BYOK provider declared by an adapter. `id` is the `x-byok-<id>` slug and
+ * is required — `{ id?: string }` is not a {@link ByokProvider}.
+ */
+export interface ByokProvider<TId extends string = string> {
+    readonly id: TId;
+    readonly label: string;
+    /**
+     * Env var names the relay may read. Names only — never put `process.env`
+     * values here. This object is imported on the client.
+     */
+    readonly env?: ReadonlyArray<string>;
+}
+/**
+ * Input for {@link defineByokProvider}. `id` cannot be optional: if `TId`
+ * includes `undefined`, `id` becomes `never` and the object is unassignable.
+ */
+export type ByokProviderInit<TId extends string> = {
+    readonly id: undefined extends TId ? never : TId;
+    readonly label: string;
+    readonly env?: string | ReadonlyArray<string>;
+};
+export declare function defineByokProvider<const TId extends string>(provider: ByokProviderInit<TId>): ByokProvider<TId>;

package/dist/esm/byok/define-provider.js

@@ -0,0 +1,19 @@
+import { isProviderId } from "./providers.js";
+//#region src/byok/define-provider.ts
+function normalizeEnv(env) {
+	if (env === void 0) return void 0;
+	return typeof env === "string" ? [env] : env;
+}
+function defineByokProvider(provider) {
+	if (!isProviderId(provider.id)) throw new Error(`Invalid BYOK provider id: ${String(provider.id)}`);
+	const env = normalizeEnv(provider.env);
+	return {
+		id: provider.id,
+		label: provider.label,
+		...env ? { env } : {}
+	};
+}
+//#endregion
+export { defineByokProvider };
+
+//# sourceMappingURL=define-provider.js.map

package/dist/esm/byok/define-provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"define-provider.js","names":[],"sources":["../../../src/byok/define-provider.ts"],"sourcesContent":["import { isProviderId } from './providers'\n\n/**\n * A BYOK provider declared by an adapter. `id` is the `x-byok-<id>` slug and\n * is required — `{ id?: string }` is not a {@link ByokProvider}.\n */\nexport interface ByokProvider<TId extends string = string> {\n  readonly id: TId\n  readonly label: string\n  /**\n   * Env var names the relay may read. Names only — never put `process.env`\n   * values here. This object is imported on the client.\n   */\n  readonly env?: ReadonlyArray<string>\n}\n\n/**\n * Input for {@link defineByokProvider}. `id` cannot be optional: if `TId`\n * includes `undefined`, `id` becomes `never` and the object is unassignable.\n */\nexport type ByokProviderInit<TId extends string> = {\n  readonly id: undefined extends TId ? never : TId\n  readonly label: string\n  readonly env?: string | ReadonlyArray<string>\n}\n\nfunction normalizeEnv(\n  env: string | ReadonlyArray<string> | undefined,\n): ReadonlyArray<string> | undefined {\n  if (env === undefined) return undefined\n  return typeof env === 'string' ? [env] : env\n}\n\nexport function defineByokProvider<const TId extends string>(\n  provider: ByokProviderInit<TId>,\n): ByokProvider<TId> {\n  if (!isProviderId(provider.id)) {\n    throw new Error(`Invalid BYOK provider id: ${String(provider.id)}`)\n  }\n  const env = normalizeEnv(provider.env)\n  return {\n    id: provider.id,\n    label: provider.label,\n    ...(env ? { env } : {}),\n  }\n}\n"],"mappings":";;AA0BA,SAAS,aACP,KACmC;CACnC,IAAI,QAAQ,KAAA,GAAW,OAAO,KAAA;CAC9B,OAAO,OAAO,QAAQ,WAAW,CAAC,GAAG,IAAI;AAC3C;AAEA,SAAgB,mBACd,UACmB;CACnB,IAAI,CAAC,aAAa,SAAS,EAAE,GAC3B,MAAM,IAAI,MAAM,6BAA6B,OAAO,SAAS,EAAE,GAAG;CAEpE,MAAM,MAAM,aAAa,SAAS,GAAG;CACrC,OAAO;EACL,IAAI,SAAS;EACb,OAAO,SAAS;EAChB,GAAI,MAAM,EAAE,IAAI,IAAI,CAAC;CACvB;AACF"}

package/dist/esm/byok/errors.d.ts

@@ -0,0 +1,13 @@
+import { ProviderId } from './providers.js';
+export declare class ByokMissingError extends Error {
+    readonly provider: ProviderId;
+    constructor(provider: ProviderId);
+}
+export declare class ByokBlockedError extends Error {
+    readonly provider: ProviderId;
+    readonly reason: 'missing' | 'locked';
+    constructor(provider: ProviderId, reason: 'missing' | 'locked');
+}
+export declare class ByokUnresolvedProviderError extends Error {
+    constructor();
+}

package/dist/esm/byok/errors.js

@@ -0,0 +1,29 @@
+//#region src/byok/errors.ts
+var ByokMissingError = class extends Error {
+	provider;
+	constructor(provider) {
+		super(`Missing ${provider} API key`);
+		this.name = "ByokMissingError";
+		this.provider = provider;
+	}
+};
+var ByokBlockedError = class extends Error {
+	provider;
+	reason;
+	constructor(provider, reason) {
+		super(reason === "locked" ? `${provider} key is locked` : `Missing ${provider} API key`);
+		this.name = "ByokBlockedError";
+		this.provider = provider;
+		this.reason = reason;
+	}
+};
+var ByokUnresolvedProviderError = class extends Error {
+	constructor() {
+		super("BYOK is enabled but no provider slug was resolved. Pass byokProvider or forwardedProps.provider.");
+		this.name = "ByokUnresolvedProviderError";
+	}
+};
+//#endregion
+export { ByokBlockedError, ByokMissingError, ByokUnresolvedProviderError };
+
+//# sourceMappingURL=errors.js.map

package/dist/esm/byok/errors.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"errors.js","names":[],"sources":["../../../src/byok/errors.ts"],"sourcesContent":["import type { ProviderId } from './providers'\n\nexport class ByokMissingError extends Error {\n  readonly provider: ProviderId\n\n  constructor(provider: ProviderId) {\n    super(`Missing ${provider} API key`)\n    this.name = 'ByokMissingError'\n    this.provider = provider\n  }\n}\n\nexport class ByokBlockedError extends Error {\n  readonly provider: ProviderId\n  readonly reason: 'missing' | 'locked'\n\n  constructor(provider: ProviderId, reason: 'missing' | 'locked') {\n    super(\n      reason === 'locked'\n        ? `${provider} key is locked`\n        : `Missing ${provider} API key`,\n    )\n    this.name = 'ByokBlockedError'\n    this.provider = provider\n    this.reason = reason\n  }\n}\n\nexport class ByokUnresolvedProviderError extends Error {\n  constructor() {\n    super(\n      'BYOK is enabled but no provider slug was resolved. Pass byokProvider or forwardedProps.provider.',\n    )\n    this.name = 'ByokUnresolvedProviderError'\n  }\n}\n"],"mappings":";AAEA,IAAa,mBAAb,cAAsC,MAAM;CAC1C;CAEA,YAAY,UAAsB;EAChC,MAAM,WAAW,SAAS,SAAS;EACnC,KAAK,OAAO;EACZ,KAAK,WAAW;CAClB;AACF;AAEA,IAAa,mBAAb,cAAsC,MAAM;CAC1C;CACA;CAEA,YAAY,UAAsB,QAA8B;EAC9D,MACE,WAAW,WACP,GAAG,SAAS,kBACZ,WAAW,SAAS,SAC1B;EACA,KAAK,OAAO;EACZ,KAAK,WAAW;EAChB,KAAK,SAAS;CAChB;AACF;AAEA,IAAa,8BAAb,cAAiD,MAAM;CACrD,cAAc;EACZ,MACE,kGACF;EACA,KAAK,OAAO;CACd;AACF"}

package/dist/esm/byok/get-key.d.ts

@@ -0,0 +1,10 @@
+import { ByokProvider } from './define-provider.js';
+import { ProviderId } from './providers.js';
+/**
+ * Read a key on the relay. Import from `@tanstack/ai/byok/server` so this
+ * `process.env` access is not in the client graph.
+ *
+ * The header wins. A {@link ByokProvider} then tries `provider.env` in order.
+ * A slug is header-only.
+ */
+export declare function getByokKey(request: Request, provider: ProviderId | ByokProvider): string | null;

package/dist/esm/byok/get-key.js

@@ -0,0 +1,26 @@
+import { byokHeaderName, resolveProviderId } from "./providers.js";
+//#region src/byok/get-key.ts
+/**
+* Read a key on the relay. Import from `@tanstack/ai/byok/server` so this
+* `process.env` access is not in the client graph.
+*
+* The header wins. A {@link ByokProvider} then tries `provider.env` in order.
+* A slug is header-only.
+*/
+function getByokKey(request, provider) {
+	const value = request.headers.get(byokHeaderName(resolveProviderId(provider)));
+	if (typeof value === "string") {
+		const trimmed = value.trim();
+		if (trimmed.length > 0) return trimmed;
+	}
+	if (typeof provider === "string") return null;
+	for (const name of provider.env ?? []) {
+		const envValue = process.env[name];
+		if (typeof envValue === "string" && envValue.length > 0) return envValue;
+	}
+	return null;
+}
+//#endregion
+export { getByokKey };
+
+//# sourceMappingURL=get-key.js.map

package/dist/esm/byok/get-key.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"get-key.js","names":[],"sources":["../../../src/byok/get-key.ts"],"sourcesContent":["import { byokHeaderName, resolveProviderId } from './providers'\nimport type { ByokProvider } from './define-provider'\nimport type { ProviderId } from './providers'\n\n/**\n * Read a key on the relay. Import from `@tanstack/ai/byok/server` so this\n * `process.env` access is not in the client graph.\n *\n * The header wins. A {@link ByokProvider} then tries `provider.env` in order.\n * A slug is header-only.\n */\nexport function getByokKey(\n  request: Request,\n  provider: ProviderId | ByokProvider,\n): string | null {\n  const value = request.headers.get(byokHeaderName(resolveProviderId(provider)))\n  if (typeof value === 'string') {\n    const trimmed = value.trim()\n    if (trimmed.length > 0) return trimmed\n  }\n  if (typeof provider === 'string') return null\n  for (const name of provider.env ?? []) {\n    const envValue = process.env[name]\n    if (typeof envValue === 'string' && envValue.length > 0) return envValue\n  }\n  return null\n}\n"],"mappings":";;;;;;;;;AAWA,SAAgB,WACd,SACA,UACe;CACf,MAAM,QAAQ,QAAQ,QAAQ,IAAI,eAAe,kBAAkB,QAAQ,CAAC,CAAC;CAC7E,IAAI,OAAO,UAAU,UAAU;EAC7B,MAAM,UAAU,MAAM,KAAK;EAC3B,IAAI,QAAQ,SAAS,GAAG,OAAO;CACjC;CACA,IAAI,OAAO,aAAa,UAAU,OAAO;CACzC,KAAK,MAAM,QAAQ,SAAS,OAAO,CAAC,GAAG;EACrC,MAAM,WAAW,QAAQ,IAAI;EAC7B,IAAI,OAAO,aAAa,YAAY,SAAS,SAAS,GAAG,OAAO;CAClE;CACA,OAAO;AACT"}

package/dist/esm/byok/missing.d.ts

@@ -0,0 +1,11 @@
+import { ByokProvider } from './define-provider.js';
+import { ProviderId } from './providers.js';
+export interface ByokMissingBody {
+    error: {
+        type: 'byok_missing';
+        provider: ProviderId;
+        message: string;
+    };
+}
+export declare function isByokMissingBody(value: unknown): value is ByokMissingBody;
+export declare function byokMissing(provider: ProviderId | ByokProvider): Response;

package/dist/esm/byok/missing.js

@@ -0,0 +1,30 @@
+import { isProviderId, resolveProviderId } from "./providers.js";
+//#region src/byok/missing.ts
+function isByokMissingBody(value) {
+	if (typeof value !== "object" || value === null) return false;
+	if (!("error" in value)) return false;
+	const error = value.error;
+	if (typeof error !== "object" || error === null) return false;
+	if (!("type" in error) || error.type !== "byok_missing") return false;
+	if (!("provider" in error) || typeof error.provider !== "string") return false;
+	if (!isProviderId(error.provider)) return false;
+	if (!("message" in error) || typeof error.message !== "string") return false;
+	return true;
+}
+function byokMissing(provider) {
+	const id = resolveProviderId(provider);
+	if (!isProviderId(id)) throw new Error(`Invalid BYOK provider id: ${id}`);
+	const body = { error: {
+		type: "byok_missing",
+		provider: id,
+		message: `Missing ${id} API key`
+	} };
+	return new Response(JSON.stringify(body), {
+		status: 401,
+		headers: { "content-type": "application/json" }
+	});
+}
+//#endregion
+export { byokMissing, isByokMissingBody };
+
+//# sourceMappingURL=missing.js.map

package/dist/esm/byok/missing.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"missing.js","names":[],"sources":["../../../src/byok/missing.ts"],"sourcesContent":["import { isProviderId, resolveProviderId } from './providers'\nimport type { ByokProvider } from './define-provider'\nimport type { ProviderId } from './providers'\n\nexport interface ByokMissingBody {\n  error: {\n    type: 'byok_missing'\n    provider: ProviderId\n    message: string\n  }\n}\n\nexport function isByokMissingBody(value: unknown): value is ByokMissingBody {\n  if (typeof value !== 'object' || value === null) return false\n  if (!('error' in value)) return false\n  const error = value.error\n  if (typeof error !== 'object' || error === null) return false\n  if (!('type' in error) || error.type !== 'byok_missing') return false\n  if (!('provider' in error) || typeof error.provider !== 'string') {\n    return false\n  }\n  if (!isProviderId(error.provider)) return false\n  if (!('message' in error) || typeof error.message !== 'string') return false\n  return true\n}\n\nexport function byokMissing(provider: ProviderId | ByokProvider): Response {\n  const id = resolveProviderId(provider)\n  if (!isProviderId(id)) {\n    throw new Error(`Invalid BYOK provider id: ${id}`)\n  }\n  const body: ByokMissingBody = {\n    error: {\n      type: 'byok_missing',\n      provider: id,\n      message: `Missing ${id} API key`,\n    },\n  }\n  return new Response(JSON.stringify(body), {\n    status: 401,\n    headers: { 'content-type': 'application/json' },\n  })\n}\n"],"mappings":";;AAYA,SAAgB,kBAAkB,OAA0C;CAC1E,IAAI,OAAO,UAAU,YAAY,UAAU,MAAM,OAAO;CACxD,IAAI,EAAE,WAAW,QAAQ,OAAO;CAChC,MAAM,QAAQ,MAAM;CACpB,IAAI,OAAO,UAAU,YAAY,UAAU,MAAM,OAAO;CACxD,IAAI,EAAE,UAAU,UAAU,MAAM,SAAS,gBAAgB,OAAO;CAChE,IAAI,EAAE,cAAc,UAAU,OAAO,MAAM,aAAa,UACtD,OAAO;CAET,IAAI,CAAC,aAAa,MAAM,QAAQ,GAAG,OAAO;CAC1C,IAAI,EAAE,aAAa,UAAU,OAAO,MAAM,YAAY,UAAU,OAAO;CACvE,OAAO;AACT;AAEA,SAAgB,YAAY,UAA+C;CACzE,MAAM,KAAK,kBAAkB,QAAQ;CACrC,IAAI,CAAC,aAAa,EAAE,GAClB,MAAM,IAAI,MAAM,6BAA6B,IAAI;CAEnD,MAAM,OAAwB,EAC5B,OAAO;EACL,MAAM;EACN,UAAU;EACV,SAAS,WAAW,GAAG;CACzB,EACF;CACA,OAAO,IAAI,SAAS,KAAK,UAAU,IAAI,GAAG;EACxC,QAAQ;EACR,SAAS,EAAE,gBAAgB,mBAAmB;CAChD,CAAC;AACH"}

package/dist/esm/byok/providers.d.ts

@@ -0,0 +1,13 @@
+/**
+ * Provider ids are open slugs, not a closed catalog. `@tanstack/ai` does not
+ * list adapters. Any matching string is a valid id and becomes `x-byok-<id>`.
+ */
+export type ProviderId = string;
+/** `[a-z][a-z0-9-]{0,63}` — lowercase, no underscores, max 64 chars. */
+export declare const BYOK_PROVIDER_ID_PATTERN: RegExp;
+export declare const BYOK_HEADER_PREFIX = "x-byok-";
+export declare function isProviderId(value: unknown): value is ProviderId;
+export declare function resolveProviderId(provider: string | {
+    readonly id: string;
+}): string;
+export declare function byokHeaderName(provider: string): string;

package/dist/esm/byok/providers.js

@@ -0,0 +1,18 @@
+//#region src/byok/providers.ts
+/** `[a-z][a-z0-9-]{0,63}` — lowercase, no underscores, max 64 chars. */
+var BYOK_PROVIDER_ID_PATTERN = /^[a-z][a-z0-9-]{0,63}$/;
+var BYOK_HEADER_PREFIX = "x-byok-";
+function isProviderId(value) {
+	return typeof value === "string" && BYOK_PROVIDER_ID_PATTERN.test(value);
+}
+function resolveProviderId(provider) {
+	return typeof provider === "string" ? provider : provider.id;
+}
+function byokHeaderName(provider) {
+	if (!isProviderId(provider)) throw new Error(`Invalid BYOK provider id: ${provider}`);
+	return `${BYOK_HEADER_PREFIX}${provider}`;
+}
+//#endregion
+export { BYOK_HEADER_PREFIX, BYOK_PROVIDER_ID_PATTERN, byokHeaderName, isProviderId, resolveProviderId };
+
+//# sourceMappingURL=providers.js.map

package/dist/esm/byok/providers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"providers.js","names":[],"sources":["../../../src/byok/providers.ts"],"sourcesContent":["/**\n * Provider ids are open slugs, not a closed catalog. `@tanstack/ai` does not\n * list adapters. Any matching string is a valid id and becomes `x-byok-<id>`.\n */\nexport type ProviderId = string\n\n/** `[a-z][a-z0-9-]{0,63}` — lowercase, no underscores, max 64 chars. */\nexport const BYOK_PROVIDER_ID_PATTERN = /^[a-z][a-z0-9-]{0,63}$/\n\nexport const BYOK_HEADER_PREFIX = 'x-byok-'\n\nexport function isProviderId(value: unknown): value is ProviderId {\n  return typeof value === 'string' && BYOK_PROVIDER_ID_PATTERN.test(value)\n}\n\nexport function resolveProviderId(\n  provider: string | { readonly id: string },\n): string {\n  return typeof provider === 'string' ? provider : provider.id\n}\n\nexport function byokHeaderName(provider: string): string {\n  if (!isProviderId(provider)) {\n    throw new Error(`Invalid BYOK provider id: ${provider}`)\n  }\n  return `${BYOK_HEADER_PREFIX}${provider}`\n}\n"],"mappings":";;AAOA,IAAa,2BAA2B;AAExC,IAAa,qBAAqB;AAElC,SAAgB,aAAa,OAAqC;CAChE,OAAO,OAAO,UAAU,YAAY,yBAAyB,KAAK,KAAK;AACzE;AAEA,SAAgB,kBACd,UACQ;CACR,OAAO,OAAO,aAAa,WAAW,WAAW,SAAS;AAC5D;AAEA,SAAgB,eAAe,UAA0B;CACvD,IAAI,CAAC,aAAa,QAAQ,GACxB,MAAM,IAAI,MAAM,6BAA6B,UAAU;CAEzD,OAAO,GAAG,qBAAqB;AACjC"}

package/dist/esm/byok/scrub.d.ts

@@ -0,0 +1,2 @@
+export declare function maskKey(key: string): string;
+export declare function scrubSecrets(input: string, secrets: ReadonlyArray<string>): string;

package/dist/esm/byok/scrub.js

@@ -0,0 +1,17 @@
+//#region src/byok/scrub.ts
+function maskKey(key) {
+	if (key.length <= 4) return "••";
+	return key.slice(-4);
+}
+function scrubSecrets(input, secrets) {
+	let next = input;
+	for (const secret of secrets) {
+		if (secret.length === 0) continue;
+		next = next.split(secret).join("[redacted]");
+	}
+	return next;
+}
+//#endregion
+export { maskKey, scrubSecrets };
+
+//# sourceMappingURL=scrub.js.map

package/dist/esm/byok/scrub.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"scrub.js","names":[],"sources":["../../../src/byok/scrub.ts"],"sourcesContent":["export function maskKey(key: string): string {\n  if (key.length <= 4) return '••'\n  return key.slice(-4)\n}\n\nexport function scrubSecrets(\n  input: string,\n  secrets: ReadonlyArray<string>,\n): string {\n  let next = input\n  for (const secret of secrets) {\n    if (secret.length === 0) continue\n    next = next.split(secret).join('[redacted]')\n  }\n  return next\n}\n"],"mappings":";AAAA,SAAgB,QAAQ,KAAqB;CAC3C,IAAI,IAAI,UAAU,GAAG,OAAO;CAC5B,OAAO,IAAI,MAAM,EAAE;AACrB;AAEA,SAAgB,aACd,OACA,SACQ;CACR,IAAI,OAAO;CACX,KAAK,MAAM,UAAU,SAAS;EAC5B,IAAI,OAAO,WAAW,GAAG;EACzB,OAAO,KAAK,MAAM,MAAM,CAAC,CAAC,KAAK,YAAY;CAC7C;CACA,OAAO;AACT"}

package/dist/esm/byok/server.d.ts

@@ -0,0 +1,3 @@
+export { getByokKey } from './get-key.js';
+export { byokMissing, isByokMissingBody } from './missing.js';
+export type { ByokMissingBody } from './missing.js';

package/dist/esm/byok/server.js

@@ -0,0 +1,3 @@
+import { byokMissing, isByokMissingBody } from "./missing.js";
+import { getByokKey } from "./get-key.js";
+export { byokMissing, getByokKey, isByokMissingBody };

package/dist/esm/byok.d.ts

@@ -0,0 +1,8 @@
+export { BYOK_PROVIDER_ID_PATTERN, BYOK_HEADER_PREFIX, byokHeaderName, isProviderId, } from './byok/providers.js';
+export type { ProviderId } from './byok/providers.js';
+export { defineByokProvider } from './byok/define-provider.js';
+export type { ByokProvider, ByokProviderInit } from './byok/define-provider.js';
+export { isByokMissingBody, byokMissing } from './byok/missing.js';
+export type { ByokMissingBody } from './byok/missing.js';
+export { ByokMissingError, ByokBlockedError, ByokUnresolvedProviderError, } from './byok/errors.js';
+export { maskKey, scrubSecrets } from './byok/scrub.js';

package/dist/esm/byok.js

@@ -0,0 +1,6 @@
+import { BYOK_HEADER_PREFIX, BYOK_PROVIDER_ID_PATTERN, byokHeaderName, isProviderId } from "./byok/providers.js";
+import { defineByokProvider } from "./byok/define-provider.js";
+import { byokMissing, isByokMissingBody } from "./byok/missing.js";
+import { ByokBlockedError, ByokMissingError, ByokUnresolvedProviderError } from "./byok/errors.js";
+import { maskKey, scrubSecrets } from "./byok/scrub.js";
+export { BYOK_HEADER_PREFIX, BYOK_PROVIDER_ID_PATTERN, ByokBlockedError, ByokMissingError, ByokUnresolvedProviderError, byokHeaderName, byokMissing, defineByokProvider, isByokMissingBody, isProviderId, maskKey, scrubSecrets };

package/dist/esm/client.d.ts

@@ -58,8 +58,15 @@ export { convertMessagesToModelMessages, generateMessageId, modelMessageToUIMess
 export { BatchStrategy, CompositeStrategy, defaultJSONParser, ImmediateStrategy, parsePartialJSON, PartialJSONParser, PunctuationStrategy, StreamProcessor, WordBoundaryStrategy, } from './activities/chat/stream/index.js';
 export type { ChunkRecording, ChunkStrategy, InternalToolCallState, JSONParser, ProcessorResult, ProcessorState, StreamProcessorEvents, StreamProcessorOptions, ToolCallState, ToolResultState, } from './activities/chat/stream/index.js';
 export { uiMessagesToWire } from './utilities/ag-ui-wire.js';
+export { mergeMetadata, tanstackMetadata, withTanstackMetadata, } from './utilities/merge-metadata.js';
+export { fromSpecTokenUsage, toSpecTokenUsage } from './utilities/ag-ui-usage.js';
+export type { SpecTokenUsage } from './utilities/ag-ui-usage.js';
+export { normalizeStreamChunk } from './utilities/normalize-stream-chunk.js';
+export { restoreInboundChunk } from './utilities/restore-inbound-chunk.js';
+export type { AdapterYieldChunk } from './utilities/adapter-yield-chunk.js';
+export { getChunkRunId, getChunkThreadId } from './utilities/chunk-ids.js';
 export type { WireMessage } from './utilities/ag-ui-wire.js';
-export type { AudioPart, ContentPart, ContentPartDataSource, ContentPartSource, ContentPartUrlSource, CustomEvent, DocumentPart, ImagePart, MediaInputMetadata, MediaInputRole, MediaPrompt, MediaPromptPart, MessagePart, ModelMessage, PersistedArtifactActivity, PersistedArtifactRef, PersistedArtifactRole, Interrupt, RunAgentResumeItem, RunErrorEvent, RunFinishedEvent, RunFinishedOutcome, SchemaInput, StreamChunk, StructuredOutputPart, TextPart, ThinkingPart, ToolCall, ToolCallPart, ToolResultPart, UIMessage, UIResourcePart, VideoPart, InferSchemaType, } from './types.js';
+export type { AudioPart, ContentPart, ContentPartDataSource, ContentPartSource, ContentPartUrlSource, CustomEvent, DocumentPart, ImagePart, MediaInputMetadata, MediaInputRole, MediaPrompt, MediaPromptPart, MessagePart, ModelMessage, PersistedArtifactActivity, PersistedArtifactRef, PersistedArtifactRole, Interrupt, RunAgentResumeItem, RunErrorEvent, RunFinishedEvent, RunFinishedOutcome, SchemaInput, StreamChunk, StructuredOutputPart, TextPart, TanStackMessageMetadata, TanStackRunMetadata, ThinkingPart, ToolCall, ToolCallPart, ToolResultPart, UIMessage, UIResourcePart, VideoPart, InferSchemaType, } from './types.js';
 export { INTERRUPT_BINDING_VERSION, canonicalizeInterruptResolutions, } from './interrupts.js';
 export { defineInterrupt, hashInterruptDefinitionSchema, INTERRUPT_PAYLOAD_METADATA_KEY, } from './interrupt-definition.js';
 export { INTERRUPT_CONTINUATION_METADATA_KEY, INTERRUPT_CONTINUATION_VERSION, genericInterruptContinuationFromDescriptor, readGenericInterruptContinuation, wrapGenericInterruptContinuation, } from './generic-interrupt-continuation.js';