@tanstack/ai 0.43.1 → 0.44.0

package/dist/esm/activities/index.js

@@ -1,17 +1,21 @@
 import { BaseTextAdapter } from "./chat/adapter.js";
-import { chat, kind as kind$3 } from "./chat/index.js";
+import { chat, kind as kind$5 } from "./chat/index.js";
 import { BaseSummarizeAdapter } from "./summarize/adapter.js";
 import { ChatStreamSummarizeAdapter } from "./summarize/chat-stream-summarize.js";
-import { kind as kind$2, summarize } from "./summarize/index.js";
+import { kind as kind$4, summarize } from "./summarize/index.js";
+import { BaseRerankAdapter } from "./rerank/adapter.js";
+import { createRerankOptions, kind as kind$3, rerank } from "./rerank/index.js";
 import { BaseImageAdapter } from "./generateImage/adapter.js";
-import { generateImage, kind as kind$1 } from "./generateImage/index.js";
+import { generateImage, kind as kind$2 } from "./generateImage/index.js";
 import { BaseAudioAdapter } from "./generateAudio/adapter.js";
 import { generateAudio, kind } from "./generateAudio/index.js";
 import { BaseVideoAdapter } from "./generateVideo/adapter.js";
-import { generateVideo, getVideoJobStatus, kind as kind$6 } from "./generateVideo/index.js";
+import { generateVideo, getVideoJobStatus, kind as kind$8 } from "./generateVideo/index.js";
 import { snapToDurationOption } from "./generateVideo/snap.js";
 import { BaseTTSAdapter } from "./generateSpeech/adapter.js";
-import { generateSpeech, kind as kind$5 } from "./generateSpeech/index.js";
+import { generateSpeech, kind as kind$7 } from "./generateSpeech/index.js";
 import { BaseTranscriptionAdapter } from "./generateTranscription/adapter.js";
-import { generateTranscription, kind as kind$4 } from "./generateTranscription/index.js";
-export { BaseAudioAdapter, BaseImageAdapter, BaseSummarizeAdapter, BaseTTSAdapter, BaseTextAdapter, BaseTranscriptionAdapter, BaseVideoAdapter, ChatStreamSummarizeAdapter, kind as audioKind, chat, generateAudio, generateImage, generateSpeech, generateTranscription, generateVideo, getVideoJobStatus, kind$1 as imageKind, snapToDurationOption, summarize, kind$2 as summarizeKind, kind$3 as textKind, kind$4 as transcriptionKind, kind$5 as ttsKind, kind$6 as videoKind };
+import { generateTranscription, kind as kind$6 } from "./generateTranscription/index.js";
+import { BaseEmbeddingAdapter } from "./embed/adapter.js";
+import { embed, kind as kind$1 } from "./embed/index.js";
+export { BaseAudioAdapter, BaseEmbeddingAdapter, BaseImageAdapter, BaseRerankAdapter, BaseSummarizeAdapter, BaseTTSAdapter, BaseTextAdapter, BaseTranscriptionAdapter, BaseVideoAdapter, ChatStreamSummarizeAdapter, kind as audioKind, chat, createRerankOptions, embed, kind$1 as embeddingKind, generateAudio, generateImage, generateSpeech, generateTranscription, generateVideo, getVideoJobStatus, kind$2 as imageKind, rerank, kind$3 as rerankKind, snapToDurationOption, summarize, kind$4 as summarizeKind, kind$5 as textKind, kind$6 as transcriptionKind, kind$7 as ttsKind, kind$8 as videoKind };

package/dist/esm/activities/middleware/types.d.ts

@@ -11,7 +11,7 @@ import { TokenUsage } from '../../types.js';
  * `'summarize'` produces text, not media, so it has no artifacts — a
  * persistence middleware stores its run record and result and nothing else.
  */
-export type GenerationActivity = 'chat' | 'image' | 'video' | 'audio' | 'tts' | 'transcription' | 'summarize';
+export type GenerationActivity = 'chat' | 'image' | 'video' | 'audio' | 'tts' | 'transcription' | 'embedding' | 'rerank' | 'summarize';
 /**
  * Stable context passed to every {@link GenerationMiddleware} hook. Created
  * once per activity call and shared across the hooks of that call.

package/dist/esm/activities/rerank/adapter.d.ts

@@ -0,0 +1,63 @@
+import { RerankAdapterResult, RerankOptions } from '../../types.js';
+/**
+ * Configuration for rerank adapter instances
+ */
+export interface RerankAdapterConfig {
+    apiKey?: string;
+    baseUrl?: string;
+    timeout?: number;
+    headers?: Record<string, string>;
+}
+/**
+ * Rerank adapter interface with pre-resolved generics.
+ *
+ * An adapter is created by a provider function: `provider('model')` → `adapter`
+ * All type resolution happens at the provider call site, not in this interface.
+ *
+ * Generic parameters:
+ * - TModel: The specific model name (e.g. 'rerank-v3.5')
+ * - TProviderOptions: Provider-specific options (already resolved)
+ */
+export interface RerankAdapter<TModel extends string = string, TProviderOptions extends object = Record<string, unknown>> {
+    /** Discriminator for adapter kind */
+    readonly kind: 'rerank';
+    /** Adapter name identifier */
+    readonly name: string;
+    /** The model this adapter is configured for */
+    readonly model: TModel;
+    /**
+     * @internal Type-only properties for inference. Not assigned at runtime.
+     */
+    '~types': {
+        providerOptions: TProviderOptions;
+    };
+    /**
+     * Rerank the given (pre-serialized) documents against the query, returning
+     * scored indices into `options.documents`. The activity layer maps these
+     * back to the caller's original documents.
+     */
+    rerank: (options: RerankOptions<TProviderOptions>) => Promise<RerankAdapterResult>;
+}
+/**
+ * A RerankAdapter with any/unknown type parameters.
+ * Useful as a constraint in generic functions and interfaces.
+ */
+export type AnyRerankAdapter = RerankAdapter<any, any>;
+/**
+ * Abstract base class for rerank adapters.
+ * Extend this class to implement a rerank adapter for a specific provider.
+ *
+ * Generic parameters match RerankAdapter - all pre-resolved by the provider function.
+ */
+export declare abstract class BaseRerankAdapter<TModel extends string = string, TProviderOptions extends object = Record<string, unknown>> implements RerankAdapter<TModel, TProviderOptions> {
+    readonly kind: "rerank";
+    abstract readonly name: string;
+    readonly model: TModel;
+    '~types': {
+        providerOptions: TProviderOptions;
+    };
+    protected config: RerankAdapterConfig;
+    constructor(config: RerankAdapterConfig | undefined, model: TModel);
+    abstract rerank(options: RerankOptions<TProviderOptions>): Promise<RerankAdapterResult>;
+    protected generateId(): string;
+}

package/dist/esm/activities/rerank/adapter.js

@@ -0,0 +1,23 @@
+//#region src/activities/rerank/adapter.ts
+/**
+* Abstract base class for rerank adapters.
+* Extend this class to implement a rerank adapter for a specific provider.
+*
+* Generic parameters match RerankAdapter - all pre-resolved by the provider function.
+*/
+var BaseRerankAdapter = class {
+	kind = "rerank";
+	model;
+	config;
+	constructor(config = {}, model) {
+		this.config = config;
+		this.model = model;
+	}
+	generateId() {
+		return `${this.name}-${Date.now()}-${Math.random().toString(36).slice(2, 9)}`;
+	}
+};
+//#endregion
+export { BaseRerankAdapter };
+
+//# sourceMappingURL=adapter.js.map

package/dist/esm/activities/rerank/adapter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"adapter.js","names":[],"sources":["../../../../src/activities/rerank/adapter.ts"],"sourcesContent":["import type { RerankAdapterResult, RerankOptions } from '../../types'\n\n/**\n * Configuration for rerank adapter instances\n */\nexport interface RerankAdapterConfig {\n  apiKey?: string\n  baseUrl?: string\n  timeout?: number\n  headers?: Record<string, string>\n}\n\n/**\n * Rerank adapter interface with pre-resolved generics.\n *\n * An adapter is created by a provider function: `provider('model')` → `adapter`\n * All type resolution happens at the provider call site, not in this interface.\n *\n * Generic parameters:\n * - TModel: The specific model name (e.g. 'rerank-v3.5')\n * - TProviderOptions: Provider-specific options (already resolved)\n */\nexport interface RerankAdapter<\n  TModel extends string = string,\n  TProviderOptions extends object = Record<string, unknown>,\n> {\n  /** Discriminator for adapter kind */\n  readonly kind: 'rerank'\n  /** Adapter name identifier */\n  readonly name: string\n  /** The model this adapter is configured for */\n  readonly model: TModel\n\n  /**\n   * @internal Type-only properties for inference. Not assigned at runtime.\n   */\n  '~types': {\n    providerOptions: TProviderOptions\n  }\n\n  /**\n   * Rerank the given (pre-serialized) documents against the query, returning\n   * scored indices into `options.documents`. The activity layer maps these\n   * back to the caller's original documents.\n   */\n  rerank: (\n    options: RerankOptions<TProviderOptions>,\n  ) => Promise<RerankAdapterResult>\n}\n\n/**\n * A RerankAdapter with any/unknown type parameters.\n * Useful as a constraint in generic functions and interfaces.\n */\nexport type AnyRerankAdapter = RerankAdapter<any, any>\n\n/**\n * Abstract base class for rerank adapters.\n * Extend this class to implement a rerank adapter for a specific provider.\n *\n * Generic parameters match RerankAdapter - all pre-resolved by the provider function.\n */\nexport abstract class BaseRerankAdapter<\n  TModel extends string = string,\n  TProviderOptions extends object = Record<string, unknown>,\n> implements RerankAdapter<TModel, TProviderOptions> {\n  readonly kind = 'rerank' as const\n  abstract readonly name: string\n  readonly model: TModel\n\n  // Type-only property - never assigned at runtime\n  declare '~types': {\n    providerOptions: TProviderOptions\n  }\n\n  protected config: RerankAdapterConfig\n\n  constructor(config: RerankAdapterConfig = {}, model: TModel) {\n    this.config = config\n    this.model = model\n  }\n\n  abstract rerank(\n    options: RerankOptions<TProviderOptions>,\n  ): Promise<RerankAdapterResult>\n\n  protected generateId(): string {\n    return `${this.name}-${Date.now()}-${Math.random().toString(36).slice(2, 9)}`\n  }\n}\n"],"mappings":";;;;;;;AA8DA,IAAsB,oBAAtB,MAGqD;CACnD,OAAgB;CAEhB;CAOA;CAEA,YAAY,SAA8B,CAAC,GAAG,OAAe;EAC3D,KAAK,SAAS;EACd,KAAK,QAAQ;CACf;CAMA,aAA+B;EAC7B,OAAO,GAAG,KAAK,KAAK,GAAG,KAAK,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC;CAC5E;AACF"}

package/dist/esm/activities/rerank/index.d.ts

@@ -0,0 +1,92 @@
+import { DebugOption } from '../../logger/types.js';
+import { GenerationMiddleware } from '../middleware/types.js';
+import { RerankAdapter } from './adapter.js';
+import { RerankResult } from '../../types.js';
+/** The adapter kind this activity handles */
+export declare const kind: "rerank";
+/** Extract provider options from a RerankAdapter via ~types */
+export type RerankProviderOptions<TAdapter> = TAdapter extends {
+    '~types': {
+        providerOptions: infer P extends object;
+    };
+} ? P : object;
+/**
+ * Options for the rerank activity. The model is extracted from the adapter's
+ * model property.
+ *
+ * @template TAdapter - The rerank adapter type
+ * @template TDocument - The document element type (string or object)
+ */
+export interface RerankActivityOptions<TAdapter extends RerankAdapter<string, RerankProviderOptions<TAdapter>>, TDocument extends string | object = string> {
+    /** The rerank adapter to use (must be created with a model) */
+    adapter: TAdapter & {
+        kind: typeof kind;
+    };
+    /** The query documents are scored against. */
+    query: string;
+    /**
+     * Documents to rerank. Either strings or JSON-serializable objects — object
+     * documents are serialized with `JSON.stringify` before being sent to the
+     * provider, and the original element (string or object) is returned in the
+     * result, preserving its type.
+     */
+    documents: Array<TDocument>;
+    /** Return only the top N results. */
+    topN?: number;
+    /** Provider-specific options */
+    modelOptions?: RerankProviderOptions<TAdapter>;
+    /** Forwarded to the provider request for cancellation. */
+    abortSignal?: AbortSignal;
+    /**
+     * Observe-only middleware notified on start, usage, success, abort, and
+     * error. Pass `otelMiddleware()` to emit OpenTelemetry spans, or implement
+     * the `GenerationMiddleware` contract for a custom backend.
+     */
+    middleware?: Array<GenerationMiddleware>;
+    /**
+     * Enable debug logging. Pass `true` to enable all categories, `false` to
+     * silence everything including errors, or a `DebugConfig` object for granular
+     * control and/or a custom `Logger`.
+     */
+    debug?: DebugOption;
+}
+/**
+ * Rerank activity - reorders documents by relevance to a query.
+ *
+ * @example Basic reranking
+ * ```ts
+ * import { rerank } from '@tanstack/ai'
+ * import { cohereRerank } from '@tanstack/ai-cohere'
+ *
+ * const { ranking, rerankedDocuments } = await rerank({
+ *   adapter: cohereRerank('rerank-v3.5'),
+ *   query: 'talk about rain',
+ *   documents: ['sunny day at the beach', 'rainy afternoon in the city'],
+ *   topN: 2,
+ * })
+ *
+ * console.log(rerankedDocuments[0]) // 'rainy afternoon in the city'
+ * ```
+ *
+ * @example Reranking object documents
+ * ```ts
+ * const { ranking } = await rerank({
+ *   adapter: cohereRerank('rerank-v3.5'),
+ *   query: 'best laptop for travel',
+ *   documents: [
+ *     { id: 1, text: 'A heavy gaming desktop' },
+ *     { id: 2, text: 'A lightweight ultrabook with all-day battery' },
+ *   ],
+ * })
+ *
+ * // ranking[0].document is the original object, fully typed.
+ * console.log(ranking[0].document.id)
+ * ```
+ */
+export declare function rerank<TAdapter extends RerankAdapter<string, RerankProviderOptions<TAdapter>>, TDocument extends string | object = string>(options: RerankActivityOptions<TAdapter, TDocument>): Promise<RerankResult<TDocument>>;
+/**
+ * Create typed options for the rerank() function without executing.
+ */
+export declare function createRerankOptions<TAdapter extends RerankAdapter<string, RerankProviderOptions<TAdapter>>, TDocument extends string | object = string>(options: RerankActivityOptions<TAdapter, TDocument>): RerankActivityOptions<TAdapter, TDocument>;
+export type { RerankAdapter, RerankAdapterConfig, AnyRerankAdapter, } from './adapter.js';
+export { BaseRerankAdapter } from './adapter.js';

package/dist/esm/activities/rerank/index.js

@@ -0,0 +1,163 @@
+import { isAbortShapedError } from "../error-payload.js";
+import { resolveDebugOption } from "../../logger/resolve.js";
+import { createGenerationContext, runGenerationAbort, runGenerationError, runGenerationFinish, runGenerationStart, runGenerationUsage } from "../middleware/run.js";
+import "./adapter.js";
+import { aiEventClient } from "@tanstack/ai-event-client";
+//#region src/activities/rerank/index.ts
+/**
+* Rerank Activity
+*
+* Reorders a set of documents by semantic relevance to a query.
+* This is a self-contained module with implementation, types, and JSDoc.
+*/
+/** The adapter kind this activity handles */
+var kind = "rerank";
+function createId(prefix) {
+	return `${prefix}-${Date.now()}-${Math.random().toString(36).slice(2, 9)}`;
+}
+/** Serialize a document for the provider. Strings pass through untouched. */
+function serializeDocument(document) {
+	return typeof document === "string" ? document : JSON.stringify(document);
+}
+function isAbortError(error, signal) {
+	if (isAbortShapedError(error)) return true;
+	return error instanceof Error ? false : signal?.aborted === true;
+}
+/**
+* Rerank activity - reorders documents by relevance to a query.
+*
+* @example Basic reranking
+* ```ts
+* import { rerank } from '@tanstack/ai'
+* import { cohereRerank } from '@tanstack/ai-cohere'
+*
+* const { ranking, rerankedDocuments } = await rerank({
+*   adapter: cohereRerank('rerank-v3.5'),
+*   query: 'talk about rain',
+*   documents: ['sunny day at the beach', 'rainy afternoon in the city'],
+*   topN: 2,
+* })
+*
+* console.log(rerankedDocuments[0]) // 'rainy afternoon in the city'
+* ```
+*
+* @example Reranking object documents
+* ```ts
+* const { ranking } = await rerank({
+*   adapter: cohereRerank('rerank-v3.5'),
+*   query: 'best laptop for travel',
+*   documents: [
+*     { id: 1, text: 'A heavy gaming desktop' },
+*     { id: 2, text: 'A lightweight ultrabook with all-day battery' },
+*   ],
+* })
+*
+* // ranking[0].document is the original object, fully typed.
+* console.log(ranking[0].document.id)
+* ```
+*/
+async function rerank(options) {
+	const { adapter, query, documents, topN, modelOptions, abortSignal, middleware } = options;
+	const model = adapter.model;
+	const requestId = createId("rerank");
+	const startTime = Date.now();
+	const logger = resolveDebugOption(options.debug);
+	if (documents.length === 0) throw new Error("rerank() requires at least one document");
+	const mwCtx = createGenerationContext({
+		requestId,
+		activity: "rerank",
+		provider: adapter.name,
+		model,
+		modelOptions,
+		createId
+	});
+	await runGenerationStart(middleware, mwCtx);
+	aiEventClient.emit("rerank:request:started", {
+		requestId,
+		provider: adapter.name,
+		model,
+		documentCount: documents.length,
+		timestamp: startTime
+	});
+	logger.request(`activity=rerank provider=${adapter.name}`, {
+		provider: adapter.name,
+		model,
+		documentCount: documents.length
+	});
+	const serialized = documents.map(serializeDocument);
+	try {
+		const result = await adapter.rerank({
+			model,
+			query,
+			documents: serialized,
+			topN,
+			modelOptions,
+			abortSignal,
+			logger
+		});
+		const ranking = result.ranking.map((r) => {
+			const document = documents[r.index];
+			if (document === void 0) throw new Error(`rerank(): provider ${adapter.name} returned out-of-range index ${r.index}`);
+			return {
+				index: r.index,
+				score: r.score,
+				document
+			};
+		});
+		const rerankedDocuments = ranking.map((r) => r.document);
+		const duration = Date.now() - startTime;
+		aiEventClient.emit("rerank:request:completed", {
+			requestId,
+			provider: adapter.name,
+			model,
+			documentCount: documents.length,
+			resultCount: ranking.length,
+			duration,
+			timestamp: Date.now()
+		});
+		aiEventClient.emit("rerank:usage", {
+			requestId,
+			model,
+			usage: result.usage,
+			timestamp: Date.now()
+		});
+		logger.output(`activity=rerank results=${ranking.length}`, { resultCount: ranking.length });
+		await runGenerationUsage(middleware, mwCtx, result.usage);
+		await runGenerationFinish(middleware, mwCtx, {
+			duration,
+			usage: result.usage
+		});
+		return {
+			id: result.id,
+			model,
+			ranking,
+			rerankedDocuments,
+			usage: result.usage
+		};
+	} catch (error) {
+		const duration = Date.now() - startTime;
+		if (isAbortError(error, abortSignal)) await runGenerationAbort(middleware, mwCtx, {
+			reason: error instanceof Error ? error.message : void 0,
+			duration
+		});
+		else await runGenerationError(middleware, mwCtx, {
+			error,
+			duration
+		});
+		logger.errors("rerank activity failed", {
+			error,
+			source: "rerank"
+		});
+		throw error;
+	}
+}
+/**
+* Create typed options for the rerank() function without executing.
+*/
+function createRerankOptions(options) {
+	return options;
+}
+//#endregion
+export { createRerankOptions, kind, rerank };
+
+//# sourceMappingURL=index.js.map

package/dist/esm/activities/rerank/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","names":[],"sources":["../../../../src/activities/rerank/index.ts"],"sourcesContent":["/**\n * Rerank Activity\n *\n * Reorders a set of documents by semantic relevance to a query.\n * This is a self-contained module with implementation, types, and JSDoc.\n */\n\nimport { aiEventClient } from '@tanstack/ai-event-client'\nimport { resolveDebugOption } from '../../logger/resolve'\nimport { isAbortShapedError } from '../error-payload'\nimport {\n  createGenerationContext,\n  runGenerationAbort,\n  runGenerationError,\n  runGenerationFinish,\n  runGenerationStart,\n  runGenerationUsage,\n} from '../middleware/run'\nimport type { InternalLogger } from '../../logger/internal-logger'\nimport type { DebugOption } from '../../logger/types'\nimport type { GenerationMiddleware } from '../middleware/types'\nimport type { RerankAdapter } from './adapter'\nimport type { RerankResult } from '../../types'\n\n// ===========================\n// Activity Kind\n// ===========================\n\n/** The adapter kind this activity handles */\nexport const kind = 'rerank' as const\n\n// ===========================\n// Type Extraction Helpers\n// ===========================\n\n/** Extract provider options from a RerankAdapter via ~types */\nexport type RerankProviderOptions<TAdapter> = TAdapter extends {\n  '~types': { providerOptions: infer P extends object }\n}\n  ? P\n  : object\n\n// ===========================\n// Activity Options Type\n// ===========================\n\n/**\n * Options for the rerank activity. The model is extracted from the adapter's\n * model property.\n *\n * @template TAdapter - The rerank adapter type\n * @template TDocument - The document element type (string or object)\n */\nexport interface RerankActivityOptions<\n  TAdapter extends RerankAdapter<string, RerankProviderOptions<TAdapter>>,\n  TDocument extends string | object = string,\n> {\n  /** The rerank adapter to use (must be created with a model) */\n  adapter: TAdapter & { kind: typeof kind }\n  /** The query documents are scored against. */\n  query: string\n  /**\n   * Documents to rerank. Either strings or JSON-serializable objects — object\n   * documents are serialized with `JSON.stringify` before being sent to the\n   * provider, and the original element (string or object) is returned in the\n   * result, preserving its type.\n   */\n  documents: Array<TDocument>\n  /** Return only the top N results. */\n  topN?: number\n  /** Provider-specific options */\n  modelOptions?: RerankProviderOptions<TAdapter>\n  /** Forwarded to the provider request for cancellation. */\n  abortSignal?: AbortSignal\n  /**\n   * Observe-only middleware notified on start, usage, success, abort, and\n   * error. Pass `otelMiddleware()` to emit OpenTelemetry spans, or implement\n   * the `GenerationMiddleware` contract for a custom backend.\n   */\n  middleware?: Array<GenerationMiddleware>\n  /**\n   * Enable debug logging. Pass `true` to enable all categories, `false` to\n   * silence everything including errors, or a `DebugConfig` object for granular\n   * control and/or a custom `Logger`.\n   */\n  debug?: DebugOption\n}\n\n// ===========================\n// Helper Functions\n// ===========================\n\nfunction createId(prefix: string): string {\n  return `${prefix}-${Date.now()}-${Math.random().toString(36).slice(2, 9)}`\n}\n\n/** Serialize a document for the provider. Strings pass through untouched. */\nfunction serializeDocument(document: string | object): string {\n  return typeof document === 'string' ? document : JSON.stringify(document)\n}\n\nfunction isAbortError(error: unknown, signal?: AbortSignal): boolean {\n  // Prefer the error's own identity over the signal state. A genuine\n  // cancellation throws an abort-shaped error (DOM `AbortError`, the OpenRouter\n  // SDK's `RequestAbortedError`, …). Classifying on `signal.aborted` alone would\n  // misroute a real failure — e.g. the out-of-range-index throw below — to the\n  // abort hook whenever a shared/long-lived signal happens to already be\n  // aborted, hiding it from `onError` observers.\n  if (isAbortShapedError(error)) return true\n  // Fall back to signal state only for non-Error throws we can't otherwise\n  // identify; a real Error with a non-abort name is never an abort.\n  return error instanceof Error ? false : signal?.aborted === true\n}\n\n// ===========================\n// Activity Implementation\n// ===========================\n\n/**\n * Rerank activity - reorders documents by relevance to a query.\n *\n * @example Basic reranking\n * ```ts\n * import { rerank } from '@tanstack/ai'\n * import { cohereRerank } from '@tanstack/ai-cohere'\n *\n * const { ranking, rerankedDocuments } = await rerank({\n *   adapter: cohereRerank('rerank-v3.5'),\n *   query: 'talk about rain',\n *   documents: ['sunny day at the beach', 'rainy afternoon in the city'],\n *   topN: 2,\n * })\n *\n * console.log(rerankedDocuments[0]) // 'rainy afternoon in the city'\n * ```\n *\n * @example Reranking object documents\n * ```ts\n * const { ranking } = await rerank({\n *   adapter: cohereRerank('rerank-v3.5'),\n *   query: 'best laptop for travel',\n *   documents: [\n *     { id: 1, text: 'A heavy gaming desktop' },\n *     { id: 2, text: 'A lightweight ultrabook with all-day battery' },\n *   ],\n * })\n *\n * // ranking[0].document is the original object, fully typed.\n * console.log(ranking[0].document.id)\n * ```\n */\nexport async function rerank<\n  TAdapter extends RerankAdapter<string, RerankProviderOptions<TAdapter>>,\n  TDocument extends string | object = string,\n>(\n  options: RerankActivityOptions<TAdapter, TDocument>,\n): Promise<RerankResult<TDocument>> {\n  const {\n    adapter,\n    query,\n    documents,\n    topN,\n    modelOptions,\n    abortSignal,\n    middleware,\n  } = options\n  const model = adapter.model\n  const requestId = createId('rerank')\n  const startTime = Date.now()\n  const logger: InternalLogger = resolveDebugOption(options.debug)\n\n  if (documents.length === 0) {\n    throw new Error('rerank() requires at least one document')\n  }\n\n  const mwCtx = createGenerationContext({\n    requestId,\n    // `rerank` joins the GenerationActivity union; otel maps it to its own\n    // gen_ai.operation.name.\n    activity: 'rerank',\n    provider: adapter.name,\n    model,\n    modelOptions,\n    createId,\n  })\n\n  await runGenerationStart(middleware, mwCtx)\n\n  aiEventClient.emit('rerank:request:started', {\n    requestId,\n    provider: adapter.name,\n    model,\n    documentCount: documents.length,\n    timestamp: startTime,\n  })\n\n  logger.request(`activity=rerank provider=${adapter.name}`, {\n    provider: adapter.name,\n    model,\n    documentCount: documents.length,\n  })\n\n  // Serialize once; reuse for the request only. Original documents are mapped\n  // back by index below so the caller's element type is preserved.\n  const serialized = documents.map(serializeDocument)\n\n  try {\n    const result = await adapter.rerank({\n      model,\n      query,\n      documents: serialized,\n      topN,\n      modelOptions,\n      abortSignal,\n      logger,\n    })\n\n    const ranking = result.ranking.map((r) => {\n      const document = documents[r.index]\n      if (document === undefined) {\n        throw new Error(\n          `rerank(): provider ${adapter.name} returned out-of-range index ${r.index}`,\n        )\n      }\n      return { index: r.index, score: r.score, document }\n    })\n    const rerankedDocuments = ranking.map((r) => r.document)\n\n    const duration = Date.now() - startTime\n\n    aiEventClient.emit('rerank:request:completed', {\n      requestId,\n      provider: adapter.name,\n      model,\n      documentCount: documents.length,\n      resultCount: ranking.length,\n      duration,\n      timestamp: Date.now(),\n    })\n\n    aiEventClient.emit('rerank:usage', {\n      requestId,\n      model,\n      usage: result.usage,\n      timestamp: Date.now(),\n    })\n\n    logger.output(`activity=rerank results=${ranking.length}`, {\n      resultCount: ranking.length,\n    })\n\n    await runGenerationUsage(middleware, mwCtx, result.usage)\n    await runGenerationFinish(middleware, mwCtx, {\n      duration,\n      usage: result.usage,\n    })\n\n    return {\n      id: result.id,\n      model,\n      ranking,\n      rerankedDocuments,\n      usage: result.usage,\n    }\n  } catch (error) {\n    const duration = Date.now() - startTime\n    if (isAbortError(error, abortSignal)) {\n      await runGenerationAbort(middleware, mwCtx, {\n        reason: error instanceof Error ? error.message : undefined,\n        duration,\n      })\n    } else {\n      await runGenerationError(middleware, mwCtx, { error, duration })\n    }\n    logger.errors('rerank activity failed', { error, source: 'rerank' })\n    throw error\n  }\n}\n\n// ===========================\n// Options Factory\n// ===========================\n\n/**\n * Create typed options for the rerank() function without executing.\n */\nexport function createRerankOptions<\n  TAdapter extends RerankAdapter<string, RerankProviderOptions<TAdapter>>,\n  TDocument extends string | object = string,\n>(\n  options: RerankActivityOptions<TAdapter, TDocument>,\n): RerankActivityOptions<TAdapter, TDocument> {\n  return options\n}\n\n// Re-export adapter types\nexport type {\n  RerankAdapter,\n  RerankAdapterConfig,\n  AnyRerankAdapter,\n} from './adapter'\nexport { BaseRerankAdapter } from './adapter'\n"],"mappings":";;;;;;;;;;;;;AA6BA,IAAa,OAAO;AA+DpB,SAAS,SAAS,QAAwB;CACxC,OAAO,GAAG,OAAO,GAAG,KAAK,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC;AACzE;;AAGA,SAAS,kBAAkB,UAAmC;CAC5D,OAAO,OAAO,aAAa,WAAW,WAAW,KAAK,UAAU,QAAQ;AAC1E;AAEA,SAAS,aAAa,OAAgB,QAA+B;CAOnE,IAAI,mBAAmB,KAAK,GAAG,OAAO;CAGtC,OAAO,iBAAiB,QAAQ,QAAQ,QAAQ,YAAY;AAC9D;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAuCA,eAAsB,OAIpB,SACkC;CAClC,MAAM,EACJ,SACA,OACA,WACA,MACA,cACA,aACA,eACE;CACJ,MAAM,QAAQ,QAAQ;CACtB,MAAM,YAAY,SAAS,QAAQ;CACnC,MAAM,YAAY,KAAK,IAAI;CAC3B,MAAM,SAAyB,mBAAmB,QAAQ,KAAK;CAE/D,IAAI,UAAU,WAAW,GACvB,MAAM,IAAI,MAAM,yCAAyC;CAG3D,MAAM,QAAQ,wBAAwB;EACpC;EAGA,UAAU;EACV,UAAU,QAAQ;EAClB;EACA;EACA;CACF,CAAC;CAED,MAAM,mBAAmB,YAAY,KAAK;CAE1C,cAAc,KAAK,0BAA0B;EAC3C;EACA,UAAU,QAAQ;EAClB;EACA,eAAe,UAAU;EACzB,WAAW;CACb,CAAC;CAED,OAAO,QAAQ,4BAA4B,QAAQ,QAAQ;EACzD,UAAU,QAAQ;EAClB;EACA,eAAe,UAAU;CAC3B,CAAC;CAID,MAAM,aAAa,UAAU,IAAI,iBAAiB;CAElD,IAAI;EACF,MAAM,SAAS,MAAM,QAAQ,OAAO;GAClC;GACA;GACA,WAAW;GACX;GACA;GACA;GACA;EACF,CAAC;EAED,MAAM,UAAU,OAAO,QAAQ,KAAK,MAAM;GACxC,MAAM,WAAW,UAAU,EAAE;GAC7B,IAAI,aAAa,KAAA,GACf,MAAM,IAAI,MACR,sBAAsB,QAAQ,KAAK,+BAA+B,EAAE,OACtE;GAEF,OAAO;IAAE,OAAO,EAAE;IAAO,OAAO,EAAE;IAAO;GAAS;EACpD,CAAC;EACD,MAAM,oBAAoB,QAAQ,KAAK,MAAM,EAAE,QAAQ;EAEvD,MAAM,WAAW,KAAK,IAAI,IAAI;EAE9B,cAAc,KAAK,4BAA4B;GAC7C;GACA,UAAU,QAAQ;GAClB;GACA,eAAe,UAAU;GACzB,aAAa,QAAQ;GACrB;GACA,WAAW,KAAK,IAAI;EACtB,CAAC;EAED,cAAc,KAAK,gBAAgB;GACjC;GACA;GACA,OAAO,OAAO;GACd,WAAW,KAAK,IAAI;EACtB,CAAC;EAED,OAAO,OAAO,2BAA2B,QAAQ,UAAU,EACzD,aAAa,QAAQ,OACvB,CAAC;EAED,MAAM,mBAAmB,YAAY,OAAO,OAAO,KAAK;EACxD,MAAM,oBAAoB,YAAY,OAAO;GAC3C;GACA,OAAO,OAAO;EAChB,CAAC;EAED,OAAO;GACL,IAAI,OAAO;GACX;GACA;GACA;GACA,OAAO,OAAO;EAChB;CACF,SAAS,OAAO;EACd,MAAM,WAAW,KAAK,IAAI,IAAI;EAC9B,IAAI,aAAa,OAAO,WAAW,GACjC,MAAM,mBAAmB,YAAY,OAAO;GAC1C,QAAQ,iBAAiB,QAAQ,MAAM,UAAU,KAAA;GACjD;EACF,CAAC;OAED,MAAM,mBAAmB,YAAY,OAAO;GAAE;GAAO;EAAS,CAAC;EAEjE,OAAO,OAAO,0BAA0B;GAAE;GAAO,QAAQ;EAAS,CAAC;EACnE,MAAM;CACR;AACF;;;;AASA,SAAgB,oBAId,SAC4C;CAC5C,OAAO;AACT"}

package/dist/esm/activities/summarize/index.d.ts

@@ -5,7 +5,11 @@ import { StreamChunk, SummarizationResult } from '../../types.js';
 /** The adapter kind this activity handles */
 export declare const kind: "summarize";
 /** Extract provider options from a SummarizeAdapter via ~types */
-export type SummarizeProviderOptions<TAdapter> = TAdapter extends SummarizeAdapter<any, any> ? TAdapter['~types']['providerOptions'] : object;
+export type SummarizeProviderOptions<TAdapter> = TAdapter extends {
+    '~types': {
+        providerOptions: infer P extends object;
+    };
+} ? P : object;
 /**
  * Options for the summarize activity.
  * The model is extracted from the adapter's model property.
@@ -54,6 +58,18 @@ export interface SummarizeActivityOptions<TAdapter extends SummarizeAdapter<stri
      * mid-summary fires `onAbort`.
      */
     middleware?: Array<GenerationMiddleware>;
+    /**
+     * Maximum duration of this activity invocation in milliseconds.
+     * No SDK-wide default — choose a value suitable for the provider and job.
+     * Composed with {@link abortSignal}; the first abort wins.
+     */
+    timeout?: number;
+    /**
+     * Caller cancellation signal (request disconnects, job/runtime cancellation).
+     * Composed with {@link timeout} into an effective signal forwarded to the
+     * adapter. Request-specific — not stored on global provider client config.
+     */
+    abortSignal?: AbortSignal;
     /**
      * Whether to stream the summarization result.
      * When true, returns an AsyncIterable<StreamChunk> for streaming output.

package/dist/esm/activities/summarize/index.js

@@ -1,6 +1,7 @@
 import { resolveDebugOption } from "../../logger/resolve.js";
 import { streamGenerationResult } from "../stream-generation-result.js";
 import { applyGenerationResultTransforms, createGenerationContext, runGenerationAbort, runGenerationError, runGenerationFinish, runGenerationStart, runGenerationUsage } from "../middleware/run.js";
+import { abortReasonMessage, createActivityAbortControls, isActivityAbortError, raceWithAbort } from "../../utilities/activity-abort.js";
 import "./adapter.js";
 import "./chat-stream-summarize.js";
 import { aiEventClient } from "@tanstack/ai-event-client";
@@ -75,12 +76,16 @@ function summarize(options) {
 * Run non-streaming summarization
 */
 async function runSummarize(options) {
-	const { adapter, text, maxLength, style, focus, modelOptions, middleware } = options;
+	const { adapter, text, maxLength, style, focus, modelOptions, middleware, timeout, abortSignal: callerAbortSignal } = options;
 	const model = adapter.model;
 	const requestId = createId("summarize");
 	const inputLength = text.length;
 	const startTime = Date.now();
 	const logger = resolveDebugOption(options.debug);
+	const abortControls = createActivityAbortControls({
+		timeout,
+		abortSignal: callerAbortSignal
+	});
 	const mwCtx = createGenerationContext({
 		requestId,
 		activity: "summarize",
@@ -111,10 +116,13 @@ async function runSummarize(options) {
 		style,
 		focus,
 		modelOptions,
-		logger
+		logger,
+		...abortControls.signal ? { abortSignal: abortControls.signal } : {}
 	};
 	try {
-		const result = await applyGenerationResultTransforms(mwCtx, await adapter.summarize(summarizeOptions));
+		const rawResult = await raceWithAbort(adapter.summarize(summarizeOptions), abortControls.signal);
+		abortControls.clear();
+		const result = await applyGenerationResultTransforms(mwCtx, rawResult);
 		const duration = Date.now() - startTime;
 		const outputLength = result.summary.length;
 		aiEventClient.emit("summarize:request:completed", {
@@ -137,9 +145,15 @@ async function runSummarize(options) {
 		});
 		return result;
 	} catch (error) {
-		await runGenerationError(middleware, mwCtx, {
+		abortControls.clear();
+		const duration = Date.now() - startTime;
+		if (isActivityAbortError(error, abortControls.signal)) await runGenerationAbort(middleware, mwCtx, {
+			reason: abortReasonMessage(error, abortControls.signal),
+			duration
+		});
+		else await runGenerationError(middleware, mwCtx, {
 			error,
-			duration: Date.now() - startTime
+			duration
 		});
 		logger.errors("summarize activity failed", {
 			error,

package/dist/esm/activities/summarize/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","names":[],"sources":["../../../../src/activities/summarize/index.ts"],"sourcesContent":["/**\n * Summarize Activity\n *\n * Generates summaries from text input.\n * This is a self-contained module with implementation, types, and JSDoc.\n */\n\nimport { aiEventClient } from '@tanstack/ai-event-client'\nimport { streamGenerationResult } from '../stream-generation-result.js'\nimport { resolveDebugOption } from '../../logger/resolve'\nimport {\n  applyGenerationResultTransforms,\n  createGenerationContext,\n  runGenerationAbort,\n  runGenerationError,\n  runGenerationFinish,\n  runGenerationStart,\n  runGenerationUsage,\n} from '../middleware/run'\nimport type { InternalLogger } from '../../logger/internal-logger'\nimport type { DebugOption } from '../../logger/types'\nimport type { GenerationMiddleware } from '../middleware/types'\nimport type { SummarizeAdapter } from './adapter'\nimport type { StreamChunk, SummarizationResult } from '../../types'\n\n// ===========================\n// Activity Kind\n// ===========================\n\n/** The adapter kind this activity handles */\nexport const kind = 'summarize' as const\n\n// ===========================\n// Type Extraction Helpers\n// ===========================\n\n/** Extract provider options from a SummarizeAdapter via ~types */\nexport type SummarizeProviderOptions<TAdapter> =\n  TAdapter extends SummarizeAdapter<any, any>\n    ? TAdapter['~types']['providerOptions']\n    : object\n\n// ===========================\n// Activity Options Type\n// ===========================\n\n/**\n * Options for the summarize activity.\n * The model is extracted from the adapter's model property.\n *\n * @template TAdapter - The summarize adapter type\n * @template TStream - Whether to stream the output\n */\nexport interface SummarizeActivityOptions<\n  TAdapter extends SummarizeAdapter<string, object>,\n  TStream extends boolean = false,\n> {\n  /** The summarize adapter to use (must be created with a model) */\n  adapter: TAdapter & { kind: typeof kind }\n  /** The text to summarize */\n  text: string\n  /** Maximum length of the summary (in words or characters, provider-dependent) */\n  maxLength?: number\n  /** Style of summary to generate */\n  style?: 'bullet-points' | 'paragraph' | 'concise'\n  /** Topics or aspects to focus on in the summary */\n  focus?: Array<string>\n  /** Provider-specific options */\n  modelOptions?: SummarizeProviderOptions<TAdapter>\n  /**\n   * Optional run identity. When set on a streaming summarize, it is stamped\n   * onto the emitted `RUN_STARTED` so a delivery-durable route keys the run's\n   * log by the same id the client rejoins with — making a mid-run reload\n   * resumable. Filed under `threadId` when persistence is wired.\n   */\n  runId?: string\n  /**\n   * Stable conversation/thread id for correlating this run when persisted — the\n   * slot a reloading client hydrates the last summary by. Pass it whenever\n   * persistence is on; `withGenerationPersistence` refuses a run without one.\n   */\n  threadId?: string\n  /**\n   * Observe-only middleware notified on start, usage, success, and error. Pass\n   * `otelMiddleware()` for OpenTelemetry, `withGenerationPersistence()` to\n   * record the run (summaries are text, so the run record holds the result and\n   * there are no artifacts to store), or implement the `GenerationMiddleware`\n   * contract for a custom backend.\n   *\n   * Streaming and non-streaming behave the same way: one `onStart`, then a\n   * terminal `onFinish` / `onError`, with the result transforms applied to the\n   * `SummarizationResult` in between. A streaming consumer that disconnects\n   * mid-summary fires `onAbort`.\n   */\n  middleware?: Array<GenerationMiddleware>\n  /**\n   * Whether to stream the summarization result.\n   * When true, returns an AsyncIterable<StreamChunk> for streaming output.\n   * When false or not provided, returns a Promise<SummarizationResult>.\n   *\n   * @default false\n   */\n  stream?: TStream\n  /**\n   * Enable debug logging. Pass `true` to enable all categories, `false` to\n   * silence everything including errors, or a `DebugConfig` object for granular\n   * control and/or a custom `Logger`.\n   */\n  debug?: DebugOption\n}\n\n// ===========================\n// Activity Result Type\n// ===========================\n\n/**\n * Result type for the summarize activity.\n * - If stream is true: AsyncIterable<StreamChunk>\n * - Otherwise: Promise<SummarizationResult>\n */\nexport type SummarizeActivityResult<TStream extends boolean> =\n  TStream extends true\n    ? AsyncIterable<StreamChunk>\n    : Promise<SummarizationResult>\n\n// ===========================\n// Helper Functions\n// ===========================\n\nfunction createId(prefix: string): string {\n  return `${prefix}-${Date.now()}-${Math.random().toString(36).slice(2, 9)}`\n}\n\n// ===========================\n// Activity Implementation\n// ===========================\n\n/**\n * Summarize activity - generates summaries from text.\n *\n * Supports both streaming and non-streaming modes.\n *\n * @example Basic summarization\n * ```ts\n * import { summarize } from '@tanstack/ai'\n * import { openaiSummarize } from '@tanstack/ai-openai'\n *\n * const result = await summarize({\n *   adapter: openaiSummarize('gpt-4o-mini'),\n *   text: 'Long article text here...'\n * })\n *\n * console.log(result.summary)\n * ```\n *\n * @example Summarization with style\n * ```ts\n * const result = await summarize({\n *   adapter: openaiSummarize('gpt-4o-mini'),\n *   text: 'Long article text here...',\n *   style: 'bullet-points',\n *   maxLength: 100\n * })\n * ```\n *\n * @example Focused summarization\n * ```ts\n * const result = await summarize({\n *   adapter: openaiSummarize('gpt-4o-mini'),\n *   text: 'Long technical document...',\n *   focus: ['key findings', 'methodology']\n * })\n * ```\n *\n * @example Streaming summarization\n * ```ts\n * for await (const chunk of summarize({\n *   adapter: openaiSummarize('gpt-4o-mini'),\n *   text: 'Long article text here...',\n *   stream: true\n * })) {\n *   if (chunk.type === 'content') {\n *     process.stdout.write(chunk.delta)\n *   }\n * }\n * ```\n */\nexport function summarize<\n  TAdapter extends SummarizeAdapter<string, object>,\n  TStream extends boolean = false,\n>(\n  options: SummarizeActivityOptions<TAdapter, TStream>,\n): SummarizeActivityResult<TStream> {\n  const { stream } = options\n\n  if (stream) {\n    return runStreamingSummarize(\n      options as SummarizeActivityOptions<\n        SummarizeAdapter<string, object>,\n        true\n      >,\n    ) as SummarizeActivityResult<TStream>\n  }\n\n  return runSummarize(\n    options as SummarizeActivityOptions<\n      SummarizeAdapter<string, object>,\n      false\n    >,\n  ) as SummarizeActivityResult<TStream>\n}\n\n/**\n * Run non-streaming summarization\n */\nasync function runSummarize(\n  options: SummarizeActivityOptions<SummarizeAdapter<string, object>, false>,\n): Promise<SummarizationResult> {\n  const { adapter, text, maxLength, style, focus, modelOptions, middleware } =\n    options\n  const model = adapter.model\n  const requestId = createId('summarize')\n  const inputLength = text.length\n  const startTime = Date.now()\n  const logger: InternalLogger = resolveDebugOption(options.debug)\n\n  const mwCtx = createGenerationContext({\n    requestId,\n    activity: 'summarize',\n    provider: adapter.name,\n    model,\n    modelOptions,\n    threadId: options.threadId,\n    runId: options.runId,\n    createId,\n  })\n\n  await runGenerationStart(middleware, mwCtx)\n\n  aiEventClient.emit('summarize:request:started', {\n    requestId,\n    provider: adapter.name,\n    model,\n    inputLength,\n    timestamp: startTime,\n  })\n\n  logger.request(`activity=summarize provider=${adapter.name}`, {\n    provider: adapter.name,\n    model,\n    inputLength,\n  })\n\n  const summarizeOptions = {\n    model,\n    text,\n    maxLength,\n    style,\n    focus,\n    modelOptions,\n    logger,\n  }\n\n  try {\n    const rawResult = await adapter.summarize(summarizeOptions)\n    // Transforms run before anything observes the result — the same order every\n    // media activity uses — so the run record and the returned value are the\n    // same object.\n    const result = await applyGenerationResultTransforms(mwCtx, rawResult)\n\n    const duration = Date.now() - startTime\n    const outputLength = result.summary.length\n\n    aiEventClient.emit('summarize:request:completed', {\n      requestId,\n      provider: adapter.name,\n      model,\n      inputLength,\n      outputLength,\n      duration,\n      timestamp: Date.now(),\n    })\n\n    logger.output(`activity=summarize length=${outputLength}`, {\n      hasSummary: !!result.summary,\n      outputLength,\n    })\n\n    if (result.usage) await runGenerationUsage(middleware, mwCtx, result.usage)\n    await runGenerationFinish(middleware, mwCtx, {\n      duration,\n      usage: result.usage,\n    })\n\n    return result\n  } catch (error) {\n    await runGenerationError(middleware, mwCtx, {\n      error,\n      duration: Date.now() - startTime,\n    })\n    logger.errors('summarize activity failed', {\n      error,\n      source: 'summarize',\n    })\n    throw error\n  }\n}\n\n/** Read a `usage` off a transformed result without asserting its shape. */\nfunction usageOf(result: unknown): SummarizationResult['usage'] | undefined {\n  if (typeof result !== 'object' || result === null) return undefined\n  const usage = (result as { usage?: unknown }).usage\n  return typeof usage === 'object' && usage !== null\n    ? (usage as SummarizationResult['usage'])\n    : undefined\n}\n\n/**\n * Run streaming summarization\n * Uses the adapter's native streaming if available, otherwise falls back\n * to non-streaming and yields the result as a single chunk.\n */\nasync function* runStreamingSummarize(\n  options: SummarizeActivityOptions<SummarizeAdapter<string, object>, true>,\n): AsyncIterable<StreamChunk> {\n  const {\n    adapter,\n    text,\n    maxLength,\n    style,\n    focus,\n    modelOptions,\n    runId,\n    threadId,\n  } = options\n  const model = adapter.model\n  const logger: InternalLogger = resolveDebugOption(options.debug)\n\n  logger.request(`activity=summarize provider=${adapter.name}`, {\n    provider: adapter.name,\n    model,\n    stream: true,\n  })\n\n  // Thread the caller's run identity through so the emitted `RUN_STARTED`\n  // carries it — keeps a delivery-durable route's log keyed by the id the\n  // client rejoins with (mid-run reload resumability). Conditional spreads keep\n  // the fields off the object entirely under `exactOptionalPropertyTypes`.\n  const summarizeOptions = {\n    model,\n    text,\n    maxLength,\n    style,\n    focus,\n    modelOptions,\n    logger,\n    ...(runId !== undefined ? { runId } : {}),\n    ...(threadId !== undefined ? { threadId } : {}),\n  }\n\n  // Use real streaming if the adapter supports it\n  if (adapter.summarizeStream) {\n    yield* runNativeSummarizeStream(\n      options,\n      adapter.summarizeStream(summarizeOptions),\n      logger,\n    )\n    return\n  }\n\n  try {\n    // Fall back to non-streaming — wrap the result with streamGenerationResult,\n    // forwarding the run identity so its RUN_STARTED matches too. The generation\n    // itself goes through `runSummarize`, so middleware (and its result\n    // transforms) run exactly as they do for a non-streaming call. Only `runId`\n    // is taken from the resolved wire identity — `threadId` stays the CALLER's,\n    // since a minted one would file the run in a slot no client can hydrate.\n    yield* streamGenerationResult(\n      (resolved) =>\n        runSummarize({ ...options, stream: false, runId: resolved.runId }),\n      {\n        ...(runId !== undefined ? { runId } : {}),\n        ...(threadId !== undefined ? { threadId } : {}),\n      },\n    )\n  } catch (error) {\n    logger.errors('summarize activity failed', {\n      error,\n      source: 'summarize',\n    })\n    throw error\n  }\n}\n\n/**\n * Drive an adapter's native `summarizeStream`, wiring the generation middleware\n * around it.\n *\n * The adapter emits a terminal `generation:result` CUSTOM chunk carrying the\n * assembled {@link SummarizationResult}; that is the one point where a result\n * exists, so the transforms run there and the REWRITTEN result is what gets\n * yielded — the client and the persisted run record then hold the same object.\n * An adapter whose stream never emits one still finishes the run, just with no\n * result recorded.\n */\nasync function* runNativeSummarizeStream(\n  options: SummarizeActivityOptions<SummarizeAdapter<string, object>, true>,\n  stream: AsyncIterable<StreamChunk>,\n  logger: InternalLogger,\n): AsyncIterable<StreamChunk> {\n  const { adapter, middleware, modelOptions } = options\n  const mwCtx = createGenerationContext({\n    requestId: createId('summarize'),\n    activity: 'summarize',\n    provider: adapter.name,\n    model: adapter.model,\n    modelOptions,\n    threadId: options.threadId,\n    runId: options.runId,\n    createId,\n  })\n\n  await runGenerationStart(middleware, mwCtx)\n\n  const startTime = Date.now()\n  // Tracks whether a terminal hook already fired, so the `finally` can report an\n  // abandoned stream without double-firing. Mirrors the streaming video path.\n  let settled = false\n  try {\n    for await (const chunk of stream) {\n      if (chunk.type === 'CUSTOM' && chunk.name === 'generation:result') {\n        const result = await applyGenerationResultTransforms<unknown>(\n          mwCtx,\n          chunk.value,\n        )\n        const usage = usageOf(result)\n        // Finish before yielding the terminal chunks: a consumer that stops\n        // reading once it has the result must not trip the abandonment path.\n        if (usage) await runGenerationUsage(middleware, mwCtx, usage)\n        await runGenerationFinish(middleware, mwCtx, {\n          duration: Date.now() - startTime,\n          usage,\n        })\n        settled = true\n        yield { ...chunk, value: result }\n        continue\n      }\n      yield chunk\n    }\n    if (!settled) {\n      await runGenerationFinish(middleware, mwCtx, {\n        duration: Date.now() - startTime,\n      })\n      settled = true\n    }\n  } catch (error) {\n    settled = true\n    await runGenerationError(middleware, mwCtx, {\n      error,\n      duration: Date.now() - startTime,\n    })\n    logger.errors('summarize activity failed', {\n      error,\n      source: 'summarize',\n    })\n    throw error\n  } finally {\n    if (!settled) {\n      // The consumer abandoned the stream mid-summary, so the generator is being\n      // unwound at a `yield`. Report a cancel, not an error, so an observer ends\n      // its span (and persistence marks the run interrupted) instead of leaving\n      // the run open forever.\n      await runGenerationAbort(middleware, mwCtx, {\n        reason: 'Summarize stream abandoned before completion',\n        duration: Date.now() - startTime,\n      })\n    }\n  }\n}\n\n// ===========================\n// Options Factory\n// ===========================\n\n/**\n * Create typed options for the summarize() function without executing.\n */\nexport function createSummarizeOptions<\n  TAdapter extends SummarizeAdapter<string, object>,\n  TStream extends boolean = false,\n>(\n  options: SummarizeActivityOptions<TAdapter, TStream>,\n): SummarizeActivityOptions<TAdapter, TStream> {\n  return options\n}\n\n// Re-export adapter types\nexport type {\n  SummarizeAdapter,\n  SummarizeAdapterConfig,\n  AnySummarizeAdapter,\n} from './adapter'\nexport { BaseSummarizeAdapter } from './adapter'\nexport {\n  ChatStreamSummarizeAdapter,\n  type ChatStreamCapable,\n  type InferTextProviderOptions,\n} from './chat-stream-summarize'\n"],"mappings":";;;;;;;;;;;;;;AA8BA,IAAa,OAAO;AAmGpB,SAAS,SAAS,QAAwB;CACxC,OAAO,GAAG,OAAO,GAAG,KAAK,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC;AACzE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAwDA,SAAgB,UAId,SACkC;CAClC,MAAM,EAAE,WAAW;CAEnB,IAAI,QACF,OAAO,sBACL,OAIF;CAGF,OAAO,aACL,OAIF;AACF;;;;AAKA,eAAe,aACb,SAC8B;CAC9B,MAAM,EAAE,SAAS,MAAM,WAAW,OAAO,OAAO,cAAc,eAC5D;CACF,MAAM,QAAQ,QAAQ;CACtB,MAAM,YAAY,SAAS,WAAW;CACtC,MAAM,cAAc,KAAK;CACzB,MAAM,YAAY,KAAK,IAAI;CAC3B,MAAM,SAAyB,mBAAmB,QAAQ,KAAK;CAE/D,MAAM,QAAQ,wBAAwB;EACpC;EACA,UAAU;EACV,UAAU,QAAQ;EAClB;EACA;EACA,UAAU,QAAQ;EAClB,OAAO,QAAQ;EACf;CACF,CAAC;CAED,MAAM,mBAAmB,YAAY,KAAK;CAE1C,cAAc,KAAK,6BAA6B;EAC9C;EACA,UAAU,QAAQ;EAClB;EACA;EACA,WAAW;CACb,CAAC;CAED,OAAO,QAAQ,+BAA+B,QAAQ,QAAQ;EAC5D,UAAU,QAAQ;EAClB;EACA;CACF,CAAC;CAED,MAAM,mBAAmB;EACvB;EACA;EACA;EACA;EACA;EACA;EACA;CACF;CAEA,IAAI;EAKF,MAAM,SAAS,MAAM,gCAAgC,OAAO,MAJpC,QAAQ,UAAU,gBAAgB,CAIW;EAErE,MAAM,WAAW,KAAK,IAAI,IAAI;EAC9B,MAAM,eAAe,OAAO,QAAQ;EAEpC,cAAc,KAAK,+BAA+B;GAChD;GACA,UAAU,QAAQ;GAClB;GACA;GACA;GACA;GACA,WAAW,KAAK,IAAI;EACtB,CAAC;EAED,OAAO,OAAO,6BAA6B,gBAAgB;GACzD,YAAY,CAAC,CAAC,OAAO;GACrB;EACF,CAAC;EAED,IAAI,OAAO,OAAO,MAAM,mBAAmB,YAAY,OAAO,OAAO,KAAK;EAC1E,MAAM,oBAAoB,YAAY,OAAO;GAC3C;GACA,OAAO,OAAO;EAChB,CAAC;EAED,OAAO;CACT,SAAS,OAAO;EACd,MAAM,mBAAmB,YAAY,OAAO;GAC1C;GACA,UAAU,KAAK,IAAI,IAAI;EACzB,CAAC;EACD,OAAO,OAAO,6BAA6B;GACzC;GACA,QAAQ;EACV,CAAC;EACD,MAAM;CACR;AACF;;AAGA,SAAS,QAAQ,QAA2D;CAC1E,IAAI,OAAO,WAAW,YAAY,WAAW,MAAM,OAAO,KAAA;CAC1D,MAAM,QAAS,OAA+B;CAC9C,OAAO,OAAO,UAAU,YAAY,UAAU,OACzC,QACD,KAAA;AACN;;;;;;AAOA,gBAAgB,sBACd,SAC4B;CAC5B,MAAM,EACJ,SACA,MACA,WACA,OACA,OACA,cACA,OACA,aACE;CACJ,MAAM,QAAQ,QAAQ;CACtB,MAAM,SAAyB,mBAAmB,QAAQ,KAAK;CAE/D,OAAO,QAAQ,+BAA+B,QAAQ,QAAQ;EAC5D,UAAU,QAAQ;EAClB;EACA,QAAQ;CACV,CAAC;CAMD,MAAM,mBAAmB;EACvB;EACA;EACA;EACA;EACA;EACA;EACA;EACA,GAAI,UAAU,KAAA,IAAY,EAAE,MAAM,IAAI,CAAC;EACvC,GAAI,aAAa,KAAA,IAAY,EAAE,SAAS,IAAI,CAAC;CAC/C;CAGA,IAAI,QAAQ,iBAAiB;EAC3B,OAAO,yBACL,SACA,QAAQ,gBAAgB,gBAAgB,GACxC,MACF;EACA;CACF;CAEA,IAAI;EAOF,OAAO,wBACJ,aACC,aAAa;GAAE,GAAG;GAAS,QAAQ;GAAO,OAAO,SAAS;EAAM,CAAC,GACnE;GACE,GAAI,UAAU,KAAA,IAAY,EAAE,MAAM,IAAI,CAAC;GACvC,GAAI,aAAa,KAAA,IAAY,EAAE,SAAS,IAAI,CAAC;EAC/C,CACF;CACF,SAAS,OAAO;EACd,OAAO,OAAO,6BAA6B;GACzC;GACA,QAAQ;EACV,CAAC;EACD,MAAM;CACR;AACF;;;;;;;;;;;;AAaA,gBAAgB,yBACd,SACA,QACA,QAC4B;CAC5B,MAAM,EAAE,SAAS,YAAY,iBAAiB;CAC9C,MAAM,QAAQ,wBAAwB;EACpC,WAAW,SAAS,WAAW;EAC/B,UAAU;EACV,UAAU,QAAQ;EAClB,OAAO,QAAQ;EACf;EACA,UAAU,QAAQ;EAClB,OAAO,QAAQ;EACf;CACF,CAAC;CAED,MAAM,mBAAmB,YAAY,KAAK;CAE1C,MAAM,YAAY,KAAK,IAAI;CAG3B,IAAI,UAAU;CACd,IAAI;EACF,WAAW,MAAM,SAAS,QAAQ;GAChC,IAAI,MAAM,SAAS,YAAY,MAAM,SAAS,qBAAqB;IACjE,MAAM,SAAS,MAAM,gCACnB,OACA,MAAM,KACR;IACA,MAAM,QAAQ,QAAQ,MAAM;IAG5B,IAAI,OAAO,MAAM,mBAAmB,YAAY,OAAO,KAAK;IAC5D,MAAM,oBAAoB,YAAY,OAAO;KAC3C,UAAU,KAAK,IAAI,IAAI;KACvB;IACF,CAAC;IACD,UAAU;IACV,MAAM;KAAE,GAAG;KAAO,OAAO;IAAO;IAChC;GACF;GACA,MAAM;EACR;EACA,IAAI,CAAC,SAAS;GACZ,MAAM,oBAAoB,YAAY,OAAO,EAC3C,UAAU,KAAK,IAAI,IAAI,UACzB,CAAC;GACD,UAAU;EACZ;CACF,SAAS,OAAO;EACd,UAAU;EACV,MAAM,mBAAmB,YAAY,OAAO;GAC1C;GACA,UAAU,KAAK,IAAI,IAAI;EACzB,CAAC;EACD,OAAO,OAAO,6BAA6B;GACzC;GACA,QAAQ;EACV,CAAC;EACD,MAAM;CACR,UAAU;EACR,IAAI,CAAC,SAKH,MAAM,mBAAmB,YAAY,OAAO;GAC1C,QAAQ;GACR,UAAU,KAAK,IAAI,IAAI;EACzB,CAAC;CAEL;AACF;;;;AASA,SAAgB,uBAId,SAC6C;CAC7C,OAAO;AACT"}
+{"version":3,"file":"index.js","names":[],"sources":["../../../../src/activities/summarize/index.ts"],"sourcesContent":["/**\n * Summarize Activity\n *\n * Generates summaries from text input.\n * This is a self-contained module with implementation, types, and JSDoc.\n */\n\nimport { aiEventClient } from '@tanstack/ai-event-client'\nimport { streamGenerationResult } from '../stream-generation-result.js'\nimport { resolveDebugOption } from '../../logger/resolve'\nimport {\n  applyGenerationResultTransforms,\n  createGenerationContext,\n  runGenerationAbort,\n  runGenerationError,\n  runGenerationFinish,\n  runGenerationStart,\n  runGenerationUsage,\n} from '../middleware/run'\nimport {\n  abortReasonMessage,\n  createActivityAbortControls,\n  isActivityAbortError,\n  raceWithAbort,\n} from '../../utilities/activity-abort'\nimport type { InternalLogger } from '../../logger/internal-logger'\nimport type { DebugOption } from '../../logger/types'\nimport type { GenerationMiddleware } from '../middleware/types'\nimport type { SummarizeAdapter } from './adapter'\nimport type { StreamChunk, SummarizationResult } from '../../types'\n\n// ===========================\n// Activity Kind\n// ===========================\n\n/** The adapter kind this activity handles */\nexport const kind = 'summarize' as const\n\n// ===========================\n// Type Extraction Helpers\n// ===========================\n\n/** Extract provider options from a SummarizeAdapter via ~types */\nexport type SummarizeProviderOptions<TAdapter> = TAdapter extends {\n  '~types': { providerOptions: infer P extends object }\n}\n  ? P\n  : object\n\n// ===========================\n// Activity Options Type\n// ===========================\n\n/**\n * Options for the summarize activity.\n * The model is extracted from the adapter's model property.\n *\n * @template TAdapter - The summarize adapter type\n * @template TStream - Whether to stream the output\n */\nexport interface SummarizeActivityOptions<\n  TAdapter extends SummarizeAdapter<string, object>,\n  TStream extends boolean = false,\n> {\n  /** The summarize adapter to use (must be created with a model) */\n  adapter: TAdapter & { kind: typeof kind }\n  /** The text to summarize */\n  text: string\n  /** Maximum length of the summary (in words or characters, provider-dependent) */\n  maxLength?: number\n  /** Style of summary to generate */\n  style?: 'bullet-points' | 'paragraph' | 'concise'\n  /** Topics or aspects to focus on in the summary */\n  focus?: Array<string>\n  /** Provider-specific options */\n  modelOptions?: SummarizeProviderOptions<TAdapter>\n  /**\n   * Optional run identity. When set on a streaming summarize, it is stamped\n   * onto the emitted `RUN_STARTED` so a delivery-durable route keys the run's\n   * log by the same id the client rejoins with — making a mid-run reload\n   * resumable. Filed under `threadId` when persistence is wired.\n   */\n  runId?: string\n  /**\n   * Stable conversation/thread id for correlating this run when persisted — the\n   * slot a reloading client hydrates the last summary by. Pass it whenever\n   * persistence is on; `withGenerationPersistence` refuses a run without one.\n   */\n  threadId?: string\n  /**\n   * Observe-only middleware notified on start, usage, success, and error. Pass\n   * `otelMiddleware()` for OpenTelemetry, `withGenerationPersistence()` to\n   * record the run (summaries are text, so the run record holds the result and\n   * there are no artifacts to store), or implement the `GenerationMiddleware`\n   * contract for a custom backend.\n   *\n   * Streaming and non-streaming behave the same way: one `onStart`, then a\n   * terminal `onFinish` / `onError`, with the result transforms applied to the\n   * `SummarizationResult` in between. A streaming consumer that disconnects\n   * mid-summary fires `onAbort`.\n   */\n  middleware?: Array<GenerationMiddleware>\n  /**\n   * Maximum duration of this activity invocation in milliseconds.\n   * No SDK-wide default — choose a value suitable for the provider and job.\n   * Composed with {@link abortSignal}; the first abort wins.\n   */\n  timeout?: number\n  /**\n   * Caller cancellation signal (request disconnects, job/runtime cancellation).\n   * Composed with {@link timeout} into an effective signal forwarded to the\n   * adapter. Request-specific — not stored on global provider client config.\n   */\n  abortSignal?: AbortSignal\n  /**\n   * Whether to stream the summarization result.\n   * When true, returns an AsyncIterable<StreamChunk> for streaming output.\n   * When false or not provided, returns a Promise<SummarizationResult>.\n   *\n   * @default false\n   */\n  stream?: TStream\n  /**\n   * Enable debug logging. Pass `true` to enable all categories, `false` to\n   * silence everything including errors, or a `DebugConfig` object for granular\n   * control and/or a custom `Logger`.\n   */\n  debug?: DebugOption\n}\n\n// ===========================\n// Activity Result Type\n// ===========================\n\n/**\n * Result type for the summarize activity.\n * - If stream is true: AsyncIterable<StreamChunk>\n * - Otherwise: Promise<SummarizationResult>\n */\nexport type SummarizeActivityResult<TStream extends boolean> =\n  TStream extends true\n    ? AsyncIterable<StreamChunk>\n    : Promise<SummarizationResult>\n\n// ===========================\n// Helper Functions\n// ===========================\n\nfunction createId(prefix: string): string {\n  return `${prefix}-${Date.now()}-${Math.random().toString(36).slice(2, 9)}`\n}\n\n// ===========================\n// Activity Implementation\n// ===========================\n\n/**\n * Summarize activity - generates summaries from text.\n *\n * Supports both streaming and non-streaming modes.\n *\n * @example Basic summarization\n * ```ts\n * import { summarize } from '@tanstack/ai'\n * import { openaiSummarize } from '@tanstack/ai-openai'\n *\n * const result = await summarize({\n *   adapter: openaiSummarize('gpt-4o-mini'),\n *   text: 'Long article text here...'\n * })\n *\n * console.log(result.summary)\n * ```\n *\n * @example Summarization with style\n * ```ts\n * const result = await summarize({\n *   adapter: openaiSummarize('gpt-4o-mini'),\n *   text: 'Long article text here...',\n *   style: 'bullet-points',\n *   maxLength: 100\n * })\n * ```\n *\n * @example Focused summarization\n * ```ts\n * const result = await summarize({\n *   adapter: openaiSummarize('gpt-4o-mini'),\n *   text: 'Long technical document...',\n *   focus: ['key findings', 'methodology']\n * })\n * ```\n *\n * @example Streaming summarization\n * ```ts\n * for await (const chunk of summarize({\n *   adapter: openaiSummarize('gpt-4o-mini'),\n *   text: 'Long article text here...',\n *   stream: true\n * })) {\n *   if (chunk.type === 'content') {\n *     process.stdout.write(chunk.delta)\n *   }\n * }\n * ```\n */\nexport function summarize<\n  TAdapter extends SummarizeAdapter<string, object>,\n  TStream extends boolean = false,\n>(\n  options: SummarizeActivityOptions<TAdapter, TStream>,\n): SummarizeActivityResult<TStream> {\n  const { stream } = options\n\n  if (stream) {\n    return runStreamingSummarize(\n      options as SummarizeActivityOptions<TAdapter, true>,\n    ) as SummarizeActivityResult<TStream>\n  }\n\n  return runSummarize(\n    options as SummarizeActivityOptions<TAdapter, false>,\n  ) as SummarizeActivityResult<TStream>\n}\n\n/**\n * Run non-streaming summarization\n */\nasync function runSummarize(\n  options: SummarizeActivityOptions<SummarizeAdapter<string, object>, false>,\n): Promise<SummarizationResult> {\n  const {\n    adapter,\n    text,\n    maxLength,\n    style,\n    focus,\n    modelOptions,\n    middleware,\n    timeout,\n    abortSignal: callerAbortSignal,\n  } = options\n  const model = adapter.model\n  const requestId = createId('summarize')\n  const inputLength = text.length\n  const startTime = Date.now()\n  const logger: InternalLogger = resolveDebugOption(options.debug)\n  const abortControls = createActivityAbortControls({\n    timeout,\n    abortSignal: callerAbortSignal,\n  })\n\n  const mwCtx = createGenerationContext({\n    requestId,\n    activity: 'summarize',\n    provider: adapter.name,\n    model,\n    modelOptions,\n    threadId: options.threadId,\n    runId: options.runId,\n    createId,\n  })\n\n  await runGenerationStart(middleware, mwCtx)\n\n  aiEventClient.emit('summarize:request:started', {\n    requestId,\n    provider: adapter.name,\n    model,\n    inputLength,\n    timestamp: startTime,\n  })\n\n  logger.request(`activity=summarize provider=${adapter.name}`, {\n    provider: adapter.name,\n    model,\n    inputLength,\n  })\n\n  const summarizeOptions = {\n    model,\n    text,\n    maxLength,\n    style,\n    focus,\n    modelOptions,\n    logger,\n    ...(abortControls.signal ? { abortSignal: abortControls.signal } : {}),\n  }\n\n  try {\n    const rawResult = await raceWithAbort(\n      adapter.summarize(summarizeOptions),\n      abortControls.signal,\n    )\n    abortControls.clear()\n    // Transforms run before anything observes the result — the same order every\n    // media activity uses — so the run record and the returned value are the\n    // same object.\n    const result = await applyGenerationResultTransforms(mwCtx, rawResult)\n\n    const duration = Date.now() - startTime\n    const outputLength = result.summary.length\n\n    aiEventClient.emit('summarize:request:completed', {\n      requestId,\n      provider: adapter.name,\n      model,\n      inputLength,\n      outputLength,\n      duration,\n      timestamp: Date.now(),\n    })\n\n    logger.output(`activity=summarize length=${outputLength}`, {\n      hasSummary: !!result.summary,\n      outputLength,\n    })\n\n    if (result.usage) await runGenerationUsage(middleware, mwCtx, result.usage)\n    await runGenerationFinish(middleware, mwCtx, {\n      duration,\n      usage: result.usage,\n    })\n\n    return result\n  } catch (error) {\n    abortControls.clear()\n    const duration = Date.now() - startTime\n    if (isActivityAbortError(error, abortControls.signal)) {\n      await runGenerationAbort(middleware, mwCtx, {\n        reason: abortReasonMessage(error, abortControls.signal),\n        duration,\n      })\n    } else {\n      await runGenerationError(middleware, mwCtx, {\n        error,\n        duration,\n      })\n    }\n    logger.errors('summarize activity failed', {\n      error,\n      source: 'summarize',\n    })\n    throw error\n  }\n}\n\n/** Read a `usage` off a transformed result without asserting its shape. */\nfunction usageOf(result: unknown): SummarizationResult['usage'] | undefined {\n  if (typeof result !== 'object' || result === null) return undefined\n  const usage = (result as { usage?: unknown }).usage\n  return typeof usage === 'object' && usage !== null\n    ? (usage as SummarizationResult['usage'])\n    : undefined\n}\n\n/**\n * Run streaming summarization\n * Uses the adapter's native streaming if available, otherwise falls back\n * to non-streaming and yields the result as a single chunk.\n */\nasync function* runStreamingSummarize(\n  options: SummarizeActivityOptions<SummarizeAdapter<string, object>, true>,\n): AsyncIterable<StreamChunk> {\n  const {\n    adapter,\n    text,\n    maxLength,\n    style,\n    focus,\n    modelOptions,\n    runId,\n    threadId,\n  } = options\n  const model = adapter.model\n  const logger: InternalLogger = resolveDebugOption(options.debug)\n\n  logger.request(`activity=summarize provider=${adapter.name}`, {\n    provider: adapter.name,\n    model,\n    stream: true,\n  })\n\n  // Thread the caller's run identity through so the emitted `RUN_STARTED`\n  // carries it — keeps a delivery-durable route's log keyed by the id the\n  // client rejoins with (mid-run reload resumability). Conditional spreads keep\n  // the fields off the object entirely under `exactOptionalPropertyTypes`.\n  const summarizeOptions = {\n    model,\n    text,\n    maxLength,\n    style,\n    focus,\n    modelOptions,\n    logger,\n    ...(runId !== undefined ? { runId } : {}),\n    ...(threadId !== undefined ? { threadId } : {}),\n  }\n\n  // Use real streaming if the adapter supports it\n  if (adapter.summarizeStream) {\n    yield* runNativeSummarizeStream(\n      options,\n      adapter.summarizeStream(summarizeOptions),\n      logger,\n    )\n    return\n  }\n\n  try {\n    // Fall back to non-streaming — wrap the result with streamGenerationResult,\n    // forwarding the run identity so its RUN_STARTED matches too. The generation\n    // itself goes through `runSummarize`, so middleware (and its result\n    // transforms) run exactly as they do for a non-streaming call. Only `runId`\n    // is taken from the resolved wire identity — `threadId` stays the CALLER's,\n    // since a minted one would file the run in a slot no client can hydrate.\n    yield* streamGenerationResult(\n      (resolved) =>\n        runSummarize({ ...options, stream: false, runId: resolved.runId }),\n      {\n        ...(runId !== undefined ? { runId } : {}),\n        ...(threadId !== undefined ? { threadId } : {}),\n      },\n    )\n  } catch (error) {\n    logger.errors('summarize activity failed', {\n      error,\n      source: 'summarize',\n    })\n    throw error\n  }\n}\n\n/**\n * Drive an adapter's native `summarizeStream`, wiring the generation middleware\n * around it.\n *\n * The adapter emits a terminal `generation:result` CUSTOM chunk carrying the\n * assembled {@link SummarizationResult}; that is the one point where a result\n * exists, so the transforms run there and the REWRITTEN result is what gets\n * yielded — the client and the persisted run record then hold the same object.\n * An adapter whose stream never emits one still finishes the run, just with no\n * result recorded.\n */\nasync function* runNativeSummarizeStream(\n  options: SummarizeActivityOptions<SummarizeAdapter<string, object>, true>,\n  stream: AsyncIterable<StreamChunk>,\n  logger: InternalLogger,\n): AsyncIterable<StreamChunk> {\n  const { adapter, middleware, modelOptions } = options\n  const mwCtx = createGenerationContext({\n    requestId: createId('summarize'),\n    activity: 'summarize',\n    provider: adapter.name,\n    model: adapter.model,\n    modelOptions,\n    threadId: options.threadId,\n    runId: options.runId,\n    createId,\n  })\n\n  await runGenerationStart(middleware, mwCtx)\n\n  const startTime = Date.now()\n  // Tracks whether a terminal hook already fired, so the `finally` can report an\n  // abandoned stream without double-firing. Mirrors the streaming video path.\n  let settled = false\n  try {\n    for await (const chunk of stream) {\n      if (chunk.type === 'CUSTOM' && chunk.name === 'generation:result') {\n        const result = await applyGenerationResultTransforms<unknown>(\n          mwCtx,\n          chunk.value,\n        )\n        const usage = usageOf(result)\n        // Finish before yielding the terminal chunks: a consumer that stops\n        // reading once it has the result must not trip the abandonment path.\n        if (usage) await runGenerationUsage(middleware, mwCtx, usage)\n        await runGenerationFinish(middleware, mwCtx, {\n          duration: Date.now() - startTime,\n          usage,\n        })\n        settled = true\n        yield { ...chunk, value: result }\n        continue\n      }\n      yield chunk\n    }\n    if (!settled) {\n      await runGenerationFinish(middleware, mwCtx, {\n        duration: Date.now() - startTime,\n      })\n      settled = true\n    }\n  } catch (error) {\n    settled = true\n    await runGenerationError(middleware, mwCtx, {\n      error,\n      duration: Date.now() - startTime,\n    })\n    logger.errors('summarize activity failed', {\n      error,\n      source: 'summarize',\n    })\n    throw error\n  } finally {\n    if (!settled) {\n      // The consumer abandoned the stream mid-summary, so the generator is being\n      // unwound at a `yield`. Report a cancel, not an error, so an observer ends\n      // its span (and persistence marks the run interrupted) instead of leaving\n      // the run open forever.\n      await runGenerationAbort(middleware, mwCtx, {\n        reason: 'Summarize stream abandoned before completion',\n        duration: Date.now() - startTime,\n      })\n    }\n  }\n}\n\n// ===========================\n// Options Factory\n// ===========================\n\n/**\n * Create typed options for the summarize() function without executing.\n */\nexport function createSummarizeOptions<\n  TAdapter extends SummarizeAdapter<string, object>,\n  TStream extends boolean = false,\n>(\n  options: SummarizeActivityOptions<TAdapter, TStream>,\n): SummarizeActivityOptions<TAdapter, TStream> {\n  return options\n}\n\n// Re-export adapter types\nexport type {\n  SummarizeAdapter,\n  SummarizeAdapterConfig,\n  AnySummarizeAdapter,\n} from './adapter'\nexport { BaseSummarizeAdapter } from './adapter'\nexport {\n  ChatStreamSummarizeAdapter,\n  type ChatStreamCapable,\n  type InferTextProviderOptions,\n} from './chat-stream-summarize'\n"],"mappings":";;;;;;;;;;;;;;;AAoCA,IAAa,OAAO;AAgHpB,SAAS,SAAS,QAAwB;CACxC,OAAO,GAAG,OAAO,GAAG,KAAK,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC;AACzE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAwDA,SAAgB,UAId,SACkC;CAClC,MAAM,EAAE,WAAW;CAEnB,IAAI,QACF,OAAO,sBACL,OACF;CAGF,OAAO,aACL,OACF;AACF;;;;AAKA,eAAe,aACb,SAC8B;CAC9B,MAAM,EACJ,SACA,MACA,WACA,OACA,OACA,cACA,YACA,SACA,aAAa,sBACX;CACJ,MAAM,QAAQ,QAAQ;CACtB,MAAM,YAAY,SAAS,WAAW;CACtC,MAAM,cAAc,KAAK;CACzB,MAAM,YAAY,KAAK,IAAI;CAC3B,MAAM,SAAyB,mBAAmB,QAAQ,KAAK;CAC/D,MAAM,gBAAgB,4BAA4B;EAChD;EACA,aAAa;CACf,CAAC;CAED,MAAM,QAAQ,wBAAwB;EACpC;EACA,UAAU;EACV,UAAU,QAAQ;EAClB;EACA;EACA,UAAU,QAAQ;EAClB,OAAO,QAAQ;EACf;CACF,CAAC;CAED,MAAM,mBAAmB,YAAY,KAAK;CAE1C,cAAc,KAAK,6BAA6B;EAC9C;EACA,UAAU,QAAQ;EAClB;EACA;EACA,WAAW;CACb,CAAC;CAED,OAAO,QAAQ,+BAA+B,QAAQ,QAAQ;EAC5D,UAAU,QAAQ;EAClB;EACA;CACF,CAAC;CAED,MAAM,mBAAmB;EACvB;EACA;EACA;EACA;EACA;EACA;EACA;EACA,GAAI,cAAc,SAAS,EAAE,aAAa,cAAc,OAAO,IAAI,CAAC;CACtE;CAEA,IAAI;EACF,MAAM,YAAY,MAAM,cACtB,QAAQ,UAAU,gBAAgB,GAClC,cAAc,MAChB;EACA,cAAc,MAAM;EAIpB,MAAM,SAAS,MAAM,gCAAgC,OAAO,SAAS;EAErE,MAAM,WAAW,KAAK,IAAI,IAAI;EAC9B,MAAM,eAAe,OAAO,QAAQ;EAEpC,cAAc,KAAK,+BAA+B;GAChD;GACA,UAAU,QAAQ;GAClB;GACA;GACA;GACA;GACA,WAAW,KAAK,IAAI;EACtB,CAAC;EAED,OAAO,OAAO,6BAA6B,gBAAgB;GACzD,YAAY,CAAC,CAAC,OAAO;GACrB;EACF,CAAC;EAED,IAAI,OAAO,OAAO,MAAM,mBAAmB,YAAY,OAAO,OAAO,KAAK;EAC1E,MAAM,oBAAoB,YAAY,OAAO;GAC3C;GACA,OAAO,OAAO;EAChB,CAAC;EAED,OAAO;CACT,SAAS,OAAO;EACd,cAAc,MAAM;EACpB,MAAM,WAAW,KAAK,IAAI,IAAI;EAC9B,IAAI,qBAAqB,OAAO,cAAc,MAAM,GAClD,MAAM,mBAAmB,YAAY,OAAO;GAC1C,QAAQ,mBAAmB,OAAO,cAAc,MAAM;GACtD;EACF,CAAC;OAED,MAAM,mBAAmB,YAAY,OAAO;GAC1C;GACA;EACF,CAAC;EAEH,OAAO,OAAO,6BAA6B;GACzC;GACA,QAAQ;EACV,CAAC;EACD,MAAM;CACR;AACF;;AAGA,SAAS,QAAQ,QAA2D;CAC1E,IAAI,OAAO,WAAW,YAAY,WAAW,MAAM,OAAO,KAAA;CAC1D,MAAM,QAAS,OAA+B;CAC9C,OAAO,OAAO,UAAU,YAAY,UAAU,OACzC,QACD,KAAA;AACN;;;;;;AAOA,gBAAgB,sBACd,SAC4B;CAC5B,MAAM,EACJ,SACA,MACA,WACA,OACA,OACA,cACA,OACA,aACE;CACJ,MAAM,QAAQ,QAAQ;CACtB,MAAM,SAAyB,mBAAmB,QAAQ,KAAK;CAE/D,OAAO,QAAQ,+BAA+B,QAAQ,QAAQ;EAC5D,UAAU,QAAQ;EAClB;EACA,QAAQ;CACV,CAAC;CAMD,MAAM,mBAAmB;EACvB;EACA;EACA;EACA;EACA;EACA;EACA;EACA,GAAI,UAAU,KAAA,IAAY,EAAE,MAAM,IAAI,CAAC;EACvC,GAAI,aAAa,KAAA,IAAY,EAAE,SAAS,IAAI,CAAC;CAC/C;CAGA,IAAI,QAAQ,iBAAiB;EAC3B,OAAO,yBACL,SACA,QAAQ,gBAAgB,gBAAgB,GACxC,MACF;EACA;CACF;CAEA,IAAI;EAOF,OAAO,wBACJ,aACC,aAAa;GAAE,GAAG;GAAS,QAAQ;GAAO,OAAO,SAAS;EAAM,CAAC,GACnE;GACE,GAAI,UAAU,KAAA,IAAY,EAAE,MAAM,IAAI,CAAC;GACvC,GAAI,aAAa,KAAA,IAAY,EAAE,SAAS,IAAI,CAAC;EAC/C,CACF;CACF,SAAS,OAAO;EACd,OAAO,OAAO,6BAA6B;GACzC;GACA,QAAQ;EACV,CAAC;EACD,MAAM;CACR;AACF;;;;;;;;;;;;AAaA,gBAAgB,yBACd,SACA,QACA,QAC4B;CAC5B,MAAM,EAAE,SAAS,YAAY,iBAAiB;CAC9C,MAAM,QAAQ,wBAAwB;EACpC,WAAW,SAAS,WAAW;EAC/B,UAAU;EACV,UAAU,QAAQ;EAClB,OAAO,QAAQ;EACf;EACA,UAAU,QAAQ;EAClB,OAAO,QAAQ;EACf;CACF,CAAC;CAED,MAAM,mBAAmB,YAAY,KAAK;CAE1C,MAAM,YAAY,KAAK,IAAI;CAG3B,IAAI,UAAU;CACd,IAAI;EACF,WAAW,MAAM,SAAS,QAAQ;GAChC,IAAI,MAAM,SAAS,YAAY,MAAM,SAAS,qBAAqB;IACjE,MAAM,SAAS,MAAM,gCACnB,OACA,MAAM,KACR;IACA,MAAM,QAAQ,QAAQ,MAAM;IAG5B,IAAI,OAAO,MAAM,mBAAmB,YAAY,OAAO,KAAK;IAC5D,MAAM,oBAAoB,YAAY,OAAO;KAC3C,UAAU,KAAK,IAAI,IAAI;KACvB;IACF,CAAC;IACD,UAAU;IACV,MAAM;KAAE,GAAG;KAAO,OAAO;IAAO;IAChC;GACF;GACA,MAAM;EACR;EACA,IAAI,CAAC,SAAS;GACZ,MAAM,oBAAoB,YAAY,OAAO,EAC3C,UAAU,KAAK,IAAI,IAAI,UACzB,CAAC;GACD,UAAU;EACZ;CACF,SAAS,OAAO;EACd,UAAU;EACV,MAAM,mBAAmB,YAAY,OAAO;GAC1C;GACA,UAAU,KAAK,IAAI,IAAI;EACzB,CAAC;EACD,OAAO,OAAO,6BAA6B;GACzC;GACA,QAAQ;EACV,CAAC;EACD,MAAM;CACR,UAAU;EACR,IAAI,CAAC,SAKH,MAAM,mBAAmB,YAAY,OAAO;GAC1C,QAAQ;GACR,UAAU,KAAK,IAAI,IAAI;EACzB,CAAC;CAEL;AACF;;;;AASA,SAAgB,uBAId,SAC6C;CAC7C,OAAO;AACT"}

package/dist/esm/index.d.ts

@@ -1,12 +1,15 @@
-export { chat, summarize, generateImage, generateAudio, generateVideo, getVideoJobStatus, generateSpeech, generateTranscription, } from './activities/index.js';
+export { chat, summarize, rerank, generateImage, generateAudio, generateVideo, getVideoJobStatus, generateSpeech, generateTranscription, embed, } from './activities/index.js';
 export { createChatOptions } from './activities/chat/index.js';
 export { createSummarizeOptions } from './activities/summarize/index.js';
+export { createRerankOptions } from './activities/rerank/index.js';
 export { createImageOptions } from './activities/generateImage/index.js';
 export { createAudioOptions } from './activities/generateAudio/index.js';
 export { createVideoOptions } from './activities/generateVideo/index.js';
 export { createSpeechOptions } from './activities/generateSpeech/index.js';
 export { createTranscriptionOptions } from './activities/generateTranscription/index.js';
-export type { AIAdapter, ImageAdapter, AnyImageAdapter, TextAdapter, AnyTextAdapter, AnySummarizeAdapter, SummarizeAdapter, AnyAudioAdapter, AudioAdapter, AnyTTSAdapter, TTSAdapter, AnyTranscriptionAdapter, TranscriptionAdapter, AnyVideoAdapter, VideoAdapter, } from './activities/index.js';
+export { createEmbedOptions } from './activities/embed/index.js';
+export type { AIAdapter, ImageAdapter, AnyImageAdapter, TextAdapter, AnyTextAdapter, AnySummarizeAdapter, SummarizeAdapter, AnyAudioAdapter, AudioAdapter, AnyTTSAdapter, TTSAdapter, AnyTranscriptionAdapter, TranscriptionAdapter, AnyVideoAdapter, VideoAdapter, AnyEmbeddingAdapter, EmbeddingAdapter, AnyRerankAdapter, RerankAdapter, } from './activities/index.js';
+export { BaseRerankAdapter } from './activities/rerank/adapter.js';
 export { toolDefinition, type ToolDefinition, type ToolDefinitionInstance, type ToolDefinitionConfig, type ServerTool, type AnyServerTool, type ClientTool, type AnyClientTool, type InferToolName, type InferToolInput, type InferToolOutput, type ApprovalCapabilityOf, type ApprovalSchemaConfig, type ApprovalSchemaOf, type InputSchemaOf, type OutputSchemaOf, type NoSchema, } from './activities/chat/tools/tool-definition.js';
 export { hashSchemaInput, normalizeApprovalSchema, type NormalizedApprovalSchema, type NormalizedSchemaInput, } from './activities/chat/tools/approval-schema.js';
 export { canonicalInterruptJson, cloneAndDeepFreezeJson, digestInterruptJson, } from './interrupt-serialization.js';
@@ -43,6 +46,8 @@ export { firstSentence, renderLazyCatalogEntry, } from './activities/chat/tools/
 export { buildBaseUsage, type BaseUsageInput } from './utilities/usage.js';
 export { resolveMediaPrompt } from './utilities/media-prompt.js';
 export type { ResolvedMediaPrompt } from './utilities/media-prompt.js';
+export { resolveEmbeddingInput, requireTextOnlyEmbeddingInput, countEmbeddingInputModalities, } from './utilities/embedding-input.js';
+export type { ResolvedEmbeddingItem } from './utilities/embedding-input.js';
 export type { SystemPrompt, NormalizedSystemPrompt } from './system-prompts.js';
 export { normalizeSystemPrompts } from './system-prompts.js';
 export { detectImageMimeType } from './utils.js';