@tanstack/ai 0.32.0 → 0.34.0

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 '@tanstack/ai-event-client'\nimport { toRunErrorPayload } from '../error-payload'\nimport { resolveDebugOption } from '../../logger/resolve'\nimport type { InternalLogger } from '../../logger/internal-logger'\nimport type { DebugOption } from '../../logger/types'\nimport type { VideoAdapter } from './adapter'\nimport type {\n  MediaPrompt,\n  MediaPromptFor,\n  StreamChunk,\n  TokenUsage,\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 * Extract the prompt type a model accepts from a VideoAdapter via ~types.\n * Mirrors `ImagePromptForModel`: models in the adapter's input-modality map\n * get a `prompt` narrowed to text + their supported part types; adapters\n * without a map fall back to the full MediaPrompt.\n */\nexport type VideoPromptForAdapter<TAdapter> =\n  TAdapter extends VideoAdapter<infer TModel, any, any, any, infer ModsByName>\n    ? string extends keyof ModsByName\n      ? MediaPrompt\n      : TModel extends keyof ModsByName\n        ? MediaPromptFor<ModsByName[TModel][number]>\n        : MediaPrompt\n    : MediaPrompt\n\n/**\n * Extract the duration type for a VideoAdapter's model via ~types.\n * Mirrors `VideoSizeForAdapter`. Falls back to `number` for adapters that\n * haven't declared per-model duration constraints.\n */\nexport type VideoDurationForAdapter<TAdapter> =\n  TAdapter extends VideoAdapter<\n    infer TModel,\n    any,\n    any,\n    any,\n    any,\n    infer TDurationMap\n  >\n    ? TModel extends keyof TDurationMap\n      ? TDurationMap[TModel]\n      : number\n    : number\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  /**\n   * Description of the desired video. Either a plain string, or — for models\n   * that support image-conditioned generation — an ordered array of content\n   * parts interleaving text with image inputs. Image parts may carry\n   * `metadata.role` (`'start_frame' | 'end_frame' | 'reference' |\n   * 'character'`) to disambiguate intent; positional fallback otherwise. The\n   * accepted part types are narrowed per model via the adapter's\n   * input-modality map.\n   */\n  prompt: VideoPromptForAdapter<TAdapter>\n  /** Video size — format depends on the provider (e.g., \"16:9\", \"1280x720\") */\n  size?: VideoSizeForAdapter<TAdapter>\n  /**\n   * Video duration in seconds. Adapters that declare a per-model duration\n   * map narrow this to the model's valid union (e.g. `4 | 6 | 8` for Veo 3).\n   * Pass `adapter.snapDuration(seconds)` to coerce raw seconds to a valid\n   * value.\n   */\n  duration?: VideoDurationForAdapter<TAdapter>\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  /**\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} & ({} 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  const logger: InternalLogger = resolveDebugOption(options.debug)\n  const providerName =\n    (adapter as { name?: string; provider?: string }).provider ??\n    (adapter as { name?: string }).name ??\n    'unknown'\n\n  logger.request(`activity=generateVideo provider=${providerName}`, {\n    provider: providerName,\n    model,\n  })\n\n  try {\n    const result = await adapter.createVideoJob({\n      model,\n      prompt,\n      size,\n      duration,\n      modelOptions,\n      logger,\n    })\n    logger.output(`activity=generateVideo jobId=${result.jobId}`, {\n      jobId: result.jobId,\n      model: result.model,\n    })\n    return result\n  } catch (error) {\n    logger.errors('generateVideo activity failed', {\n      error,\n      source: 'generateVideo',\n    })\n    throw error\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  const logger: InternalLogger = resolveDebugOption(options.debug)\n  const providerName =\n    (adapter as { name?: string; provider?: string }).provider ??\n    (adapter as { name?: string }).name ??\n    'unknown'\n\n  const threadId = createId('thread')\n\n  yield {\n    type: 'RUN_STARTED',\n    runId,\n    threadId,\n    timestamp: Date.now(),\n  } as StreamChunk\n\n  logger.request(\n    `activity=generateVideo provider=${providerName} stream=true`,\n    {\n      provider: providerName,\n      model,\n    },\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      logger,\n    })\n\n    yield {\n      type: 'CUSTOM',\n      name: 'video:job:created',\n      value: { jobId: jobResult.jobId },\n      timestamp: Date.now(),\n    } as StreamChunk\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      } as StreamChunk\n\n      if (statusResult.status === 'completed') {\n        const urlResult = await adapter.getVideoUrl(jobResult.jobId)\n\n        logger.output(\n          `activity=generateVideo jobId=${jobResult.jobId} status=completed`,\n          {\n            jobId: jobResult.jobId,\n            url: urlResult.url,\n          },\n        )\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            ...(urlResult.usage ? { usage: urlResult.usage } : {}),\n          },\n          timestamp: Date.now(),\n        } as StreamChunk\n\n        yield {\n          type: 'RUN_FINISHED',\n          runId,\n          threadId,\n          finishReason: 'stop',\n          timestamp: Date.now(),\n        } as StreamChunk\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: unknown) {\n    const payload = toRunErrorPayload(error, 'Video generation failed')\n    logger.errors('generateVideo activity failed', {\n      message: payload.message,\n      code: payload.code,\n      source: 'generateVideo',\n    })\n    yield {\n      type: 'RUN_ERROR',\n      runId,\n      threadId,\n      message: payload.message,\n      code: payload.code,\n      error: payload,\n      timestamp: Date.now(),\n    } as StreamChunk\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  usage?: TokenUsage\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      if (urlResult.usage) {\n        aiEventClient.emit('video:usage', {\n          requestId,\n          model: adapter.model,\n          usage: urlResult.usage,\n          timestamp: Date.now(),\n        })\n      }\n      return {\n        status: statusResult.status,\n        progress: statusResult.progress,\n        url: urlResult.url,\n        ...(urlResult.usage ? { usage: urlResult.usage } : {}),\n      }\n    } catch (error) {\n      const errorMessage =\n        error instanceof Error ? error.message : 'Failed to get video URL'\n      aiEventClient.emit('video:request:completed', {\n        requestId,\n        provider: adapter.name,\n        model: adapter.model,\n        requestType: 'status',\n        jobId,\n        status: 'failed',\n        progress: statusResult.progress,\n        error: errorMessage,\n        duration: Date.now() - startTime,\n        timestamp: Date.now(),\n      })\n      // Provider reported completed but result fetch failed — treat as failed\n      return {\n        status: 'failed' as const,\n        progress: statusResult.progress,\n        error: errorMessage,\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":";;;AA8BO,MAAM,OAAO;AA6DpB,SAAS,SAAS,QAAwB;AACxC,SAAO,GAAG,MAAM,IAAI,KAAK,IAAA,CAAK,IAAI,KAAK,OAAA,EAAS,SAAS,EAAE,EAAE,MAAM,GAAG,CAAC,CAAC;AAC1E;AA2LO,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;AACtB,QAAM,SAAyB,mBAAmB,QAAQ,KAAK;AAC/D,QAAM,eACH,QAAiD,YACjD,QAA8B,QAC/B;AAEF,SAAO,QAAQ,mCAAmC,YAAY,IAAI;AAAA,IAChE,UAAU;AAAA,IACV;AAAA,EAAA,CACD;AAED,MAAI;AACF,UAAM,SAAS,MAAM,QAAQ,eAAe;AAAA,MAC1C;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IAAA,CACD;AACD,WAAO,OAAO,gCAAgC,OAAO,KAAK,IAAI;AAAA,MAC5D,OAAO,OAAO;AAAA,MACd,OAAO,OAAO;AAAA,IAAA,CACf;AACD,WAAO;AAAA,EACT,SAAS,OAAO;AACd,WAAO,OAAO,iCAAiC;AAAA,MAC7C;AAAA,MACA,QAAQ;AAAA,IAAA,CACT;AACD,UAAM;AAAA,EACR;AACF;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;AAC3C,QAAM,SAAyB,mBAAmB,QAAQ,KAAK;AAC/D,QAAM,eACH,QAAiD,YACjD,QAA8B,QAC/B;AAEF,QAAM,WAAW,SAAS,QAAQ;AAElC,QAAM;AAAA,IACJ,MAAM;AAAA,IACN;AAAA,IACA;AAAA,IACA,WAAW,KAAK,IAAA;AAAA,EAAI;AAGtB,SAAO;AAAA,IACL,mCAAmC,YAAY;AAAA,IAC/C;AAAA,MACE,UAAU;AAAA,MACV;AAAA,IAAA;AAAA,EACF;AAGF,MAAI;AAEF,UAAM,YAAY,MAAM,QAAQ,eAAe;AAAA,MAC7C;AAAA,MACA;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,eAAO;AAAA,UACL,gCAAgC,UAAU,KAAK;AAAA,UAC/C;AAAA,YACE,OAAO,UAAU;AAAA,YACjB,KAAK,UAAU;AAAA,UAAA;AAAA,QACjB;AAGF,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,YACrB,GAAI,UAAU,QAAQ,EAAE,OAAO,UAAU,MAAA,IAAU,CAAA;AAAA,UAAC;AAAA,UAEtD,WAAW,KAAK,IAAA;AAAA,QAAI;AAGtB,cAAM;AAAA,UACJ,MAAM;AAAA,UACN;AAAA,UACA;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,OAAgB;AACvB,UAAM,UAAU,kBAAkB,OAAO,yBAAyB;AAClE,WAAO,OAAO,iCAAiC;AAAA,MAC7C,SAAS,QAAQ;AAAA,MACjB,MAAM,QAAQ;AAAA,MACd,QAAQ;AAAA,IAAA,CACT;AACD,UAAM;AAAA,MACJ,MAAM;AAAA,MACN;AAAA,MACA;AAAA,MACA,SAAS,QAAQ;AAAA,MACjB,MAAM,QAAQ;AAAA,MACd,OAAO;AAAA,MACP,WAAW,KAAK,IAAA;AAAA,IAAI;AAAA,EAExB;AACF;AA2BA,eAAsB,kBAEpB,SASC;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,UAAI,UAAU,OAAO;AACnB,sBAAc,KAAK,eAAe;AAAA,UAChC;AAAA,UACA,OAAO,QAAQ;AAAA,UACf,OAAO,UAAU;AAAA,UACjB,WAAW,KAAK,IAAA;AAAA,QAAI,CACrB;AAAA,MACH;AACA,aAAO;AAAA,QACL,QAAQ,aAAa;AAAA,QACrB,UAAU,aAAa;AAAA,QACvB,KAAK,UAAU;AAAA,QACf,GAAI,UAAU,QAAQ,EAAE,OAAO,UAAU,MAAA,IAAU,CAAA;AAAA,MAAC;AAAA,IAExD,SAAS,OAAO;AACd,YAAM,eACJ,iBAAiB,QAAQ,MAAM,UAAU;AAC3C,oBAAc,KAAK,2BAA2B;AAAA,QAC5C;AAAA,QACA,UAAU,QAAQ;AAAA,QAClB,OAAO,QAAQ;AAAA,QACf,aAAa;AAAA,QACb;AAAA,QACA,QAAQ;AAAA,QACR,UAAU,aAAa;AAAA,QACvB,OAAO;AAAA,QACP,UAAU,KAAK,IAAA,IAAQ;AAAA,QACvB,WAAW,KAAK,IAAA;AAAA,MAAI,CACrB;AAED,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,UAAU,aAAa;AAAA,QACvB,OAAO;AAAA,MAAA;AAAA,IAEX;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;"}
+{"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 { toRunErrorPayload } from '../error-payload'\nimport { resolveDebugOption } from '../../logger/resolve'\nimport {\n  createGenerationContext,\n  runGenerationAbort,\n  runGenerationError,\n  runGenerationFinish,\n  runGenerationStart,\n  runGenerationUsage,\n} from '../middleware'\nimport type { InternalLogger } from '../../logger/internal-logger'\nimport type { DebugOption } from '../../logger/types'\nimport type { GenerationMiddleware } from '../middleware'\nimport type { VideoAdapter } from './adapter'\nimport type {\n  MediaPrompt,\n  MediaPromptFor,\n  StreamChunk,\n  TokenUsage,\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 * Extract the prompt type a model accepts from a VideoAdapter via ~types.\n * Mirrors `ImagePromptForModel`: models in the adapter's input-modality map\n * get a `prompt` narrowed to text + their supported part types; adapters\n * without a map fall back to the full MediaPrompt.\n */\nexport type VideoPromptForAdapter<TAdapter> =\n  TAdapter extends VideoAdapter<infer TModel, any, any, any, infer ModsByName>\n    ? string extends keyof ModsByName\n      ? MediaPrompt\n      : TModel extends keyof ModsByName\n        ? MediaPromptFor<ModsByName[TModel][number]>\n        : MediaPrompt\n    : MediaPrompt\n\n/**\n * Extract the duration type for a VideoAdapter's model via ~types.\n * Mirrors `VideoSizeForAdapter`. Falls back to `number` for adapters that\n * haven't declared per-model duration constraints.\n */\nexport type VideoDurationForAdapter<TAdapter> =\n  TAdapter extends VideoAdapter<\n    infer TModel,\n    any,\n    any,\n    any,\n    any,\n    infer TDurationMap\n  >\n    ? TModel extends keyof TDurationMap\n      ? TDurationMap[TModel]\n      : number\n    : number\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  /**\n   * Description of the desired video. Either a plain string, or — for models\n   * that support image-conditioned generation — an ordered array of content\n   * parts interleaving text with image inputs. Image parts may carry\n   * `metadata.role` (`'start_frame' | 'end_frame' | 'reference' |\n   * 'character'`) to disambiguate intent; positional fallback otherwise. The\n   * accepted part types are narrowed per model via the adapter's\n   * input-modality map.\n   */\n  prompt: VideoPromptForAdapter<TAdapter>\n  /** Video size — format depends on the provider (e.g., \"16:9\", \"1280x720\") */\n  size?: VideoSizeForAdapter<TAdapter>\n  /**\n   * Video duration in seconds. Adapters that declare a per-model duration\n   * map narrow this to the model's valid union (e.g. `4 | 6 | 8` for Veo 3).\n   * Pass `adapter.snapDuration(seconds)` to coerce raw seconds to a valid\n   * value.\n   */\n  duration?: VideoDurationForAdapter<TAdapter>\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  /**\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   * Observe-only middleware notified on start, usage, success, and error. Pass\n   * `otelMiddleware()` to emit OpenTelemetry spans, or implement the\n   * `GenerationMiddleware` contract for a custom backend. In streaming mode the\n   * span covers the full create→poll→complete lifecycle; in non-streaming mode\n   * it covers job submission. An abandoned stream fires `onAbort`.\n   */\n  middleware?: Array<GenerationMiddleware>\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, middleware } = options\n  const model = adapter.model\n  const requestId = createId('video')\n  const startTime = Date.now()\n  const logger: InternalLogger = resolveDebugOption(options.debug)\n  const providerName =\n    (adapter as { name?: string; provider?: string }).provider ??\n    (adapter as { name?: string }).name ??\n    'unknown'\n\n  const mwCtx = createGenerationContext({\n    requestId,\n    activity: 'video',\n    provider: adapter.name,\n    model,\n    modelOptions,\n    createId,\n  })\n\n  await runGenerationStart(middleware, mwCtx)\n\n  logger.request(`activity=generateVideo provider=${providerName}`, {\n    provider: providerName,\n    model,\n  })\n\n  try {\n    const result = await adapter.createVideoJob({\n      model,\n      prompt,\n      size,\n      duration,\n      modelOptions,\n      logger,\n    })\n    logger.output(`activity=generateVideo jobId=${result.jobId}`, {\n      jobId: result.jobId,\n      model: result.model,\n    })\n    // Non-streaming create only submits the job; usage isn't known until the\n    // job completes via polling, so the span covers submission only.\n    await runGenerationFinish(middleware, mwCtx, {\n      duration: Date.now() - startTime,\n    })\n    return result\n  } catch (error) {\n    await runGenerationError(middleware, mwCtx, {\n      error,\n      duration: Date.now() - startTime,\n    })\n    logger.errors('generateVideo activity failed', {\n      error,\n      source: 'generateVideo',\n    })\n    throw error\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, middleware } = options\n  const model = adapter.model\n  const runId = options.runId ?? createId('run')\n  const requestId = createId('video')\n  const obsStartTime = Date.now()\n  const pollingInterval = options.pollingInterval ?? 2000\n  const maxDuration = options.maxDuration ?? 600_000\n  const logger: InternalLogger = resolveDebugOption(options.debug)\n  const providerName =\n    (adapter as { name?: string; provider?: string }).provider ??\n    (adapter as { name?: string }).name ??\n    'unknown'\n\n  const threadId = createId('thread')\n\n  yield {\n    type: 'RUN_STARTED',\n    runId,\n    threadId,\n    timestamp: Date.now(),\n  } as StreamChunk\n\n  const mwCtx = createGenerationContext({\n    requestId,\n    activity: 'video',\n    provider: adapter.name,\n    model,\n    modelOptions,\n    createId,\n  })\n\n  await runGenerationStart(middleware, mwCtx)\n\n  logger.request(\n    `activity=generateVideo provider=${providerName} stream=true`,\n    {\n      provider: providerName,\n      model,\n    },\n  )\n\n  // Tracks whether a terminal observer event (finish/error) has already fired,\n  // so the `finally` below can fire one on abandonment without double-firing.\n  let settled = false\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      logger,\n    })\n\n    yield {\n      type: 'CUSTOM',\n      name: 'video:job:created',\n      value: { jobId: jobResult.jobId },\n      timestamp: Date.now(),\n    } as StreamChunk\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      } as StreamChunk\n\n      if (statusResult.status === 'completed') {\n        const urlResult = await adapter.getVideoUrl(jobResult.jobId)\n\n        logger.output(\n          `activity=generateVideo jobId=${jobResult.jobId} status=completed`,\n          {\n            jobId: jobResult.jobId,\n            url: urlResult.url,\n          },\n        )\n\n        // Fire finish before yielding the terminal chunks: the generation has\n        // succeeded, so a consumer that stops reading after `generation:result`\n        // (without pulling `RUN_FINISHED`) must not trip the abandonment path in\n        // `finally`, which would otherwise report a spurious cancellation.\n        if (urlResult.usage)\n          await runGenerationUsage(middleware, mwCtx, urlResult.usage)\n        await runGenerationFinish(middleware, mwCtx, {\n          duration: Date.now() - obsStartTime,\n          usage: urlResult.usage,\n        })\n        settled = true\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            ...(urlResult.usage ? { usage: urlResult.usage } : {}),\n          },\n          timestamp: Date.now(),\n        } as StreamChunk\n\n        yield {\n          type: 'RUN_FINISHED',\n          runId,\n          threadId,\n          finishReason: 'stop',\n          timestamp: Date.now(),\n        } as StreamChunk\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: unknown) {\n    const payload = toRunErrorPayload(error, 'Video generation failed')\n    // Mark settled before firing onError: if a user error-hook throws, the\n    // `finally` below must still not double-fire onAbort over the same op\n    // (which would mask the original error and end the span twice).\n    settled = true\n    await runGenerationError(middleware, mwCtx, {\n      error,\n      duration: Date.now() - obsStartTime,\n    })\n    logger.errors('generateVideo activity failed', {\n      message: payload.message,\n      code: payload.code,\n      source: 'generateVideo',\n    })\n    yield {\n      type: 'RUN_ERROR',\n      runId,\n      threadId,\n      message: payload.message,\n      code: payload.code,\n      error: payload,\n      timestamp: Date.now(),\n    } as StreamChunk\n  } finally {\n    if (!settled) {\n      // The consumer abandoned the stream (broke the `for await` loop or\n      // disconnected) before completion, so the generator is being unwound at\n      // a `yield` without reaching finish/error. Fire `onAbort` — a cancel, not\n      // an error — so otelMiddleware ends its span instead of leaking it.\n      await runGenerationAbort(middleware, mwCtx, {\n        reason: 'Video generation stream abandoned before completion',\n        duration: Date.now() - obsStartTime,\n      })\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  usage?: TokenUsage\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      if (urlResult.usage) {\n        aiEventClient.emit('video:usage', {\n          requestId,\n          model: adapter.model,\n          usage: urlResult.usage,\n          timestamp: Date.now(),\n        })\n      }\n      return {\n        status: statusResult.status,\n        progress: statusResult.progress,\n        url: urlResult.url,\n        ...(urlResult.usage ? { usage: urlResult.usage } : {}),\n      }\n    } catch (error) {\n      const errorMessage =\n        error instanceof Error ? error.message : 'Failed to get video URL'\n      aiEventClient.emit('video:request:completed', {\n        requestId,\n        provider: adapter.name,\n        model: adapter.model,\n        requestType: 'status',\n        jobId,\n        status: 'failed',\n        progress: statusResult.progress,\n        error: errorMessage,\n        duration: Date.now() - startTime,\n        timestamp: Date.now(),\n      })\n      // Provider reported completed but result fetch failed — treat as failed\n      return {\n        status: 'failed' as const,\n        progress: statusResult.progress,\n        error: errorMessage,\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":";;;;AAuCO,MAAM,OAAO;AA6DpB,SAAS,SAAS,QAAwB;AACxC,SAAO,GAAG,MAAM,IAAI,KAAK,IAAA,CAAK,IAAI,KAAK,OAAA,EAAS,SAAS,EAAE,EAAE,MAAM,GAAG,CAAC,CAAC;AAC1E;AAmMO,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,cAAc,eAAe;AACtE,QAAM,QAAQ,QAAQ;AACtB,QAAM,YAAY,SAAS,OAAO;AAClC,QAAM,YAAY,KAAK,IAAA;AACvB,QAAM,SAAyB,mBAAmB,QAAQ,KAAK;AAC/D,QAAM,eACH,QAAiD,YACjD,QAA8B,QAC/B;AAEF,QAAM,QAAQ,wBAAwB;AAAA,IACpC;AAAA,IACA,UAAU;AAAA,IACV,UAAU,QAAQ;AAAA,IAClB;AAAA,IACA;AAAA,IACA;AAAA,EAAA,CACD;AAED,QAAM,mBAAmB,YAAY,KAAK;AAE1C,SAAO,QAAQ,mCAAmC,YAAY,IAAI;AAAA,IAChE,UAAU;AAAA,IACV;AAAA,EAAA,CACD;AAED,MAAI;AACF,UAAM,SAAS,MAAM,QAAQ,eAAe;AAAA,MAC1C;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IAAA,CACD;AACD,WAAO,OAAO,gCAAgC,OAAO,KAAK,IAAI;AAAA,MAC5D,OAAO,OAAO;AAAA,MACd,OAAO,OAAO;AAAA,IAAA,CACf;AAGD,UAAM,oBAAoB,YAAY,OAAO;AAAA,MAC3C,UAAU,KAAK,QAAQ;AAAA,IAAA,CACxB;AACD,WAAO;AAAA,EACT,SAAS,OAAO;AACd,UAAM,mBAAmB,YAAY,OAAO;AAAA,MAC1C;AAAA,MACA,UAAU,KAAK,QAAQ;AAAA,IAAA,CACxB;AACD,WAAO,OAAO,iCAAiC;AAAA,MAC7C;AAAA,MACA,QAAQ;AAAA,IAAA,CACT;AACD,UAAM;AAAA,EACR;AACF;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,cAAc,eAAe;AACtE,QAAM,QAAQ,QAAQ;AACtB,QAAM,QAAQ,QAAQ,SAAS,SAAS,KAAK;AAC7C,QAAM,YAAY,SAAS,OAAO;AAClC,QAAM,eAAe,KAAK,IAAA;AAC1B,QAAM,kBAAkB,QAAQ,mBAAmB;AACnD,QAAM,cAAc,QAAQ,eAAe;AAC3C,QAAM,SAAyB,mBAAmB,QAAQ,KAAK;AAC/D,QAAM,eACH,QAAiD,YACjD,QAA8B,QAC/B;AAEF,QAAM,WAAW,SAAS,QAAQ;AAElC,QAAM;AAAA,IACJ,MAAM;AAAA,IACN;AAAA,IACA;AAAA,IACA,WAAW,KAAK,IAAA;AAAA,EAAI;AAGtB,QAAM,QAAQ,wBAAwB;AAAA,IACpC;AAAA,IACA,UAAU;AAAA,IACV,UAAU,QAAQ;AAAA,IAClB;AAAA,IACA;AAAA,IACA;AAAA,EAAA,CACD;AAED,QAAM,mBAAmB,YAAY,KAAK;AAE1C,SAAO;AAAA,IACL,mCAAmC,YAAY;AAAA,IAC/C;AAAA,MACE,UAAU;AAAA,MACV;AAAA,IAAA;AAAA,EACF;AAKF,MAAI,UAAU;AACd,MAAI;AAEF,UAAM,YAAY,MAAM,QAAQ,eAAe;AAAA,MAC7C;AAAA,MACA;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,eAAO;AAAA,UACL,gCAAgC,UAAU,KAAK;AAAA,UAC/C;AAAA,YACE,OAAO,UAAU;AAAA,YACjB,KAAK,UAAU;AAAA,UAAA;AAAA,QACjB;AAOF,YAAI,UAAU;AACZ,gBAAM,mBAAmB,YAAY,OAAO,UAAU,KAAK;AAC7D,cAAM,oBAAoB,YAAY,OAAO;AAAA,UAC3C,UAAU,KAAK,IAAA,IAAQ;AAAA,UACvB,OAAO,UAAU;AAAA,QAAA,CAClB;AACD,kBAAU;AAEV,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,YACrB,GAAI,UAAU,QAAQ,EAAE,OAAO,UAAU,MAAA,IAAU,CAAA;AAAA,UAAC;AAAA,UAEtD,WAAW,KAAK,IAAA;AAAA,QAAI;AAGtB,cAAM;AAAA,UACJ,MAAM;AAAA,UACN;AAAA,UACA;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,OAAgB;AACvB,UAAM,UAAU,kBAAkB,OAAO,yBAAyB;AAIlE,cAAU;AACV,UAAM,mBAAmB,YAAY,OAAO;AAAA,MAC1C;AAAA,MACA,UAAU,KAAK,QAAQ;AAAA,IAAA,CACxB;AACD,WAAO,OAAO,iCAAiC;AAAA,MAC7C,SAAS,QAAQ;AAAA,MACjB,MAAM,QAAQ;AAAA,MACd,QAAQ;AAAA,IAAA,CACT;AACD,UAAM;AAAA,MACJ,MAAM;AAAA,MACN;AAAA,MACA;AAAA,MACA,SAAS,QAAQ;AAAA,MACjB,MAAM,QAAQ;AAAA,MACd,OAAO;AAAA,MACP,WAAW,KAAK,IAAA;AAAA,IAAI;AAAA,EAExB,UAAA;AACE,QAAI,CAAC,SAAS;AAKZ,YAAM,mBAAmB,YAAY,OAAO;AAAA,QAC1C,QAAQ;AAAA,QACR,UAAU,KAAK,QAAQ;AAAA,MAAA,CACxB;AAAA,IACH;AAAA,EACF;AACF;AA2BA,eAAsB,kBAEpB,SASC;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,UAAI,UAAU,OAAO;AACnB,sBAAc,KAAK,eAAe;AAAA,UAChC;AAAA,UACA,OAAO,QAAQ;AAAA,UACf,OAAO,UAAU;AAAA,UACjB,WAAW,KAAK,IAAA;AAAA,QAAI,CACrB;AAAA,MACH;AACA,aAAO;AAAA,QACL,QAAQ,aAAa;AAAA,QACrB,UAAU,aAAa;AAAA,QACvB,KAAK,UAAU;AAAA,QACf,GAAI,UAAU,QAAQ,EAAE,OAAO,UAAU,MAAA,IAAU,CAAA;AAAA,MAAC;AAAA,IAExD,SAAS,OAAO;AACd,YAAM,eACJ,iBAAiB,QAAQ,MAAM,UAAU;AAC3C,oBAAc,KAAK,2BAA2B;AAAA,QAC5C;AAAA,QACA,UAAU,QAAQ;AAAA,QAClB,OAAO,QAAQ;AAAA,QACf,aAAa;AAAA,QACb;AAAA,QACA,QAAQ;AAAA,QACR,UAAU,aAAa;AAAA,QACvB,OAAO;AAAA,QACP,UAAU,KAAK,IAAA,IAAQ;AAAA,QACvB,WAAW,KAAK,IAAA;AAAA,MAAI,CACrB;AAED,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,UAAU,aAAa;AAAA,QACvB,OAAO;AAAA,MAAA;AAAA,IAEX;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/middleware/index.d.ts

@@ -0,0 +1,2 @@
+export type { GenerationActivity, GenerationMiddleware, GenerationMiddlewareContext, GenerationUsageInfo, GenerationFinishInfo, GenerationAbortInfo, GenerationErrorInfo, AnyGenerationMiddleware, } from './types.js';
+export { createGenerationContext, runGenerationStart, runGenerationUsage, runGenerationFinish, runGenerationAbort, runGenerationError, } from './run.js';

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

@@ -0,0 +1,20 @@
+import { GenerationAbortInfo, GenerationErrorInfo, GenerationFinishInfo, GenerationMiddleware, GenerationMiddlewareContext, GenerationUsageInfo } from './types.js';
+/**
+ * Build the stable context for a single media-activity call.
+ *
+ * Media activities are always server-side and carry no user runtime context,
+ * so `source` is fixed to `'server'` and `context` to `undefined`.
+ */
+export declare function createGenerationContext(args: {
+    requestId: string;
+    activity: GenerationMiddlewareContext['activity'];
+    provider: string;
+    model: string;
+    modelOptions?: unknown;
+    createId: (prefix: string) => string;
+}): GenerationMiddlewareContext;
+export declare function runGenerationStart(middleware: ReadonlyArray<GenerationMiddleware> | undefined, ctx: GenerationMiddlewareContext): Promise<void>;
+export declare function runGenerationUsage(middleware: ReadonlyArray<GenerationMiddleware> | undefined, ctx: GenerationMiddlewareContext, usage: GenerationUsageInfo): Promise<void>;
+export declare function runGenerationFinish(middleware: ReadonlyArray<GenerationMiddleware> | undefined, ctx: GenerationMiddlewareContext, info: GenerationFinishInfo): Promise<void>;
+export declare function runGenerationAbort(middleware: ReadonlyArray<GenerationMiddleware> | undefined, ctx: GenerationMiddlewareContext, info: GenerationAbortInfo): Promise<void>;
+export declare function runGenerationError(middleware: ReadonlyArray<GenerationMiddleware> | undefined, ctx: GenerationMiddlewareContext, info: GenerationErrorInfo): Promise<void>;

package/dist/esm/activities/middleware/run.js

@@ -0,0 +1,42 @@
+function createGenerationContext(args) {
+  return {
+    requestId: args.requestId,
+    activity: args.activity,
+    provider: args.provider,
+    model: args.model,
+    modelOptions: args.modelOptions,
+    source: "server",
+    createId: args.createId,
+    context: void 0
+  };
+}
+async function run(middleware, invoke) {
+  if (!middleware || middleware.length === 0) return;
+  for (const mw of middleware) {
+    await invoke(mw);
+  }
+}
+function runGenerationStart(middleware, ctx) {
+  return run(middleware, (mw) => mw.onStart?.(ctx));
+}
+function runGenerationUsage(middleware, ctx, usage) {
+  return run(middleware, (mw) => mw.onUsage?.(ctx, usage));
+}
+function runGenerationFinish(middleware, ctx, info) {
+  return run(middleware, (mw) => mw.onFinish?.(ctx, info));
+}
+function runGenerationAbort(middleware, ctx, info) {
+  return run(middleware, (mw) => mw.onAbort?.(ctx, info));
+}
+function runGenerationError(middleware, ctx, info) {
+  return run(middleware, (mw) => mw.onError?.(ctx, info));
+}
+export {
+  createGenerationContext,
+  runGenerationAbort,
+  runGenerationError,
+  runGenerationFinish,
+  runGenerationStart,
+  runGenerationUsage
+};
+//# sourceMappingURL=run.js.map

package/dist/esm/activities/middleware/run.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"run.js","sources":["../../../../src/activities/middleware/run.ts"],"sourcesContent":["import type {\n  GenerationAbortInfo,\n  GenerationErrorInfo,\n  GenerationFinishInfo,\n  GenerationMiddleware,\n  GenerationMiddlewareContext,\n  GenerationUsageInfo,\n} from './types'\n\n/**\n * Build the stable context for a single media-activity call.\n *\n * Media activities are always server-side and carry no user runtime context,\n * so `source` is fixed to `'server'` and `context` to `undefined`.\n */\nexport function createGenerationContext(args: {\n  requestId: string\n  activity: GenerationMiddlewareContext['activity']\n  provider: string\n  model: string\n  modelOptions?: unknown\n  createId: (prefix: string) => string\n}): GenerationMiddlewareContext {\n  return {\n    requestId: args.requestId,\n    activity: args.activity,\n    provider: args.provider,\n    model: args.model,\n    modelOptions: args.modelOptions,\n    source: 'server',\n    createId: args.createId,\n    context: undefined,\n  }\n}\n\n/**\n * Run a single lifecycle hook across each middleware in registration order,\n * awaiting each. Exceptions PROPAGATE (matching `chat()` middleware) — a\n * broken middleware fails the activity rather than being silently swallowed.\n */\nasync function run(\n  middleware: ReadonlyArray<GenerationMiddleware> | undefined,\n  invoke: (mw: GenerationMiddleware) => void | Promise<void>,\n): Promise<void> {\n  if (!middleware || middleware.length === 0) return\n  for (const mw of middleware) {\n    await invoke(mw)\n  }\n}\n\nexport function runGenerationStart(\n  middleware: ReadonlyArray<GenerationMiddleware> | undefined,\n  ctx: GenerationMiddlewareContext,\n): Promise<void> {\n  return run(middleware, (mw) => mw.onStart?.(ctx))\n}\n\nexport function runGenerationUsage(\n  middleware: ReadonlyArray<GenerationMiddleware> | undefined,\n  ctx: GenerationMiddlewareContext,\n  usage: GenerationUsageInfo,\n): Promise<void> {\n  return run(middleware, (mw) => mw.onUsage?.(ctx, usage))\n}\n\nexport function runGenerationFinish(\n  middleware: ReadonlyArray<GenerationMiddleware> | undefined,\n  ctx: GenerationMiddlewareContext,\n  info: GenerationFinishInfo,\n): Promise<void> {\n  return run(middleware, (mw) => mw.onFinish?.(ctx, info))\n}\n\nexport function runGenerationAbort(\n  middleware: ReadonlyArray<GenerationMiddleware> | undefined,\n  ctx: GenerationMiddlewareContext,\n  info: GenerationAbortInfo,\n): Promise<void> {\n  return run(middleware, (mw) => mw.onAbort?.(ctx, info))\n}\n\nexport function runGenerationError(\n  middleware: ReadonlyArray<GenerationMiddleware> | undefined,\n  ctx: GenerationMiddlewareContext,\n  info: GenerationErrorInfo,\n): Promise<void> {\n  return run(middleware, (mw) => mw.onError?.(ctx, info))\n}\n"],"names":[],"mappings":"AAeO,SAAS,wBAAwB,MAOR;AAC9B,SAAO;AAAA,IACL,WAAW,KAAK;AAAA,IAChB,UAAU,KAAK;AAAA,IACf,UAAU,KAAK;AAAA,IACf,OAAO,KAAK;AAAA,IACZ,cAAc,KAAK;AAAA,IACnB,QAAQ;AAAA,IACR,UAAU,KAAK;AAAA,IACf,SAAS;AAAA,EAAA;AAEb;AAOA,eAAe,IACb,YACA,QACe;AACf,MAAI,CAAC,cAAc,WAAW,WAAW,EAAG;AAC5C,aAAW,MAAM,YAAY;AAC3B,UAAM,OAAO,EAAE;AAAA,EACjB;AACF;AAEO,SAAS,mBACd,YACA,KACe;AACf,SAAO,IAAI,YAAY,CAAC,OAAO,GAAG,UAAU,GAAG,CAAC;AAClD;AAEO,SAAS,mBACd,YACA,KACA,OACe;AACf,SAAO,IAAI,YAAY,CAAC,OAAO,GAAG,UAAU,KAAK,KAAK,CAAC;AACzD;AAEO,SAAS,oBACd,YACA,KACA,MACe;AACf,SAAO,IAAI,YAAY,CAAC,OAAO,GAAG,WAAW,KAAK,IAAI,CAAC;AACzD;AAEO,SAAS,mBACd,YACA,KACA,MACe;AACf,SAAO,IAAI,YAAY,CAAC,OAAO,GAAG,UAAU,KAAK,IAAI,CAAC;AACxD;AAEO,SAAS,mBACd,YACA,KACA,MACe;AACf,SAAO,IAAI,YAAY,CAAC,OAAO,GAAG,UAAU,KAAK,IAAI,CAAC;AACxD;"}

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

@@ -0,0 +1,118 @@
+import { TokenUsage } from '../../types.js';
+/**
+ * The activity an observability event describes.
+ *
+ * Mirrors the public surface a caller reaches for: `'chat'` for `chat()`, and
+ * the media kinds for the `generate*` activities. `'tts'` matches the speech
+ * adapter's kind (the public discriminator avoids inventing a parallel
+ * `'speech'`/`'text'` vocabulary). `otelMiddleware` maps each to its
+ * `gen_ai.operation.name`.
+ */
+export type GenerationActivity = 'chat' | 'image' | 'video' | 'audio' | 'tts' | 'transcription';
+/**
+ * Stable context passed to every {@link GenerationMiddleware} hook. Created
+ * once per activity call and shared across the hooks of that call.
+ *
+ * Carries only fields every activity can honor. `ChatMiddlewareContext`
+ * structurally includes all of these plus chat-only state (messages,
+ * iteration, capabilities, …), which is why a chat middleware that reads those
+ * extra fields is not assignable to `GenerationMiddleware`.
+ */
+export interface GenerationMiddlewareContext<TContext = unknown> {
+    /**
+     * Stable id correlating the `onStart` / `onFinish` / `onError` / `onAbort`
+     * hooks of a single activity call.
+     */
+    requestId: string;
+    /** Which activity this call is. Discriminates media from chat. */
+    activity: GenerationActivity;
+    /** Provider/adapter name (e.g. `"openai"`). Emitted as `gen_ai.system`. */
+    provider: string;
+    /** Model id. Emitted as `gen_ai.request.model`. */
+    model: string;
+    /**
+     * Provider-specific options passed to the activity, if any. Typed `unknown`
+     * because each activity's options are strongly typed per model; a supertype
+     * of `ChatMiddlewareContext`'s `modelOptions`.
+     */
+    modelOptions?: unknown;
+    /** Where the call originates. Always `'server'` for media activities. */
+    source: 'client' | 'server';
+    /** Generate a unique id with the given prefix. */
+    createId: (prefix: string) => string;
+    /** Runtime context provided by the activity options, if any. */
+    context: TContext;
+}
+/**
+ * Token usage passed to {@link GenerationMiddleware.onUsage}. Kept as an
+ * interface extending `TokenUsage` to preserve declaration merging for this
+ * publicly exported type.
+ */
+export interface GenerationUsageInfo extends TokenUsage {
+}
+/** Information passed to {@link GenerationMiddleware.onFinish}. */
+export interface GenerationFinishInfo {
+    /** Wall-clock duration of the activity call, in milliseconds. */
+    duration: number;
+    /** Unified usage, when the provider reported it. */
+    usage?: TokenUsage | undefined;
+}
+/** Information passed to {@link GenerationMiddleware.onAbort}. */
+export interface GenerationAbortInfo {
+    /** The reason for the abort, if provided. */
+    reason?: string;
+    /** Wall-clock duration until the abort, in milliseconds. */
+    duration: number;
+}
+/** Information passed to {@link GenerationMiddleware.onError}. */
+export interface GenerationErrorInfo {
+    /** The thrown value (typically an `Error`). */
+    error: unknown;
+    /** Wall-clock duration until the failure, in milliseconds. */
+    duration: number;
+}
+/**
+ * Activity-agnostic, observe-only middleware.
+ *
+ * A thin lifecycle observer registerable on any activity via its `middleware`
+ * option. Unlike `ChatMiddleware` (which can also rewrite config, chunks, and
+ * tool calls), these hooks only observe — the right fit for the single
+ * request → response shape of media activities. Pass `otelMiddleware()` for
+ * OpenTelemetry, or implement the hooks directly for a custom backend.
+ *
+ * Hooks are awaited in registration order. A hook that throws PROPAGATES and
+ * fails the activity — matching `chat()` middleware semantics. Keep them cheap;
+ * they run inline with the request.
+ *
+ * Exactly one of `onFinish` / `onAbort` / `onError` fires per call.
+ *
+ * @example
+ * ```ts
+ * import { generateImage } from '@tanstack/ai'
+ * import { otelMiddleware } from '@tanstack/ai/middlewares/otel'
+ * import { openaiImage } from '@tanstack/ai-openai'
+ * import { trace } from '@opentelemetry/api'
+ *
+ * await generateImage({
+ *   adapter: openaiImage('gpt-image-1'),
+ *   prompt: 'A serene mountain landscape at sunset',
+ *   middleware: [otelMiddleware({ tracer: trace.getTracer('my-app') })],
+ * })
+ * ```
+ */
+export interface GenerationMiddleware<TContext = unknown> {
+    /** Optional name, surfaced in diagnostics. */
+    name?: string;
+    /** Called before the adapter request begins. */
+    onStart?: (ctx: GenerationMiddlewareContext<TContext>) => void | Promise<void>;
+    /** Called when the provider reports usage, before `onFinish`. */
+    onUsage?: (ctx: GenerationMiddlewareContext<TContext>, usage: GenerationUsageInfo) => void | Promise<void>;
+    /** Called after the activity completes successfully. */
+    onFinish?: (ctx: GenerationMiddlewareContext<TContext>, info: GenerationFinishInfo) => void | Promise<void>;
+    /** Called when the activity is aborted (e.g. an abandoned stream). */
+    onAbort?: (ctx: GenerationMiddlewareContext<TContext>, info: GenerationAbortInfo) => void | Promise<void>;
+    /** Called when the activity throws before completing. */
+    onError?: (ctx: GenerationMiddlewareContext<TContext>, info: GenerationErrorInfo) => void | Promise<void>;
+}
+/** A `GenerationMiddleware` with a permissive context — for use as a constraint. */
+export type AnyGenerationMiddleware = GenerationMiddleware<any>;

package/dist/esm/index.d.ts

@@ -13,11 +13,13 @@ export { MCPDuplicateToolNameError } from './activities/chat/mcp/manager.js';
 export { convertSchemaToJsonSchema, isStandardSchema, parseWithStandardSchema, StandardSchemaValidationError, } from './activities/chat/tools/schema-converter.js';
 export { streamToText, toServerSentEventsStream, toServerSentEventsResponse, toHttpStream, toHttpResponse, } from './stream-to-response.js';
 export { ToolCallManager } from './activities/chat/tools/tool-calls.js';
+export { DISCOVERY_TOOL_NAME } from './activities/chat/tools/lazy-tool-manager.js';
 export type { ProviderTool } from './tools/provider-tool.js';
 export { brandProviderTool } from './tools/provider-tool.js';
 export { maxIterations, untilFinishReason, combineStrategies, } from './activities/chat/agent-loop-strategies.js';
 export { createToolRegistry, createFrozenRegistry, type ToolRegistry, } from './tool-registry.js';
 export type { ChatMiddleware, ChatMiddlewareContext, ChatMiddlewarePhase, ChatMiddlewareConfig, StructuredOutputMiddlewareConfig, ToolCallHookContext, BeforeToolCallDecision, AfterToolCallInfo, IterationInfo, ToolPhaseCompleteInfo, UsageInfo, FinishInfo, AbortInfo, ErrorInfo, } from './activities/chat/middleware/index.js';
+export type { GenerationMiddleware, GenerationMiddlewareContext, GenerationActivity, GenerationUsageInfo, GenerationFinishInfo, GenerationAbortInfo, GenerationErrorInfo, AnyGenerationMiddleware, } from './activities/middleware/index.js';
 export { createCapability, defineChatMiddleware, createChatMiddleware, } from './activities/chat/middleware/index.js';
 export type { Capability, CapabilityHandle, CapabilityContext, CapabilityGetter, CapabilityProvider, } from './activities/chat/middleware/index.js';
 export * from './types.js';

package/dist/esm/index.js

@@ -10,6 +10,7 @@ import { MCPDuplicateToolNameError } from "./activities/chat/mcp/manager.js";
 import { StandardSchemaValidationError, convertSchemaToJsonSchema, isStandardSchema, parseWithStandardSchema } from "./activities/chat/tools/schema-converter.js";
 import { streamToText, toHttpResponse, toHttpStream, toServerSentEventsResponse, toServerSentEventsStream } from "./stream-to-response.js";
 import { ToolCallManager } from "./activities/chat/tools/tool-calls.js";
+import { DISCOVERY_TOOL_NAME } from "./activities/chat/tools/lazy-tool-manager.js";
 import { brandProviderTool } from "./tools/provider-tool.js";
 import { combineStrategies, maxIterations, untilFinishReason } from "./activities/chat/agent-loop-strategies.js";
 import { createFrozenRegistry, createToolRegistry } from "./tool-registry.js";
@@ -35,6 +36,7 @@ export {
   BatchStrategy,
   CompositeStrategy,
   ConsoleLogger,
+  DISCOVERY_TOOL_NAME,
   EventType,
   ImmediateStrategy,
   MCPDuplicateToolNameError,

package/dist/esm/index.js.map

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

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

@@ -1,13 +1,16 @@
 import { AttributeValue, Meter, Span, SpanOptions, Tracer } from '@opentelemetry/api';
 import { ChatMiddleware, ChatMiddlewareContext } from '../activities/chat/middleware/types.js';
+import { GenerationMiddleware, GenerationMiddlewareContext } from '../activities/middleware/types.js';
 /**
  * Scope (role) of an OTel span emitted by this middleware.
  *
  * - `chat` — the root span for a single `chat()` call
  * - `iteration` — one per agent-loop iteration (one model call)
  * - `tool` — one per tool execution inside an iteration
+ * - `generation` — the single span for a media activity call
+ *   (`generateImage`, `generateVideo`, `generateSpeech`, …)
  */
-export type OtelSpanScope = 'chat' | 'iteration' | 'tool';
+export type OtelSpanScope = 'chat' | 'iteration' | 'tool' | 'generation';
 /**
  * Alias retained for backwards compatibility. Prefer {@link OtelSpanScope}.
  *
@@ -33,6 +36,9 @@ export type OtelSpanInfo<TScope extends OtelSpanScope = OtelSpanScope> = TScope
     iteration: number;
     toolName: string;
     toolCallId: string;
+} : TScope extends 'generation' ? {
+    kind: 'generation';
+    ctx: GenerationMiddlewareContext;
 } : never;
 export interface OtelMiddlewareOptions {
     /** OTel `Tracer` used to start root, iteration, and tool spans. */
@@ -72,4 +78,4 @@ export interface OtelMiddlewareOptions {
     /** Fires just before every `span.end()`. */
     onSpanEnd?: (info: OtelSpanInfo, span: Span) => void;
 }
-export declare function otelMiddleware(options: OtelMiddlewareOptions): ChatMiddleware;
+export declare function otelMiddleware(options: OtelMiddlewareOptions): GenerationMiddleware & ChatMiddleware;