@tanstack/ai 0.6.2 → 0.8.0

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

@@ -1,9 +1,17 @@
-import { aiEventClient } from "../../event-client.js";
+import { aiEventClient } from "@tanstack/ai-event-client";
 const kind = "video";
 function createId(prefix) {
   return `${prefix}-${Date.now()}-${Math.random().toString(36).slice(2, 9)}`;
 }
-async function generateVideo(options) {
+function generateVideo(options) {
+  if (options.stream) {
+    return runStreamingVideoGeneration(
+      options
+    );
+  }
+  return runCreateVideoJob(options);
+}
+async function runCreateVideoJob(options) {
   const { adapter, prompt, size, duration, modelOptions } = options;
   const model = adapter.model;
   return adapter.createVideoJob({
@@ -14,6 +22,87 @@ async function generateVideo(options) {
     modelOptions
   });
 }
+function sleep(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+async function* runStreamingVideoGeneration(options) {
+  const { adapter, prompt, size, duration, modelOptions } = options;
+  const model = adapter.model;
+  const runId = options.runId ?? createId("run");
+  const pollingInterval = options.pollingInterval ?? 2e3;
+  const maxDuration = options.maxDuration ?? 6e5;
+  yield {
+    type: "RUN_STARTED",
+    runId,
+    timestamp: Date.now()
+  };
+  try {
+    const jobResult = await adapter.createVideoJob({
+      model,
+      prompt,
+      size,
+      duration,
+      modelOptions
+    });
+    yield {
+      type: "CUSTOM",
+      name: "video:job:created",
+      value: { jobId: jobResult.jobId },
+      timestamp: Date.now()
+    };
+    const startTime = Date.now();
+    while (Date.now() - startTime < maxDuration) {
+      await sleep(pollingInterval);
+      const statusResult = await adapter.getVideoStatus(jobResult.jobId);
+      yield {
+        type: "CUSTOM",
+        name: "video:status",
+        value: {
+          jobId: jobResult.jobId,
+          status: statusResult.status,
+          progress: statusResult.progress,
+          error: statusResult.error
+        },
+        timestamp: Date.now()
+      };
+      if (statusResult.status === "completed") {
+        const urlResult = await adapter.getVideoUrl(jobResult.jobId);
+        yield {
+          type: "CUSTOM",
+          name: "generation:result",
+          value: {
+            jobId: jobResult.jobId,
+            status: "completed",
+            url: urlResult.url,
+            expiresAt: urlResult.expiresAt
+          },
+          timestamp: Date.now()
+        };
+        yield {
+          type: "RUN_FINISHED",
+          runId,
+          finishReason: "stop",
+          timestamp: Date.now()
+        };
+        return;
+      }
+      if (statusResult.status === "failed") {
+        throw new Error(statusResult.error || "Video generation failed");
+      }
+    }
+    throw new Error("Video generation timed out");
+  } catch (error) {
+    yield {
+      type: "RUN_ERROR",
+      runId,
+      error: {
+        message: error.message || "Video generation failed",
+        code: error.code
+      },
+      timestamp: Date.now()
+    };
+  }
+}
 async function getVideoJobStatus(options) {
   const { adapter, jobId } = options;
   const requestId = createId("video-status");

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

@@ -1 +1 @@
-{"version":3,"file":"index.js","sources":["../../../../src/activities/generateVideo/index.ts"],"sourcesContent":["/**\n * Video Activity (Experimental)\n *\n * Generates videos from text prompts using a jobs/polling architecture.\n * This is a self-contained module with implementation, types, and JSDoc.\n *\n * @experimental Video generation is an experimental feature and may change.\n */\n\nimport { aiEventClient } from '../../event-client.js'\nimport type { VideoAdapter } from './adapter'\nimport type {\n  VideoJobResult,\n  VideoStatusResult,\n  VideoUrlResult,\n} from '../../types'\n\n// ===========================\n// Activity Kind\n// ===========================\n\n/** The adapter kind this activity handles */\nexport const kind = 'video' as const\n\n// ===========================\n// Type Extraction Helpers\n// ===========================\n\n/**\n * Extract provider options from a VideoAdapter via ~types.\n */\nexport type VideoProviderOptions<TAdapter> =\n  TAdapter extends VideoAdapter<any, any, any, any>\n    ? TAdapter['~types']['providerOptions']\n    : object\n\n/**\n * Extract the size type for a VideoAdapter's model via ~types.\n */\nexport type VideoSizeForAdapter<TAdapter> =\n  TAdapter extends VideoAdapter<infer TModel, any, any, infer TSizeMap>\n    ? TModel extends keyof TSizeMap\n      ? TSizeMap[TModel]\n      : string\n    : string\n\n// ===========================\n// Activity Options Types\n\nfunction createId(prefix: string): string {\n  return `${prefix}-${Date.now()}-${Math.random().toString(36).slice(2, 9)}`\n}\n// ===========================\n\n/**\n * Base options shared by all video activity operations.\n * The model is extracted from the adapter's model property.\n */\ninterface VideoActivityBaseOptions<\n  TAdapter extends VideoAdapter<string, any, any, any>,\n> {\n  /** The video adapter to use (must be created with a model) */\n  adapter: TAdapter & { kind: typeof kind }\n}\n\n/**\n * Options for creating a new video generation job.\n * The model is extracted from the adapter's model property.\n *\n * @experimental Video generation is an experimental feature and may change.\n */\nexport type VideoCreateOptions<\n  TAdapter extends VideoAdapter<string, any, any, any>,\n> = VideoActivityBaseOptions<TAdapter> & {\n  /** Request type - create a new job (default if not specified) */\n  request?: 'create'\n  /** Text description of the desired video */\n  prompt: string\n  /** Video size — format depends on the provider (e.g., \"16:9\", \"1280x720\") */\n  size?: VideoSizeForAdapter<TAdapter>\n  /** Video duration in seconds */\n  duration?: number\n} & ({} extends VideoProviderOptions<TAdapter>\n    ? {\n        /** Provider-specific options for video generation */ modelOptions?: VideoProviderOptions<TAdapter>\n      }\n    : {\n        /** Provider-specific options for video generation */ modelOptions: VideoProviderOptions<TAdapter>\n      })\n\n/**\n * Options for polling the status of a video generation job.\n *\n * @experimental Video generation is an experimental feature and may change.\n */\nexport interface VideoStatusOptions<\n  TAdapter extends VideoAdapter<string, any, any, any>,\n> extends VideoActivityBaseOptions<TAdapter> {\n  /** Request type - get job status */\n  request: 'status'\n  /** The job ID to check status for */\n  jobId: string\n}\n\n/**\n * Options for getting the URL of a completed video.\n *\n * @experimental Video generation is an experimental feature and may change.\n */\nexport interface VideoUrlOptions<\n  TAdapter extends VideoAdapter<string, any, any, any>,\n> extends VideoActivityBaseOptions<TAdapter> {\n  /** Request type - get video URL */\n  request: 'url'\n  /** The job ID to get URL for */\n  jobId: string\n}\n\n/**\n * Union type for all video activity options.\n * Discriminated by the `request` field.\n *\n * @experimental Video generation is an experimental feature and may change.\n */\nexport type VideoActivityOptions<\n  TAdapter extends VideoAdapter<string, any, any, any>,\n  TRequest extends 'create' | 'status' | 'url' = 'create',\n> = TRequest extends 'status'\n  ? VideoStatusOptions<TAdapter>\n  : TRequest extends 'url'\n    ? VideoUrlOptions<TAdapter>\n    : VideoCreateOptions<TAdapter>\n\n// ===========================\n// Activity Result Types\n// ===========================\n\n/**\n * Result type for the video activity, based on request type.\n *\n * @experimental Video generation is an experimental feature and may change.\n */\nexport type VideoActivityResult<\n  TRequest extends 'create' | 'status' | 'url' = 'create',\n> = TRequest extends 'status'\n  ? Promise<VideoStatusResult>\n  : TRequest extends 'url'\n    ? Promise<VideoUrlResult>\n    : Promise<VideoJobResult>\n\n// ===========================\n// Activity Implementation\n// ===========================\n\n/**\n * Generate video - creates a video generation job from a text prompt.\n *\n * Uses AI video generation models to create videos based on natural language descriptions.\n * Unlike image generation, video generation is asynchronous and requires polling for completion.\n *\n * @experimental Video generation is an experimental feature and may change.\n *\n * @example Create a video generation job\n * ```ts\n * import { generateVideo } from '@tanstack/ai'\n * import { openaiVideo } from '@tanstack/ai-openai'\n *\n * // Start a video generation job\n * const { jobId } = await generateVideo({\n *   adapter: openaiVideo('sora-2'),\n *   prompt: 'A cat chasing a dog in a sunny park'\n * })\n *\n * console.log('Job started:', jobId)\n * ```\n */\nexport async function generateVideo<\n  TAdapter extends VideoAdapter<string, any, any, any>,\n>(options: VideoCreateOptions<TAdapter>): Promise<VideoJobResult> {\n  const { adapter, prompt, size, duration, modelOptions } = options\n  const model = adapter.model\n\n  return adapter.createVideoJob({\n    model,\n    prompt,\n    size,\n    duration,\n    modelOptions,\n  })\n}\n\n/**\n * Get video job status - returns the current status, progress, and URL if available.\n *\n * This function combines status checking and URL retrieval. If the job is completed,\n * it will automatically fetch and include the video URL.\n *\n * @experimental Video generation is an experimental feature and may change.\n *\n * @example Check job status\n * ```ts\n * import { getVideoJobStatus } from '@tanstack/ai'\n * import { openaiVideo } from '@tanstack/ai-openai'\n *\n * const result = await getVideoJobStatus({\n *   adapter: openaiVideo('sora-2'),\n *   jobId: 'job-123'\n * })\n *\n * console.log('Status:', result.status)\n * console.log('Progress:', result.progress)\n * if (result.url) {\n *   console.log('Video URL:', result.url)\n * }\n * ```\n */\nexport async function getVideoJobStatus<\n  TAdapter extends VideoAdapter<string, any, any, any>,\n>(options: {\n  adapter: TAdapter & { kind: typeof kind }\n  jobId: string\n}): Promise<{\n  status: 'pending' | 'processing' | 'completed' | 'failed'\n  progress?: number\n  url?: string\n  error?: string\n}> {\n  const { adapter, jobId } = options\n  const requestId = createId('video-status')\n  const startTime = Date.now()\n\n  aiEventClient.emit('video:request:started', {\n    requestId,\n    provider: adapter.name,\n    model: adapter.model,\n    requestType: 'status',\n    jobId,\n    timestamp: startTime,\n  })\n\n  // Get status first\n  const statusResult = await adapter.getVideoStatus(jobId)\n\n  // If completed, also get the URL\n  if (statusResult.status === 'completed') {\n    try {\n      const urlResult = await adapter.getVideoUrl(jobId)\n      aiEventClient.emit('video:request:completed', {\n        requestId,\n        provider: adapter.name,\n        model: adapter.model,\n        requestType: 'status',\n        jobId,\n        status: statusResult.status,\n        progress: statusResult.progress,\n        url: urlResult.url,\n        duration: Date.now() - startTime,\n        timestamp: Date.now(),\n      })\n      return {\n        status: statusResult.status,\n        progress: statusResult.progress,\n        url: urlResult.url,\n      }\n    } catch (error) {\n      aiEventClient.emit('video:request:completed', {\n        requestId,\n        provider: adapter.name,\n        model: adapter.model,\n        requestType: 'status',\n        jobId,\n        status: statusResult.status,\n        progress: statusResult.progress,\n        error:\n          error instanceof Error ? error.message : 'Failed to get video URL',\n        duration: Date.now() - startTime,\n        timestamp: Date.now(),\n      })\n      // If URL fetch fails, still return status\n      return {\n        status: statusResult.status,\n        progress: statusResult.progress,\n        error:\n          error instanceof Error ? error.message : 'Failed to get video URL',\n      }\n    }\n  }\n\n  aiEventClient.emit('video:request:completed', {\n    requestId,\n    provider: adapter.name,\n    model: adapter.model,\n    requestType: 'status',\n    jobId,\n    status: statusResult.status,\n    progress: statusResult.progress,\n    error: statusResult.error,\n    duration: Date.now() - startTime,\n    timestamp: Date.now(),\n  })\n\n  // Return status for non-completed jobs\n  return {\n    status: statusResult.status,\n    progress: statusResult.progress,\n    error: statusResult.error,\n  }\n}\n\n// ===========================\n// Options Factory\n// ===========================\n\n/**\n * Create typed options for the generateVideo() function without executing.\n */\nexport function createVideoOptions<\n  TAdapter extends VideoAdapter<string, any, any, any>,\n>(options: VideoCreateOptions<TAdapter>): VideoCreateOptions<TAdapter> {\n  return options\n}\n\n// Re-export adapter types\nexport type {\n  VideoAdapter,\n  VideoAdapterConfig,\n  AnyVideoAdapter,\n} from './adapter'\nexport { BaseVideoAdapter } from './adapter'\n"],"names":[],"mappings":";AAsBO,MAAM,OAAO;AA2BpB,SAAS,SAAS,QAAwB;AACxC,SAAO,GAAG,MAAM,IAAI,KAAK,IAAA,CAAK,IAAI,KAAK,OAAA,EAAS,SAAS,EAAE,EAAE,MAAM,GAAG,CAAC,CAAC;AAC1E;AA6HA,eAAsB,cAEpB,SAAgE;AAChE,QAAM,EAAE,SAAS,QAAQ,MAAM,UAAU,iBAAiB;AAC1D,QAAM,QAAQ,QAAQ;AAEtB,SAAO,QAAQ,eAAe;AAAA,IAC5B;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EAAA,CACD;AACH;AA2BA,eAAsB,kBAEpB,SAQC;AACD,QAAM,EAAE,SAAS,MAAA,IAAU;AAC3B,QAAM,YAAY,SAAS,cAAc;AACzC,QAAM,YAAY,KAAK,IAAA;AAEvB,gBAAc,KAAK,yBAAyB;AAAA,IAC1C;AAAA,IACA,UAAU,QAAQ;AAAA,IAClB,OAAO,QAAQ;AAAA,IACf,aAAa;AAAA,IACb;AAAA,IACA,WAAW;AAAA,EAAA,CACZ;AAGD,QAAM,eAAe,MAAM,QAAQ,eAAe,KAAK;AAGvD,MAAI,aAAa,WAAW,aAAa;AACvC,QAAI;AACF,YAAM,YAAY,MAAM,QAAQ,YAAY,KAAK;AACjD,oBAAc,KAAK,2BAA2B;AAAA,QAC5C;AAAA,QACA,UAAU,QAAQ;AAAA,QAClB,OAAO,QAAQ;AAAA,QACf,aAAa;AAAA,QACb;AAAA,QACA,QAAQ,aAAa;AAAA,QACrB,UAAU,aAAa;AAAA,QACvB,KAAK,UAAU;AAAA,QACf,UAAU,KAAK,IAAA,IAAQ;AAAA,QACvB,WAAW,KAAK,IAAA;AAAA,MAAI,CACrB;AACD,aAAO;AAAA,QACL,QAAQ,aAAa;AAAA,QACrB,UAAU,aAAa;AAAA,QACvB,KAAK,UAAU;AAAA,MAAA;AAAA,IAEnB,SAAS,OAAO;AACd,oBAAc,KAAK,2BAA2B;AAAA,QAC5C;AAAA,QACA,UAAU,QAAQ;AAAA,QAClB,OAAO,QAAQ;AAAA,QACf,aAAa;AAAA,QACb;AAAA,QACA,QAAQ,aAAa;AAAA,QACrB,UAAU,aAAa;AAAA,QACvB,OACE,iBAAiB,QAAQ,MAAM,UAAU;AAAA,QAC3C,UAAU,KAAK,IAAA,IAAQ;AAAA,QACvB,WAAW,KAAK,IAAA;AAAA,MAAI,CACrB;AAED,aAAO;AAAA,QACL,QAAQ,aAAa;AAAA,QACrB,UAAU,aAAa;AAAA,QACvB,OACE,iBAAiB,QAAQ,MAAM,UAAU;AAAA,MAAA;AAAA,IAE/C;AAAA,EACF;AAEA,gBAAc,KAAK,2BAA2B;AAAA,IAC5C;AAAA,IACA,UAAU,QAAQ;AAAA,IAClB,OAAO,QAAQ;AAAA,IACf,aAAa;AAAA,IACb;AAAA,IACA,QAAQ,aAAa;AAAA,IACrB,UAAU,aAAa;AAAA,IACvB,OAAO,aAAa;AAAA,IACpB,UAAU,KAAK,IAAA,IAAQ;AAAA,IACvB,WAAW,KAAK,IAAA;AAAA,EAAI,CACrB;AAGD,SAAO;AAAA,IACL,QAAQ,aAAa;AAAA,IACrB,UAAU,aAAa;AAAA,IACvB,OAAO,aAAa;AAAA,EAAA;AAExB;AASO,SAAS,mBAEd,SAAqE;AACrE,SAAO;AACT;"}
+{"version":3,"file":"index.js","sources":["../../../../src/activities/generateVideo/index.ts"],"sourcesContent":["/**\n * Video Activity (Experimental)\n *\n * Generates videos from text prompts using a jobs/polling architecture.\n * This is a self-contained module with implementation, types, and JSDoc.\n *\n * @experimental Video generation is an experimental feature and may change.\n */\n\nimport { aiEventClient } from '@tanstack/ai-event-client'\nimport type { VideoAdapter } from './adapter'\nimport type {\n  StreamChunk,\n  VideoJobResult,\n  VideoStatusResult,\n  VideoUrlResult,\n} from '../../types'\n\n// ===========================\n// Activity Kind\n// ===========================\n\n/** The adapter kind this activity handles */\nexport const kind = 'video' as const\n\n// ===========================\n// Type Extraction Helpers\n// ===========================\n\n/**\n * Extract provider options from a VideoAdapter via ~types.\n */\nexport type VideoProviderOptions<TAdapter> =\n  TAdapter extends VideoAdapter<any, any, any, any>\n    ? TAdapter['~types']['providerOptions']\n    : object\n\n/**\n * Extract the size type for a VideoAdapter's model via ~types.\n */\nexport type VideoSizeForAdapter<TAdapter> =\n  TAdapter extends VideoAdapter<infer TModel, any, any, infer TSizeMap>\n    ? TModel extends keyof TSizeMap\n      ? TSizeMap[TModel]\n      : string\n    : string\n\n// ===========================\n// Activity Options Types\n\nfunction createId(prefix: string): string {\n  return `${prefix}-${Date.now()}-${Math.random().toString(36).slice(2, 9)}`\n}\n// ===========================\n\n/**\n * Base options shared by all video activity operations.\n * The model is extracted from the adapter's model property.\n */\ninterface VideoActivityBaseOptions<\n  TAdapter extends VideoAdapter<string, any, any, any>,\n> {\n  /** The video adapter to use (must be created with a model) */\n  adapter: TAdapter & { kind: typeof kind }\n}\n\n/**\n * Options for creating a new video generation job.\n * The model is extracted from the adapter's model property.\n *\n * @template TAdapter - The video adapter type\n * @template TStream - Whether to stream the output\n *\n * @experimental Video generation is an experimental feature and may change.\n */\nexport type VideoCreateOptions<\n  TAdapter extends VideoAdapter<string, any, any, any>,\n  TStream extends boolean = false,\n> = VideoActivityBaseOptions<TAdapter> & {\n  /** Request type - create a new job (default if not specified) */\n  request?: 'create'\n  /** Text description of the desired video */\n  prompt: string\n  /** Video size — format depends on the provider (e.g., \"16:9\", \"1280x720\") */\n  size?: VideoSizeForAdapter<TAdapter>\n  /** Video duration in seconds */\n  duration?: number\n  /**\n   * Whether to stream the video generation lifecycle.\n   * When true, returns an AsyncIterable<StreamChunk> that handles the full\n   * job lifecycle: create job, poll for status, yield updates, and yield final result.\n   * When false or not provided, returns a Promise<VideoJobResult>.\n   *\n   * @default false\n   */\n  stream?: TStream\n  /** Polling interval in milliseconds (stream mode only). @default 2000 */\n  pollingInterval?: number\n  /** Maximum time to wait before timing out in milliseconds (stream mode only). @default 600000 */\n  maxDuration?: number\n  /** Custom run ID (stream mode only) */\n  runId?: string\n} & ({} extends VideoProviderOptions<TAdapter>\n    ? {\n        /** Provider-specific options for video generation */ modelOptions?: VideoProviderOptions<TAdapter>\n      }\n    : {\n        /** Provider-specific options for video generation */ modelOptions: VideoProviderOptions<TAdapter>\n      })\n\n/**\n * Options for polling the status of a video generation job.\n *\n * @experimental Video generation is an experimental feature and may change.\n */\nexport interface VideoStatusOptions<\n  TAdapter extends VideoAdapter<string, any, any, any>,\n> extends VideoActivityBaseOptions<TAdapter> {\n  /** Request type - get job status */\n  request: 'status'\n  /** The job ID to check status for */\n  jobId: string\n}\n\n/**\n * Options for getting the URL of a completed video.\n *\n * @experimental Video generation is an experimental feature and may change.\n */\nexport interface VideoUrlOptions<\n  TAdapter extends VideoAdapter<string, any, any, any>,\n> extends VideoActivityBaseOptions<TAdapter> {\n  /** Request type - get video URL */\n  request: 'url'\n  /** The job ID to get URL for */\n  jobId: string\n}\n\n/**\n * Union type for all video activity options.\n * Discriminated by the `request` field.\n *\n * @experimental Video generation is an experimental feature and may change.\n */\nexport type VideoActivityOptions<\n  TAdapter extends VideoAdapter<string, any, any, any>,\n  TRequest extends 'create' | 'status' | 'url' = 'create',\n  TStream extends boolean = false,\n> = TRequest extends 'status'\n  ? VideoStatusOptions<TAdapter>\n  : TRequest extends 'url'\n    ? VideoUrlOptions<TAdapter>\n    : VideoCreateOptions<TAdapter, TStream>\n\n// ===========================\n// Activity Result Types\n// ===========================\n\n/**\n * Result type for the video activity, based on request type and streaming.\n * - If stream is true (create request): AsyncIterable<StreamChunk>\n * - Otherwise: Promise<VideoJobResult | VideoStatusResult | VideoUrlResult>\n *\n * @experimental Video generation is an experimental feature and may change.\n */\nexport type VideoActivityResult<\n  TRequest extends 'create' | 'status' | 'url' = 'create',\n  TStream extends boolean = false,\n> = TRequest extends 'status'\n  ? Promise<VideoStatusResult>\n  : TRequest extends 'url'\n    ? Promise<VideoUrlResult>\n    : TStream extends true\n      ? AsyncIterable<StreamChunk>\n      : Promise<VideoJobResult>\n\n// ===========================\n// Activity Implementation\n// ===========================\n\n/**\n * Generate video - creates a video generation job from a text prompt.\n *\n * Uses AI video generation models to create videos based on natural language descriptions.\n * Unlike image generation, video generation is asynchronous and requires polling for completion.\n *\n * When `stream: true` is passed, handles the full job lifecycle automatically:\n * create job → poll for status → stream updates → yield final result.\n *\n * @experimental Video generation is an experimental feature and may change.\n *\n * @example Create a video generation job\n * ```ts\n * import { generateVideo } from '@tanstack/ai'\n * import { openaiVideo } from '@tanstack/ai-openai'\n *\n * // Start a video generation job\n * const { jobId } = await generateVideo({\n *   adapter: openaiVideo('sora-2'),\n *   prompt: 'A cat chasing a dog in a sunny park'\n * })\n *\n * console.log('Job started:', jobId)\n * ```\n *\n * @example Stream the full video generation lifecycle\n * ```ts\n * import { generateVideo, toServerSentEventsResponse } from '@tanstack/ai'\n * import { openaiVideo } from '@tanstack/ai-openai'\n *\n * const stream = generateVideo({\n *   adapter: openaiVideo('sora-2'),\n *   prompt: 'A cat chasing a dog in a sunny park',\n *   stream: true,\n *   pollingInterval: 3000,\n * })\n *\n * return toServerSentEventsResponse(stream)\n * ```\n */\nexport function generateVideo<\n  TAdapter extends VideoAdapter<string, any, any, any>,\n  TStream extends boolean = false,\n>(\n  options: VideoCreateOptions<TAdapter, TStream>,\n): VideoActivityResult<'create', TStream> {\n  if (options.stream) {\n    return runStreamingVideoGeneration(\n      options as VideoCreateOptions<TAdapter, true>,\n    ) as VideoActivityResult<'create', TStream>\n  }\n\n  return runCreateVideoJob(options) as VideoActivityResult<'create', TStream>\n}\n\n/**\n * Internal implementation of non-streaming video job creation.\n */\nasync function runCreateVideoJob<\n  TAdapter extends VideoAdapter<string, any, any, any>,\n>(options: VideoCreateOptions<TAdapter, boolean>): Promise<VideoJobResult> {\n  const { adapter, prompt, size, duration, modelOptions } = options\n  const model = adapter.model\n\n  return adapter.createVideoJob({\n    model,\n    prompt,\n    size,\n    duration,\n    modelOptions,\n  })\n}\n\nfunction sleep(ms: number): Promise<void> {\n  return new Promise((resolve) => setTimeout(resolve, ms))\n}\n\n/**\n * Internal streaming implementation for video generation.\n * Handles the full job lifecycle: create job → poll for status → stream updates → yield final result.\n */\nasync function* runStreamingVideoGeneration<\n  TAdapter extends VideoAdapter<string, any, any, any>,\n>(options: VideoCreateOptions<TAdapter, true>): AsyncIterable<StreamChunk> {\n  const { adapter, prompt, size, duration, modelOptions } = options\n  const model = adapter.model\n  const runId = options.runId ?? createId('run')\n  const pollingInterval = options.pollingInterval ?? 2000\n  const maxDuration = options.maxDuration ?? 600_000\n\n  yield {\n    type: 'RUN_STARTED',\n    runId,\n    timestamp: Date.now(),\n  }\n\n  try {\n    // Create the video generation job\n    const jobResult = await adapter.createVideoJob({\n      model,\n      prompt,\n      size,\n      duration,\n      modelOptions,\n    })\n\n    yield {\n      type: 'CUSTOM',\n      name: 'video:job:created',\n      value: { jobId: jobResult.jobId },\n      timestamp: Date.now(),\n    }\n\n    // Poll for completion\n    const startTime = Date.now()\n    while (Date.now() - startTime < maxDuration) {\n      await sleep(pollingInterval)\n\n      const statusResult = await adapter.getVideoStatus(jobResult.jobId)\n\n      yield {\n        type: 'CUSTOM',\n        name: 'video:status',\n        value: {\n          jobId: jobResult.jobId,\n          status: statusResult.status,\n          progress: statusResult.progress,\n          error: statusResult.error,\n        },\n        timestamp: Date.now(),\n      }\n\n      if (statusResult.status === 'completed') {\n        const urlResult = await adapter.getVideoUrl(jobResult.jobId)\n\n        yield {\n          type: 'CUSTOM',\n          name: 'generation:result',\n          value: {\n            jobId: jobResult.jobId,\n            status: 'completed',\n            url: urlResult.url,\n            expiresAt: urlResult.expiresAt,\n          },\n          timestamp: Date.now(),\n        }\n\n        yield {\n          type: 'RUN_FINISHED',\n          runId,\n          finishReason: 'stop',\n          timestamp: Date.now(),\n        }\n        return\n      }\n\n      if (statusResult.status === 'failed') {\n        throw new Error(statusResult.error || 'Video generation failed')\n      }\n    }\n\n    throw new Error('Video generation timed out')\n  } catch (error: any) {\n    yield {\n      type: 'RUN_ERROR',\n      runId,\n      error: {\n        message: error.message || 'Video generation failed',\n        code: error.code,\n      },\n      timestamp: Date.now(),\n    }\n  }\n}\n\n/**\n * Get video job status - returns the current status, progress, and URL if available.\n *\n * This function combines status checking and URL retrieval. If the job is completed,\n * it will automatically fetch and include the video URL.\n *\n * @experimental Video generation is an experimental feature and may change.\n *\n * @example Check job status\n * ```ts\n * import { getVideoJobStatus } from '@tanstack/ai'\n * import { openaiVideo } from '@tanstack/ai-openai'\n *\n * const result = await getVideoJobStatus({\n *   adapter: openaiVideo('sora-2'),\n *   jobId: 'job-123'\n * })\n *\n * console.log('Status:', result.status)\n * console.log('Progress:', result.progress)\n * if (result.url) {\n *   console.log('Video URL:', result.url)\n * }\n * ```\n */\nexport async function getVideoJobStatus<\n  TAdapter extends VideoAdapter<string, any, any, any>,\n>(options: {\n  adapter: TAdapter & { kind: typeof kind }\n  jobId: string\n}): Promise<{\n  status: 'pending' | 'processing' | 'completed' | 'failed'\n  progress?: number\n  url?: string\n  error?: string\n}> {\n  const { adapter, jobId } = options\n  const requestId = createId('video-status')\n  const startTime = Date.now()\n\n  aiEventClient.emit('video:request:started', {\n    requestId,\n    provider: adapter.name,\n    model: adapter.model,\n    requestType: 'status',\n    jobId,\n    timestamp: startTime,\n  })\n\n  // Get status first\n  const statusResult = await adapter.getVideoStatus(jobId)\n\n  // If completed, also get the URL\n  if (statusResult.status === 'completed') {\n    try {\n      const urlResult = await adapter.getVideoUrl(jobId)\n      aiEventClient.emit('video:request:completed', {\n        requestId,\n        provider: adapter.name,\n        model: adapter.model,\n        requestType: 'status',\n        jobId,\n        status: statusResult.status,\n        progress: statusResult.progress,\n        url: urlResult.url,\n        duration: Date.now() - startTime,\n        timestamp: Date.now(),\n      })\n      return {\n        status: statusResult.status,\n        progress: statusResult.progress,\n        url: urlResult.url,\n      }\n    } catch (error) {\n      aiEventClient.emit('video:request:completed', {\n        requestId,\n        provider: adapter.name,\n        model: adapter.model,\n        requestType: 'status',\n        jobId,\n        status: statusResult.status,\n        progress: statusResult.progress,\n        error:\n          error instanceof Error ? error.message : 'Failed to get video URL',\n        duration: Date.now() - startTime,\n        timestamp: Date.now(),\n      })\n      // If URL fetch fails, still return status\n      return {\n        status: statusResult.status,\n        progress: statusResult.progress,\n        error:\n          error instanceof Error ? error.message : 'Failed to get video URL',\n      }\n    }\n  }\n\n  aiEventClient.emit('video:request:completed', {\n    requestId,\n    provider: adapter.name,\n    model: adapter.model,\n    requestType: 'status',\n    jobId,\n    status: statusResult.status,\n    progress: statusResult.progress,\n    error: statusResult.error,\n    duration: Date.now() - startTime,\n    timestamp: Date.now(),\n  })\n\n  // Return status for non-completed jobs\n  return {\n    status: statusResult.status,\n    progress: statusResult.progress,\n    error: statusResult.error,\n  }\n}\n\n// ===========================\n// Options Factory\n// ===========================\n\n/**\n * Create typed options for the generateVideo() function without executing.\n */\nexport function createVideoOptions<\n  TAdapter extends VideoAdapter<string, any, any, any>,\n  TStream extends boolean = false,\n>(\n  options: VideoCreateOptions<TAdapter, TStream>,\n): VideoCreateOptions<TAdapter, TStream> {\n  return options\n}\n\n// Re-export adapter types\nexport type {\n  VideoAdapter,\n  VideoAdapterConfig,\n  AnyVideoAdapter,\n} from './adapter'\nexport { BaseVideoAdapter } from './adapter'\n"],"names":[],"mappings":";AAuBO,MAAM,OAAO;AA2BpB,SAAS,SAAS,QAAwB;AACxC,SAAO,GAAG,MAAM,IAAI,KAAK,IAAA,CAAK,IAAI,KAAK,OAAA,EAAS,SAAS,EAAE,EAAE,MAAM,GAAG,CAAC,CAAC;AAC1E;AAwKO,SAAS,cAId,SACwC;AACxC,MAAI,QAAQ,QAAQ;AAClB,WAAO;AAAA,MACL;AAAA,IAAA;AAAA,EAEJ;AAEA,SAAO,kBAAkB,OAAO;AAClC;AAKA,eAAe,kBAEb,SAAyE;AACzE,QAAM,EAAE,SAAS,QAAQ,MAAM,UAAU,iBAAiB;AAC1D,QAAM,QAAQ,QAAQ;AAEtB,SAAO,QAAQ,eAAe;AAAA,IAC5B;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EAAA,CACD;AACH;AAEA,SAAS,MAAM,IAA2B;AACxC,SAAO,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,EAAE,CAAC;AACzD;AAMA,gBAAgB,4BAEd,SAAyE;AACzE,QAAM,EAAE,SAAS,QAAQ,MAAM,UAAU,iBAAiB;AAC1D,QAAM,QAAQ,QAAQ;AACtB,QAAM,QAAQ,QAAQ,SAAS,SAAS,KAAK;AAC7C,QAAM,kBAAkB,QAAQ,mBAAmB;AACnD,QAAM,cAAc,QAAQ,eAAe;AAE3C,QAAM;AAAA,IACJ,MAAM;AAAA,IACN;AAAA,IACA,WAAW,KAAK,IAAA;AAAA,EAAI;AAGtB,MAAI;AAEF,UAAM,YAAY,MAAM,QAAQ,eAAe;AAAA,MAC7C;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IAAA,CACD;AAED,UAAM;AAAA,MACJ,MAAM;AAAA,MACN,MAAM;AAAA,MACN,OAAO,EAAE,OAAO,UAAU,MAAA;AAAA,MAC1B,WAAW,KAAK,IAAA;AAAA,IAAI;AAItB,UAAM,YAAY,KAAK,IAAA;AACvB,WAAO,KAAK,QAAQ,YAAY,aAAa;AAC3C,YAAM,MAAM,eAAe;AAE3B,YAAM,eAAe,MAAM,QAAQ,eAAe,UAAU,KAAK;AAEjE,YAAM;AAAA,QACJ,MAAM;AAAA,QACN,MAAM;AAAA,QACN,OAAO;AAAA,UACL,OAAO,UAAU;AAAA,UACjB,QAAQ,aAAa;AAAA,UACrB,UAAU,aAAa;AAAA,UACvB,OAAO,aAAa;AAAA,QAAA;AAAA,QAEtB,WAAW,KAAK,IAAA;AAAA,MAAI;AAGtB,UAAI,aAAa,WAAW,aAAa;AACvC,cAAM,YAAY,MAAM,QAAQ,YAAY,UAAU,KAAK;AAE3D,cAAM;AAAA,UACJ,MAAM;AAAA,UACN,MAAM;AAAA,UACN,OAAO;AAAA,YACL,OAAO,UAAU;AAAA,YACjB,QAAQ;AAAA,YACR,KAAK,UAAU;AAAA,YACf,WAAW,UAAU;AAAA,UAAA;AAAA,UAEvB,WAAW,KAAK,IAAA;AAAA,QAAI;AAGtB,cAAM;AAAA,UACJ,MAAM;AAAA,UACN;AAAA,UACA,cAAc;AAAA,UACd,WAAW,KAAK,IAAA;AAAA,QAAI;AAEtB;AAAA,MACF;AAEA,UAAI,aAAa,WAAW,UAAU;AACpC,cAAM,IAAI,MAAM,aAAa,SAAS,yBAAyB;AAAA,MACjE;AAAA,IACF;AAEA,UAAM,IAAI,MAAM,4BAA4B;AAAA,EAC9C,SAAS,OAAY;AACnB,UAAM;AAAA,MACJ,MAAM;AAAA,MACN;AAAA,MACA,OAAO;AAAA,QACL,SAAS,MAAM,WAAW;AAAA,QAC1B,MAAM,MAAM;AAAA,MAAA;AAAA,MAEd,WAAW,KAAK,IAAA;AAAA,IAAI;AAAA,EAExB;AACF;AA2BA,eAAsB,kBAEpB,SAQC;AACD,QAAM,EAAE,SAAS,MAAA,IAAU;AAC3B,QAAM,YAAY,SAAS,cAAc;AACzC,QAAM,YAAY,KAAK,IAAA;AAEvB,gBAAc,KAAK,yBAAyB;AAAA,IAC1C;AAAA,IACA,UAAU,QAAQ;AAAA,IAClB,OAAO,QAAQ;AAAA,IACf,aAAa;AAAA,IACb;AAAA,IACA,WAAW;AAAA,EAAA,CACZ;AAGD,QAAM,eAAe,MAAM,QAAQ,eAAe,KAAK;AAGvD,MAAI,aAAa,WAAW,aAAa;AACvC,QAAI;AACF,YAAM,YAAY,MAAM,QAAQ,YAAY,KAAK;AACjD,oBAAc,KAAK,2BAA2B;AAAA,QAC5C;AAAA,QACA,UAAU,QAAQ;AAAA,QAClB,OAAO,QAAQ;AAAA,QACf,aAAa;AAAA,QACb;AAAA,QACA,QAAQ,aAAa;AAAA,QACrB,UAAU,aAAa;AAAA,QACvB,KAAK,UAAU;AAAA,QACf,UAAU,KAAK,IAAA,IAAQ;AAAA,QACvB,WAAW,KAAK,IAAA;AAAA,MAAI,CACrB;AACD,aAAO;AAAA,QACL,QAAQ,aAAa;AAAA,QACrB,UAAU,aAAa;AAAA,QACvB,KAAK,UAAU;AAAA,MAAA;AAAA,IAEnB,SAAS,OAAO;AACd,oBAAc,KAAK,2BAA2B;AAAA,QAC5C;AAAA,QACA,UAAU,QAAQ;AAAA,QAClB,OAAO,QAAQ;AAAA,QACf,aAAa;AAAA,QACb;AAAA,QACA,QAAQ,aAAa;AAAA,QACrB,UAAU,aAAa;AAAA,QACvB,OACE,iBAAiB,QAAQ,MAAM,UAAU;AAAA,QAC3C,UAAU,KAAK,IAAA,IAAQ;AAAA,QACvB,WAAW,KAAK,IAAA;AAAA,MAAI,CACrB;AAED,aAAO;AAAA,QACL,QAAQ,aAAa;AAAA,QACrB,UAAU,aAAa;AAAA,QACvB,OACE,iBAAiB,QAAQ,MAAM,UAAU;AAAA,MAAA;AAAA,IAE/C;AAAA,EACF;AAEA,gBAAc,KAAK,2BAA2B;AAAA,IAC5C;AAAA,IACA,UAAU,QAAQ;AAAA,IAClB,OAAO,QAAQ;AAAA,IACf,aAAa;AAAA,IACb;AAAA,IACA,QAAQ,aAAa;AAAA,IACrB,UAAU,aAAa;AAAA,IACvB,OAAO,aAAa;AAAA,IACpB,UAAU,KAAK,IAAA,IAAQ;AAAA,IACvB,WAAW,KAAK,IAAA;AAAA,EAAI,CACrB;AAGD,SAAO;AAAA,IACL,QAAQ,aAAa;AAAA,IACrB,UAAU,aAAa;AAAA,IACvB,OAAO,aAAa;AAAA,EAAA;AAExB;AASO,SAAS,mBAId,SACuC;AACvC,SAAO;AACT;"}

package/dist/esm/activities/stream-generation-result.d.ts

@@ -0,0 +1,14 @@
+import { StreamChunk } from '../types.js';
+/**
+ * Wrap a one-shot generation result as a StreamChunk async iterable.
+ *
+ * This allows non-streaming activities (image, speech, transcription, summarize)
+ * to be sent over the same streaming transport as chat.
+ *
+ * @param generator - An async function that performs the generation and returns the result
+ * @param options - Optional configuration (runId)
+ * @returns An AsyncIterable of StreamChunks with RUN_STARTED, CUSTOM(generation:result), and RUN_FINISHED events
+ */
+export declare function streamGenerationResult<TResult>(generator: () => Promise<TResult>, options?: {
+    runId?: string;
+}): AsyncIterable<StreamChunk>;

package/dist/esm/activities/stream-generation-result.js

@@ -0,0 +1,40 @@
+function createId(prefix) {
+  return `${prefix}-${Date.now()}-${Math.random().toString(36).slice(2, 9)}`;
+}
+async function* streamGenerationResult(generator, options) {
+  const runId = createId("run");
+  yield {
+    type: "RUN_STARTED",
+    runId,
+    timestamp: Date.now()
+  };
+  try {
+    const result = await generator();
+    yield {
+      type: "CUSTOM",
+      name: "generation:result",
+      value: result,
+      timestamp: Date.now()
+    };
+    yield {
+      type: "RUN_FINISHED",
+      runId,
+      finishReason: "stop",
+      timestamp: Date.now()
+    };
+  } catch (error) {
+    yield {
+      type: "RUN_ERROR",
+      runId,
+      error: {
+        message: error.message || "Generation failed",
+        code: error.code
+      },
+      timestamp: Date.now()
+    };
+  }
+}
+export {
+  streamGenerationResult
+};
+//# sourceMappingURL=stream-generation-result.js.map

package/dist/esm/activities/stream-generation-result.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream-generation-result.js","sources":["../../../src/activities/stream-generation-result.ts"],"sourcesContent":["/**\n * Internal helper for wrapping one-shot generation results as StreamChunk\n * async iterables. NOT exported from the package — used only by activity\n * implementations to support `stream: true`.\n */\n\nimport type { StreamChunk } from '../types'\n\nfunction createId(prefix: string): string {\n  return `${prefix}-${Date.now()}-${Math.random().toString(36).slice(2, 9)}`\n}\n\n/**\n * Wrap a one-shot generation result as a StreamChunk async iterable.\n *\n * This allows non-streaming activities (image, speech, transcription, summarize)\n * to be sent over the same streaming transport as chat.\n *\n * @param generator - An async function that performs the generation and returns the result\n * @param options - Optional configuration (runId)\n * @returns An AsyncIterable of StreamChunks with RUN_STARTED, CUSTOM(generation:result), and RUN_FINISHED events\n */\nexport async function* streamGenerationResult<TResult>(\n  generator: () => Promise<TResult>,\n  options?: { runId?: string },\n): AsyncIterable<StreamChunk> {\n  const runId = options?.runId ?? createId('run')\n\n  yield {\n    type: 'RUN_STARTED',\n    runId,\n    timestamp: Date.now(),\n  }\n\n  try {\n    const result = await generator()\n\n    yield {\n      type: 'CUSTOM',\n      name: 'generation:result',\n      value: result as unknown,\n      timestamp: Date.now(),\n    }\n\n    yield {\n      type: 'RUN_FINISHED',\n      runId,\n      finishReason: 'stop',\n      timestamp: Date.now(),\n    }\n  } catch (error: any) {\n    yield {\n      type: 'RUN_ERROR',\n      runId,\n      error: {\n        message: error.message || 'Generation failed',\n        code: error.code,\n      },\n      timestamp: Date.now(),\n    }\n  }\n}\n"],"names":[],"mappings":"AAQA,SAAS,SAAS,QAAwB;AACxC,SAAO,GAAG,MAAM,IAAI,KAAK,IAAA,CAAK,IAAI,KAAK,OAAA,EAAS,SAAS,EAAE,EAAE,MAAM,GAAG,CAAC,CAAC;AAC1E;AAYA,gBAAuB,uBACrB,WACA,SAC4B;AAC5B,QAAM,QAA0B,SAAS,KAAK;AAE9C,QAAM;AAAA,IACJ,MAAM;AAAA,IACN;AAAA,IACA,WAAW,KAAK,IAAA;AAAA,EAAI;AAGtB,MAAI;AACF,UAAM,SAAS,MAAM,UAAA;AAErB,UAAM;AAAA,MACJ,MAAM;AAAA,MACN,MAAM;AAAA,MACN,OAAO;AAAA,MACP,WAAW,KAAK,IAAA;AAAA,IAAI;AAGtB,UAAM;AAAA,MACJ,MAAM;AAAA,MACN;AAAA,MACA,cAAc;AAAA,MACd,WAAW,KAAK,IAAA;AAAA,IAAI;AAAA,EAExB,SAAS,OAAY;AACnB,UAAM;AAAA,MACJ,MAAM;AAAA,MACN;AAAA,MACA,OAAO;AAAA,QACL,SAAS,MAAM,WAAW;AAAA,QAC1B,MAAM,MAAM;AAAA,MAAA;AAAA,MAEd,WAAW,KAAK,IAAA;AAAA,IAAI;AAAA,EAExB;AACF;"}

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

@@ -1,4 +1,5 @@
-import { aiEventClient } from "../../event-client.js";
+import { aiEventClient } from "@tanstack/ai-event-client";
+import { streamGenerationResult } from "../stream-generation-result.js";
 const kind = "summarize";
 function createId(prefix) {
   return `${prefix}-${Date.now()}-${Math.random().toString(36).slice(2, 9)}`;
@@ -62,23 +63,7 @@ async function* runStreamingSummarize(options) {
     yield* adapter.summarizeStream(summarizeOptions);
     return;
   }
-  const result = await adapter.summarize(summarizeOptions);
-  yield {
-    type: "TEXT_MESSAGE_CONTENT",
-    messageId: result.id,
-    model: result.model,
-    timestamp: Date.now(),
-    delta: result.summary,
-    content: result.summary
-  };
-  yield {
-    type: "RUN_FINISHED",
-    runId: result.id,
-    model: result.model,
-    timestamp: Date.now(),
-    finishReason: "stop",
-    usage: result.usage
-  };
+  yield* streamGenerationResult(() => adapter.summarize(summarizeOptions));
 }
 function createSummarizeOptions(options) {
   return options;

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

@@ -1 +1 @@
-{"version":3,"file":"index.js","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 '../../event-client.js'\nimport type { SummarizeAdapter } from './adapter'\nimport type {\n  StreamChunk,\n  SummarizationOptions,\n  SummarizationResult,\n} 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   * 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\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 unknown as SummarizeActivityOptions<\n        SummarizeAdapter<string, object>,\n        true\n      >,\n    ) as SummarizeActivityResult<TStream>\n  }\n\n  return runSummarize(\n    options as unknown 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 } = options\n  const model = adapter.model\n  const requestId = createId('summarize')\n  const inputLength = text.length\n  const startTime = Date.now()\n\n  aiEventClient.emit('summarize:request:started', {\n    requestId,\n    provider: adapter.name,\n    model,\n    inputLength,\n    timestamp: startTime,\n  })\n\n  const summarizeOptions: SummarizationOptions = {\n    model,\n    text,\n    maxLength,\n    style,\n    focus,\n  }\n\n  const result = await adapter.summarize(summarizeOptions)\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  return result\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 { adapter, text, maxLength, style, focus } = options\n  const model = adapter.model\n\n  const summarizeOptions: SummarizationOptions = {\n    model,\n    text,\n    maxLength,\n    style,\n    focus,\n  }\n\n  // Use real streaming if the adapter supports it\n  if (adapter.summarizeStream) {\n    yield* adapter.summarizeStream(summarizeOptions)\n    return\n  }\n\n  // Fall back to non-streaming and yield as a single chunk\n  const result = await adapter.summarize(summarizeOptions)\n\n  // Yield TEXT_MESSAGE_CONTENT event with the summary\n  yield {\n    type: 'TEXT_MESSAGE_CONTENT',\n    messageId: result.id,\n    model: result.model,\n    timestamp: Date.now(),\n    delta: result.summary,\n    content: result.summary,\n  }\n\n  // Yield RUN_FINISHED event\n  yield {\n    type: 'RUN_FINISHED',\n    runId: result.id,\n    model: result.model,\n    timestamp: Date.now(),\n    finishReason: 'stop',\n    usage: result.usage,\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'\n"],"names":[],"mappings":";AAoBO,MAAM,OAAO;AAmEpB,SAAS,SAAS,QAAwB;AACxC,SAAO,GAAG,MAAM,IAAI,KAAK,IAAA,CAAK,IAAI,KAAK,OAAA,EAAS,SAAS,EAAE,EAAE,MAAM,GAAG,CAAC,CAAC;AAC1E;AAwDO,SAAS,UAId,SACkC;AAClC,QAAM,EAAE,WAAW;AAEnB,MAAI,QAAQ;AACV,WAAO;AAAA,MACL;AAAA,IAAA;AAAA,EAKJ;AAEA,SAAO;AAAA,IACL;AAAA,EAAA;AAKJ;AAKA,eAAe,aACb,SAC8B;AAC9B,QAAM,EAAE,SAAS,MAAM,WAAW,OAAO,UAAU;AACnD,QAAM,QAAQ,QAAQ;AACtB,QAAM,YAAY,SAAS,WAAW;AACtC,QAAM,cAAc,KAAK;AACzB,QAAM,YAAY,KAAK,IAAA;AAEvB,gBAAc,KAAK,6BAA6B;AAAA,IAC9C;AAAA,IACA,UAAU,QAAQ;AAAA,IAClB;AAAA,IACA;AAAA,IACA,WAAW;AAAA,EAAA,CACZ;AAED,QAAM,mBAAyC;AAAA,IAC7C;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EAAA;AAGF,QAAM,SAAS,MAAM,QAAQ,UAAU,gBAAgB;AAEvD,QAAM,WAAW,KAAK,IAAA,IAAQ;AAC9B,QAAM,eAAe,OAAO,QAAQ;AAEpC,gBAAc,KAAK,+BAA+B;AAAA,IAChD;AAAA,IACA,UAAU,QAAQ;AAAA,IAClB;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA,WAAW,KAAK,IAAA;AAAA,EAAI,CACrB;AAED,SAAO;AACT;AAOA,gBAAgB,sBACd,SAC4B;AAC5B,QAAM,EAAE,SAAS,MAAM,WAAW,OAAO,UAAU;AACnD,QAAM,QAAQ,QAAQ;AAEtB,QAAM,mBAAyC;AAAA,IAC7C;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EAAA;AAIF,MAAI,QAAQ,iBAAiB;AAC3B,WAAO,QAAQ,gBAAgB,gBAAgB;AAC/C;AAAA,EACF;AAGA,QAAM,SAAS,MAAM,QAAQ,UAAU,gBAAgB;AAGvD,QAAM;AAAA,IACJ,MAAM;AAAA,IACN,WAAW,OAAO;AAAA,IAClB,OAAO,OAAO;AAAA,IACd,WAAW,KAAK,IAAA;AAAA,IAChB,OAAO,OAAO;AAAA,IACd,SAAS,OAAO;AAAA,EAAA;AAIlB,QAAM;AAAA,IACJ,MAAM;AAAA,IACN,OAAO,OAAO;AAAA,IACd,OAAO,OAAO;AAAA,IACd,WAAW,KAAK,IAAA;AAAA,IAChB,cAAc;AAAA,IACd,OAAO,OAAO;AAAA,EAAA;AAElB;AASO,SAAS,uBAId,SAC6C;AAC7C,SAAO;AACT;"}
+{"version":3,"file":"index.js","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 type { SummarizeAdapter } from './adapter'\nimport type {\n  StreamChunk,\n  SummarizationOptions,\n  SummarizationResult,\n} 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   * 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\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 unknown as SummarizeActivityOptions<\n        SummarizeAdapter<string, object>,\n        true\n      >,\n    ) as SummarizeActivityResult<TStream>\n  }\n\n  return runSummarize(\n    options as unknown 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 } = options\n  const model = adapter.model\n  const requestId = createId('summarize')\n  const inputLength = text.length\n  const startTime = Date.now()\n\n  aiEventClient.emit('summarize:request:started', {\n    requestId,\n    provider: adapter.name,\n    model,\n    inputLength,\n    timestamp: startTime,\n  })\n\n  const summarizeOptions: SummarizationOptions = {\n    model,\n    text,\n    maxLength,\n    style,\n    focus,\n  }\n\n  const result = await adapter.summarize(summarizeOptions)\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  return result\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 { adapter, text, maxLength, style, focus } = options\n  const model = adapter.model\n\n  const summarizeOptions: SummarizationOptions = {\n    model,\n    text,\n    maxLength,\n    style,\n    focus,\n  }\n\n  // Use real streaming if the adapter supports it\n  if (adapter.summarizeStream) {\n    yield* adapter.summarizeStream(summarizeOptions)\n    return\n  }\n\n  // Fall back to non-streaming — wrap result with streamGenerationResult\n  yield* streamGenerationResult(() => adapter.summarize(summarizeOptions))\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'\n"],"names":[],"mappings":";;AAqBO,MAAM,OAAO;AAmEpB,SAAS,SAAS,QAAwB;AACxC,SAAO,GAAG,MAAM,IAAI,KAAK,IAAA,CAAK,IAAI,KAAK,OAAA,EAAS,SAAS,EAAE,EAAE,MAAM,GAAG,CAAC,CAAC;AAC1E;AAwDO,SAAS,UAId,SACkC;AAClC,QAAM,EAAE,WAAW;AAEnB,MAAI,QAAQ;AACV,WAAO;AAAA,MACL;AAAA,IAAA;AAAA,EAKJ;AAEA,SAAO;AAAA,IACL;AAAA,EAAA;AAKJ;AAKA,eAAe,aACb,SAC8B;AAC9B,QAAM,EAAE,SAAS,MAAM,WAAW,OAAO,UAAU;AACnD,QAAM,QAAQ,QAAQ;AACtB,QAAM,YAAY,SAAS,WAAW;AACtC,QAAM,cAAc,KAAK;AACzB,QAAM,YAAY,KAAK,IAAA;AAEvB,gBAAc,KAAK,6BAA6B;AAAA,IAC9C;AAAA,IACA,UAAU,QAAQ;AAAA,IAClB;AAAA,IACA;AAAA,IACA,WAAW;AAAA,EAAA,CACZ;AAED,QAAM,mBAAyC;AAAA,IAC7C;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EAAA;AAGF,QAAM,SAAS,MAAM,QAAQ,UAAU,gBAAgB;AAEvD,QAAM,WAAW,KAAK,IAAA,IAAQ;AAC9B,QAAM,eAAe,OAAO,QAAQ;AAEpC,gBAAc,KAAK,+BAA+B;AAAA,IAChD;AAAA,IACA,UAAU,QAAQ;AAAA,IAClB;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA,WAAW,KAAK,IAAA;AAAA,EAAI,CACrB;AAED,SAAO;AACT;AAOA,gBAAgB,sBACd,SAC4B;AAC5B,QAAM,EAAE,SAAS,MAAM,WAAW,OAAO,UAAU;AACnD,QAAM,QAAQ,QAAQ;AAEtB,QAAM,mBAAyC;AAAA,IAC7C;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EAAA;AAIF,MAAI,QAAQ,iBAAiB;AAC3B,WAAO,QAAQ,gBAAgB,gBAAgB;AAC/C;AAAA,EACF;AAGA,SAAO,uBAAuB,MAAM,QAAQ,UAAU,gBAAgB,CAAC;AACzE;AASO,SAAS,uBAId,SAC6C;AAC7C,SAAO;AACT;"}

package/dist/esm/index.d.ts

@@ -11,9 +11,11 @@ export { convertSchemaToJsonSchema } from './activities/chat/tools/schema-conver
 export { streamToText, toServerSentEventsStream, toServerSentEventsResponse, toHttpStream, toHttpResponse, } from './stream-to-response.js';
 export { ToolCallManager } from './activities/chat/tools/tool-calls.js';
 export { maxIterations, untilFinishReason, combineStrategies, } from './activities/chat/agent-loop-strategies.js';
+export type { ChatMiddleware, ChatMiddlewareContext, ChatMiddlewarePhase, ChatMiddlewareConfig, ToolCallHookContext, BeforeToolCallDecision, AfterToolCallInfo, IterationInfo, ToolPhaseCompleteInfo, UsageInfo, FinishInfo, AbortInfo, ErrorInfo, } from './activities/chat/middleware/index.js';
 export * from './types.js';
 export { detectImageMimeType } from './utils.js';
-export * from './event-client.js';
+export { realtimeToken } from './realtime/index.js';
+export type { RealtimeToken, RealtimeTokenAdapter, RealtimeTokenOptions, RealtimeSessionConfig, VADConfig, RealtimeMessage, RealtimeMessagePart, RealtimeTextPart, RealtimeAudioPart, RealtimeToolCallPart, RealtimeToolResultPart, RealtimeImagePart, RealtimeStatus, RealtimeMode, AudioVisualization, RealtimeEvent, RealtimeEventPayloads, RealtimeEventHandler, RealtimeErrorCode, RealtimeError, } from './realtime/index.js';
 export { convertMessagesToModelMessages, generateMessageId, uiMessageToModelMessages, modelMessageToUIMessage, modelMessagesToUIMessages, normalizeToUIMessage, } from './activities/chat/messages.js';
 export { StreamProcessor, createReplayStream, ImmediateStrategy, PunctuationStrategy, BatchStrategy, WordBoundaryStrategy, CompositeStrategy, PartialJSONParser, defaultJSONParser, parsePartialJSON, } from './activities/chat/stream/index.js';
 export type { ChunkStrategy, ChunkRecording, InternalToolCallState, ProcessorResult, ProcessorState, StreamProcessorEvents, StreamProcessorOptions, ToolCallState, ToolResultState, JSONParser, } from './activities/chat/stream/index.js';

package/dist/esm/index.js

@@ -10,7 +10,7 @@ import { streamToText, toHttpResponse, toHttpStream, toServerSentEventsResponse,
 import { ToolCallManager } from "./activities/chat/tools/tool-calls.js";
 import { combineStrategies, maxIterations, untilFinishReason } from "./activities/chat/agent-loop-strategies.js";
 import { detectImageMimeType } from "./utils.js";
-import { aiEventClient } from "./event-client.js";
+import { realtimeToken } from "./realtime/index.js";
 import { convertMessagesToModelMessages, generateMessageId, modelMessageToUIMessage, modelMessagesToUIMessages, normalizeToUIMessage, uiMessageToModelMessages } from "./activities/chat/messages.js";
 import { createModel, extendAdapter } from "./extend-adapter.js";
 import { StreamProcessor, createReplayStream } from "./activities/chat/stream/processor.js";
@@ -25,7 +25,6 @@ export {
   StreamProcessor,
   ToolCallManager,
   WordBoundaryStrategy,
-  aiEventClient,
   chat,
   combineStrategies,
   convertMessagesToModelMessages,
@@ -52,6 +51,7 @@ export {
   modelMessagesToUIMessages,
   normalizeToUIMessage,
   parsePartialJSON,
+  realtimeToken,
   streamToText,
   summarize,
   toHttpResponse,

package/dist/esm/middlewares/content-guard.d.ts

@@ -0,0 +1,77 @@
+import { ChatMiddleware } from '../activities/chat/middleware/types.js';
+/**
+ * A content guard rule — either a regex pattern with replacement, or a transform function.
+ */
+export type ContentGuardRule = {
+    pattern: RegExp;
+    replacement: string;
+} | {
+    fn: (text: string) => string;
+};
+/**
+ * Information passed to the onFiltered callback.
+ */
+export interface ContentFilteredInfo {
+    /** The message ID being filtered */
+    messageId: string;
+    /** The original text before filtering */
+    original: string;
+    /** The filtered text after rules applied */
+    filtered: string;
+    /** Which strategy was used */
+    strategy: 'delta' | 'buffered';
+}
+/**
+ * Options for the content guard middleware.
+ */
+export interface ContentGuardMiddlewareOptions {
+    /**
+     * Rules to apply to text content. Each rule is either a regex pattern
+     * with a replacement string, or a custom transform function.
+     * Rules are applied in order. Each rule receives the output of the previous.
+     */
+    rules: Array<ContentGuardRule>;
+    /**
+     * Matching strategy:
+     * - 'delta': Apply rules to each delta as it arrives. Fast, real-time,
+     *   but patterns spanning chunk boundaries may be missed.
+     * - 'buffered': Accumulate content and apply rules to settled portions,
+     *   holding back a look-behind buffer to catch cross-boundary patterns.
+     *
+     * @default 'buffered'
+     */
+    strategy?: 'delta' | 'buffered';
+    /**
+     * Number of characters to hold back before emitting (buffered strategy only).
+     * Should be at least as long as the longest pattern you expect to match.
+     * Buffer is flushed when the stream ends.
+     *
+     * @default 50
+     */
+    bufferSize?: number;
+    /**
+     * If true, drop the entire chunk when any rule changes the content.
+     * @default false
+     */
+    blockOnMatch?: boolean;
+    /**
+     * Callback when content is filtered by any rule.
+     */
+    onFiltered?: (info: ContentFilteredInfo) => void;
+}
+/**
+ * Creates a middleware that filters or transforms streamed text content.
+ *
+ * @example
+ * ```ts
+ * import { contentGuardMiddleware } from '@tanstack/ai/middlewares'
+ *
+ * const guard = contentGuardMiddleware({
+ *   rules: [
+ *     { pattern: /\b\d{3}-\d{2}-\d{4}\b/g, replacement: '[SSN REDACTED]' },
+ *   ],
+ *   strategy: 'buffered',
+ * })
+ * ```
+ */
+export declare function contentGuardMiddleware(options: ContentGuardMiddlewareOptions): ChatMiddleware;

package/dist/esm/middlewares/content-guard.js

@@ -0,0 +1,155 @@
+function applyRules(text, rules) {
+  let result = text;
+  for (const rule of rules) {
+    if ("pattern" in rule) {
+      result = result.replace(rule.pattern, rule.replacement);
+    } else {
+      result = rule.fn(result);
+    }
+  }
+  return result;
+}
+function contentGuardMiddleware(options) {
+  const {
+    rules,
+    strategy = "buffered",
+    bufferSize = 50,
+    blockOnMatch = false,
+    onFiltered
+  } = options;
+  if (strategy === "delta") {
+    return createDeltaStrategy(rules, blockOnMatch, onFiltered);
+  }
+  return createBufferedStrategy(rules, bufferSize, blockOnMatch, onFiltered);
+}
+function createDeltaStrategy(rules, blockOnMatch, onFiltered) {
+  return {
+    name: "content-guard",
+    onChunk(_ctx, chunk) {
+      if (chunk.type !== "TEXT_MESSAGE_CONTENT") return;
+      const original = chunk.delta;
+      const filtered = applyRules(original, rules);
+      if (filtered === original) return;
+      if (onFiltered) {
+        onFiltered({
+          messageId: chunk.messageId,
+          original,
+          filtered,
+          strategy: "delta"
+        });
+      }
+      if (blockOnMatch) return null;
+      return {
+        ...chunk,
+        delta: filtered,
+        content: void 0
+      };
+    }
+  };
+}
+function createBufferedStrategy(rules, bufferSize, blockOnMatch, onFiltered) {
+  let rawAccumulated = "";
+  let emittedFilteredLength = 0;
+  let lastMessageId = "";
+  function resetState() {
+    rawAccumulated = "";
+    emittedFilteredLength = 0;
+    lastMessageId = "";
+  }
+  function flushBuffer() {
+    if (rawAccumulated.length === 0) return null;
+    const filtered = applyRules(rawAccumulated, rules);
+    if (blockOnMatch && filtered !== rawAccumulated) {
+      if (onFiltered) {
+        onFiltered({
+          messageId: lastMessageId,
+          original: rawAccumulated,
+          filtered,
+          strategy: "buffered"
+        });
+      }
+      resetState();
+      return null;
+    }
+    const remaining = filtered.slice(emittedFilteredLength);
+    if (remaining.length > 0) {
+      if (filtered !== rawAccumulated && onFiltered) {
+        onFiltered({
+          messageId: lastMessageId,
+          original: rawAccumulated,
+          filtered,
+          strategy: "buffered"
+        });
+      }
+      const flushed = {
+        type: "TEXT_MESSAGE_CONTENT",
+        messageId: lastMessageId,
+        delta: remaining,
+        content: filtered,
+        timestamp: Date.now()
+      };
+      resetState();
+      return flushed;
+    }
+    resetState();
+    return null;
+  }
+  return {
+    name: "content-guard",
+    onStart() {
+      resetState();
+    },
+    onChunk(_ctx, chunk) {
+      if (chunk.type === "TEXT_MESSAGE_END" || chunk.type === "RUN_FINISHED") {
+        const flushed = flushBuffer();
+        if (flushed) return [flushed, chunk];
+        return;
+      }
+      if (chunk.type !== "TEXT_MESSAGE_CONTENT") return;
+      const pending = [];
+      if (lastMessageId && chunk.messageId !== lastMessageId) {
+        const flushed = flushBuffer();
+        if (flushed) pending.push(flushed);
+      }
+      rawAccumulated += chunk.delta;
+      lastMessageId = chunk.messageId;
+      const filtered = applyRules(rawAccumulated, rules);
+      const safeFilteredEnd = Math.max(0, filtered.length - bufferSize);
+      if (safeFilteredEnd <= emittedFilteredLength) {
+        return pending.length > 0 ? pending : null;
+      }
+      if (blockOnMatch && filtered !== rawAccumulated) {
+        if (onFiltered) {
+          onFiltered({
+            messageId: chunk.messageId,
+            original: rawAccumulated,
+            filtered,
+            strategy: "buffered"
+          });
+        }
+        return pending.length > 0 ? pending : null;
+      }
+      const newDelta = filtered.slice(emittedFilteredLength, safeFilteredEnd);
+      if (filtered !== rawAccumulated && onFiltered) {
+        onFiltered({
+          messageId: chunk.messageId,
+          original: rawAccumulated,
+          filtered,
+          strategy: "buffered"
+        });
+      }
+      emittedFilteredLength = safeFilteredEnd;
+      const emitChunk = {
+        ...chunk,
+        delta: newDelta,
+        content: filtered.slice(0, safeFilteredEnd)
+      };
+      pending.push(emitChunk);
+      return pending.length === 1 ? pending[0] : pending;
+    }
+  };
+}
+export {
+  contentGuardMiddleware
+};
+//# sourceMappingURL=content-guard.js.map

package/dist/esm/middlewares/content-guard.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"content-guard.js","sources":["../../../src/middlewares/content-guard.ts"],"sourcesContent":["import type {\n  ChatMiddleware,\n  ChatMiddlewareContext,\n} from '../activities/chat/middleware/types'\nimport type { StreamChunk } from '../types'\n\n/**\n * A content guard rule — either a regex pattern with replacement, or a transform function.\n */\nexport type ContentGuardRule =\n  | { pattern: RegExp; replacement: string }\n  | { fn: (text: string) => string }\n\n/**\n * Information passed to the onFiltered callback.\n */\nexport interface ContentFilteredInfo {\n  /** The message ID being filtered */\n  messageId: string\n  /** The original text before filtering */\n  original: string\n  /** The filtered text after rules applied */\n  filtered: string\n  /** Which strategy was used */\n  strategy: 'delta' | 'buffered'\n}\n\n/**\n * Options for the content guard middleware.\n */\nexport interface ContentGuardMiddlewareOptions {\n  /**\n   * Rules to apply to text content. Each rule is either a regex pattern\n   * with a replacement string, or a custom transform function.\n   * Rules are applied in order. Each rule receives the output of the previous.\n   */\n  rules: Array<ContentGuardRule>\n\n  /**\n   * Matching strategy:\n   * - 'delta': Apply rules to each delta as it arrives. Fast, real-time,\n   *   but patterns spanning chunk boundaries may be missed.\n   * - 'buffered': Accumulate content and apply rules to settled portions,\n   *   holding back a look-behind buffer to catch cross-boundary patterns.\n   *\n   * @default 'buffered'\n   */\n  strategy?: 'delta' | 'buffered'\n\n  /**\n   * Number of characters to hold back before emitting (buffered strategy only).\n   * Should be at least as long as the longest pattern you expect to match.\n   * Buffer is flushed when the stream ends.\n   *\n   * @default 50\n   */\n  bufferSize?: number\n\n  /**\n   * If true, drop the entire chunk when any rule changes the content.\n   * @default false\n   */\n  blockOnMatch?: boolean\n\n  /**\n   * Callback when content is filtered by any rule.\n   */\n  onFiltered?: (info: ContentFilteredInfo) => void\n}\n\n/**\n * Apply all rules to a string, returning the transformed result.\n */\nfunction applyRules(text: string, rules: Array<ContentGuardRule>): string {\n  let result = text\n  for (const rule of rules) {\n    if ('pattern' in rule) {\n      result = result.replace(rule.pattern, rule.replacement)\n    } else {\n      result = rule.fn(result)\n    }\n  }\n  return result\n}\n\n/**\n * Creates a middleware that filters or transforms streamed text content.\n *\n * @example\n * ```ts\n * import { contentGuardMiddleware } from '@tanstack/ai/middlewares'\n *\n * const guard = contentGuardMiddleware({\n *   rules: [\n *     { pattern: /\\b\\d{3}-\\d{2}-\\d{4}\\b/g, replacement: '[SSN REDACTED]' },\n *   ],\n *   strategy: 'buffered',\n * })\n * ```\n */\nexport function contentGuardMiddleware(\n  options: ContentGuardMiddlewareOptions,\n): ChatMiddleware {\n  const {\n    rules,\n    strategy = 'buffered',\n    bufferSize = 50,\n    blockOnMatch = false,\n    onFiltered,\n  } = options\n\n  if (strategy === 'delta') {\n    return createDeltaStrategy(rules, blockOnMatch, onFiltered)\n  }\n  return createBufferedStrategy(rules, bufferSize, blockOnMatch, onFiltered)\n}\n\nfunction createDeltaStrategy(\n  rules: Array<ContentGuardRule>,\n  blockOnMatch: boolean,\n  onFiltered?: (info: ContentFilteredInfo) => void,\n): ChatMiddleware {\n  return {\n    name: 'content-guard',\n\n    onChunk(_ctx: ChatMiddlewareContext, chunk: StreamChunk) {\n      if (chunk.type !== 'TEXT_MESSAGE_CONTENT') return\n\n      const original = chunk.delta\n      const filtered = applyRules(original, rules)\n\n      if (filtered === original) return // unchanged, pass through\n\n      if (onFiltered) {\n        onFiltered({\n          messageId: chunk.messageId,\n          original,\n          filtered,\n          strategy: 'delta',\n        })\n      }\n\n      if (blockOnMatch) return null // drop chunk\n\n      return {\n        ...chunk,\n        delta: filtered,\n        content: undefined,\n      } as StreamChunk\n    },\n  }\n}\n\nfunction createBufferedStrategy(\n  rules: Array<ContentGuardRule>,\n  bufferSize: number,\n  blockOnMatch: boolean,\n  onFiltered?: (info: ContentFilteredInfo) => void,\n): ChatMiddleware {\n  let rawAccumulated = ''\n  let emittedFilteredLength = 0\n  let lastMessageId = ''\n\n  function resetState() {\n    rawAccumulated = ''\n    emittedFilteredLength = 0\n    lastMessageId = ''\n  }\n\n  function flushBuffer(): StreamChunk | null {\n    if (rawAccumulated.length === 0) return null\n\n    const filtered = applyRules(rawAccumulated, rules)\n\n    if (blockOnMatch && filtered !== rawAccumulated) {\n      if (onFiltered) {\n        onFiltered({\n          messageId: lastMessageId,\n          original: rawAccumulated,\n          filtered,\n          strategy: 'buffered',\n        })\n      }\n      resetState()\n      return null\n    }\n\n    const remaining = filtered.slice(emittedFilteredLength)\n    if (remaining.length > 0) {\n      if (filtered !== rawAccumulated && onFiltered) {\n        onFiltered({\n          messageId: lastMessageId,\n          original: rawAccumulated,\n          filtered,\n          strategy: 'buffered',\n        })\n      }\n\n      const flushed = {\n        type: 'TEXT_MESSAGE_CONTENT',\n        messageId: lastMessageId,\n        delta: remaining,\n        content: filtered,\n        timestamp: Date.now(),\n      } as StreamChunk\n\n      resetState()\n      return flushed\n    }\n\n    resetState()\n    return null\n  }\n\n  return {\n    name: 'content-guard',\n\n    onStart() {\n      resetState()\n    },\n\n    onChunk(_ctx: ChatMiddlewareContext, chunk: StreamChunk) {\n      // Flush buffer on stream end events\n      if (chunk.type === 'TEXT_MESSAGE_END' || chunk.type === 'RUN_FINISHED') {\n        const flushed = flushBuffer()\n        if (flushed) return [flushed, chunk]\n        return // pass through end event\n      }\n\n      if (chunk.type !== 'TEXT_MESSAGE_CONTENT') return // pass through\n\n      // Flush buffer on message boundary change\n      const pending: Array<StreamChunk> = []\n      if (lastMessageId && chunk.messageId !== lastMessageId) {\n        const flushed = flushBuffer()\n        if (flushed) pending.push(flushed)\n      }\n\n      rawAccumulated += chunk.delta\n      lastMessageId = chunk.messageId\n\n      // Apply rules to full accumulated text, buffer in filtered space\n      const filtered = applyRules(rawAccumulated, rules)\n      const safeFilteredEnd = Math.max(0, filtered.length - bufferSize)\n\n      if (safeFilteredEnd <= emittedFilteredLength) {\n        return pending.length > 0 ? pending : null\n      }\n\n      if (blockOnMatch && filtered !== rawAccumulated) {\n        if (onFiltered) {\n          onFiltered({\n            messageId: chunk.messageId,\n            original: rawAccumulated,\n            filtered,\n            strategy: 'buffered',\n          })\n        }\n        return pending.length > 0 ? pending : null\n      }\n\n      const newDelta = filtered.slice(emittedFilteredLength, safeFilteredEnd)\n\n      if (filtered !== rawAccumulated && onFiltered) {\n        onFiltered({\n          messageId: chunk.messageId,\n          original: rawAccumulated,\n          filtered,\n          strategy: 'buffered',\n        })\n      }\n\n      emittedFilteredLength = safeFilteredEnd\n\n      const emitChunk = {\n        ...chunk,\n        delta: newDelta,\n        content: filtered.slice(0, safeFilteredEnd),\n      } as StreamChunk\n\n      pending.push(emitChunk)\n      return pending.length === 1 ? pending[0]! : pending\n    },\n  }\n}\n"],"names":[],"mappings":"AAyEA,SAAS,WAAW,MAAc,OAAwC;AACxE,MAAI,SAAS;AACb,aAAW,QAAQ,OAAO;AACxB,QAAI,aAAa,MAAM;AACrB,eAAS,OAAO,QAAQ,KAAK,SAAS,KAAK,WAAW;AAAA,IACxD,OAAO;AACL,eAAS,KAAK,GAAG,MAAM;AAAA,IACzB;AAAA,EACF;AACA,SAAO;AACT;AAiBO,SAAS,uBACd,SACgB;AAChB,QAAM;AAAA,IACJ;AAAA,IACA,WAAW;AAAA,IACX,aAAa;AAAA,IACb,eAAe;AAAA,IACf;AAAA,EAAA,IACE;AAEJ,MAAI,aAAa,SAAS;AACxB,WAAO,oBAAoB,OAAO,cAAc,UAAU;AAAA,EAC5D;AACA,SAAO,uBAAuB,OAAO,YAAY,cAAc,UAAU;AAC3E;AAEA,SAAS,oBACP,OACA,cACA,YACgB;AAChB,SAAO;AAAA,IACL,MAAM;AAAA,IAEN,QAAQ,MAA6B,OAAoB;AACvD,UAAI,MAAM,SAAS,uBAAwB;AAE3C,YAAM,WAAW,MAAM;AACvB,YAAM,WAAW,WAAW,UAAU,KAAK;AAE3C,UAAI,aAAa,SAAU;AAE3B,UAAI,YAAY;AACd,mBAAW;AAAA,UACT,WAAW,MAAM;AAAA,UACjB;AAAA,UACA;AAAA,UACA,UAAU;AAAA,QAAA,CACX;AAAA,MACH;AAEA,UAAI,aAAc,QAAO;AAEzB,aAAO;AAAA,QACL,GAAG;AAAA,QACH,OAAO;AAAA,QACP,SAAS;AAAA,MAAA;AAAA,IAEb;AAAA,EAAA;AAEJ;AAEA,SAAS,uBACP,OACA,YACA,cACA,YACgB;AAChB,MAAI,iBAAiB;AACrB,MAAI,wBAAwB;AAC5B,MAAI,gBAAgB;AAEpB,WAAS,aAAa;AACpB,qBAAiB;AACjB,4BAAwB;AACxB,oBAAgB;AAAA,EAClB;AAEA,WAAS,cAAkC;AACzC,QAAI,eAAe,WAAW,EAAG,QAAO;AAExC,UAAM,WAAW,WAAW,gBAAgB,KAAK;AAEjD,QAAI,gBAAgB,aAAa,gBAAgB;AAC/C,UAAI,YAAY;AACd,mBAAW;AAAA,UACT,WAAW;AAAA,UACX,UAAU;AAAA,UACV;AAAA,UACA,UAAU;AAAA,QAAA,CACX;AAAA,MACH;AACA,iBAAA;AACA,aAAO;AAAA,IACT;AAEA,UAAM,YAAY,SAAS,MAAM,qBAAqB;AACtD,QAAI,UAAU,SAAS,GAAG;AACxB,UAAI,aAAa,kBAAkB,YAAY;AAC7C,mBAAW;AAAA,UACT,WAAW;AAAA,UACX,UAAU;AAAA,UACV;AAAA,UACA,UAAU;AAAA,QAAA,CACX;AAAA,MACH;AAEA,YAAM,UAAU;AAAA,QACd,MAAM;AAAA,QACN,WAAW;AAAA,QACX,OAAO;AAAA,QACP,SAAS;AAAA,QACT,WAAW,KAAK,IAAA;AAAA,MAAI;AAGtB,iBAAA;AACA,aAAO;AAAA,IACT;AAEA,eAAA;AACA,WAAO;AAAA,EACT;AAEA,SAAO;AAAA,IACL,MAAM;AAAA,IAEN,UAAU;AACR,iBAAA;AAAA,IACF;AAAA,IAEA,QAAQ,MAA6B,OAAoB;AAEvD,UAAI,MAAM,SAAS,sBAAsB,MAAM,SAAS,gBAAgB;AACtE,cAAM,UAAU,YAAA;AAChB,YAAI,QAAS,QAAO,CAAC,SAAS,KAAK;AACnC;AAAA,MACF;AAEA,UAAI,MAAM,SAAS,uBAAwB;AAG3C,YAAM,UAA8B,CAAA;AACpC,UAAI,iBAAiB,MAAM,cAAc,eAAe;AACtD,cAAM,UAAU,YAAA;AAChB,YAAI,QAAS,SAAQ,KAAK,OAAO;AAAA,MACnC;AAEA,wBAAkB,MAAM;AACxB,sBAAgB,MAAM;AAGtB,YAAM,WAAW,WAAW,gBAAgB,KAAK;AACjD,YAAM,kBAAkB,KAAK,IAAI,GAAG,SAAS,SAAS,UAAU;AAEhE,UAAI,mBAAmB,uBAAuB;AAC5C,eAAO,QAAQ,SAAS,IAAI,UAAU;AAAA,MACxC;AAEA,UAAI,gBAAgB,aAAa,gBAAgB;AAC/C,YAAI,YAAY;AACd,qBAAW;AAAA,YACT,WAAW,MAAM;AAAA,YACjB,UAAU;AAAA,YACV;AAAA,YACA,UAAU;AAAA,UAAA,CACX;AAAA,QACH;AACA,eAAO,QAAQ,SAAS,IAAI,UAAU;AAAA,MACxC;AAEA,YAAM,WAAW,SAAS,MAAM,uBAAuB,eAAe;AAEtE,UAAI,aAAa,kBAAkB,YAAY;AAC7C,mBAAW;AAAA,UACT,WAAW,MAAM;AAAA,UACjB,UAAU;AAAA,UACV;AAAA,UACA,UAAU;AAAA,QAAA,CACX;AAAA,MACH;AAEA,8BAAwB;AAExB,YAAM,YAAY;AAAA,QAChB,GAAG;AAAA,QACH,OAAO;AAAA,QACP,SAAS,SAAS,MAAM,GAAG,eAAe;AAAA,MAAA;AAG5C,cAAQ,KAAK,SAAS;AACtB,aAAO,QAAQ,WAAW,IAAI,QAAQ,CAAC,IAAK;AAAA,IAC9C;AAAA,EAAA;AAEJ;"}

package/dist/esm/middlewares/index.d.ts

@@ -0,0 +1,2 @@
+export { toolCacheMiddleware, type ToolCacheMiddlewareOptions, type ToolCacheStorage, type ToolCacheEntry, } from './tool-cache.js';
+export { contentGuardMiddleware, type ContentGuardMiddlewareOptions, type ContentGuardRule, type ContentFilteredInfo, } from './content-guard.js';

package/dist/esm/middlewares/index.js

@@ -0,0 +1,7 @@
+import { contentGuardMiddleware } from "./content-guard.js";
+import { toolCacheMiddleware } from "../activities/chat/middleware/tool-cache-middleware.js";
+export {
+  contentGuardMiddleware,
+  toolCacheMiddleware
+};
+//# sourceMappingURL=index.js.map

package/dist/esm/middlewares/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sources":[],"sourcesContent":[],"names":[],"mappings":";;"}

package/dist/esm/middlewares/tool-cache.d.ts

@@ -0,0 +1 @@
+export { toolCacheMiddleware, type ToolCacheMiddlewareOptions, type ToolCacheStorage, type ToolCacheEntry, } from '../activities/chat/middleware/tool-cache-middleware.js';