UNPKG

@tanstack/ai

Version:

Core TanStack AI library - Open source AI SDK

87 lines (86 loc) 3.98 kB
import { VideoGenerationOptions, VideoJobResult, VideoStatusResult, VideoUrlResult } from '../../types.js'; /** * Configuration for video adapter instances * * @experimental Video generation is an experimental feature and may change. */ export interface VideoAdapterConfig { apiKey?: string; baseUrl?: string; timeout?: number; maxRetries?: number; headers?: Record<string, string>; } /** * Video adapter interface with pre-resolved generics. * * An adapter is created by a provider function: `provider('model')` → `adapter` * All type resolution happens at the provider call site, not in this interface. * * @experimental Video generation is an experimental feature and may change. * * Generic parameters: * - TModel: The specific model name (e.g., 'sora-2') * - TProviderOptions: Provider-specific options (already resolved) * - TModelProviderOptionsByName: Map from model name to its specific provider options * - TModelSizeByName: Map from model name to its supported sizes */ export interface VideoAdapter<TModel extends string = string, TProviderOptions extends object = Record<string, unknown>, TModelProviderOptionsByName extends Record<string, any> = Record<string, any>, TModelSizeByName extends Record<string, string> = Record<string, string>> { /** Discriminator for adapter kind - used to determine API shape */ readonly kind: 'video'; /** Adapter name identifier */ readonly name: string; /** The model this adapter is configured for */ readonly model: TModel; /** * @internal Type-only properties for inference. Not assigned at runtime. */ '~types': { providerOptions: TProviderOptions; modelProviderOptionsByName: TModelProviderOptionsByName; modelSizeByName: TModelSizeByName; }; /** * Create a new video generation job. * Returns a job ID that can be used to poll for status and retrieve the video. */ createVideoJob: (options: VideoGenerationOptions<TProviderOptions, TModelSizeByName[TModel]>) => Promise<VideoJobResult>; /** * Get the current status of a video generation job. */ getVideoStatus: (jobId: string) => Promise<VideoStatusResult>; /** * Get the URL to download/view the generated video. * Should only be called after status is 'completed'. */ getVideoUrl: (jobId: string) => Promise<VideoUrlResult>; } /** * A VideoAdapter with any/unknown type parameters. * Useful as a constraint in generic functions and interfaces. */ export type AnyVideoAdapter = VideoAdapter<any, any, any, any>; /** * Abstract base class for video generation adapters. * Extend this class to implement a video adapter for a specific provider. * * @experimental Video generation is an experimental feature and may change. * * Generic parameters match VideoAdapter - all pre-resolved by the provider function. */ export declare abstract class BaseVideoAdapter<TModel extends string = string, TProviderOptions extends object = Record<string, unknown>, TModelProviderOptionsByName extends Record<string, any> = Record<string, any>, TModelSizeByName extends Record<string, string> = Record<string, string>> implements VideoAdapter<TModel, TProviderOptions, TModelProviderOptionsByName, TModelSizeByName> { readonly kind: "video"; abstract readonly name: string; readonly model: TModel; '~types': { providerOptions: TProviderOptions; modelProviderOptionsByName: TModelProviderOptionsByName; modelSizeByName: TModelSizeByName; }; protected config: VideoAdapterConfig; constructor(config: VideoAdapterConfig | undefined, model: TModel); abstract createVideoJob(options: VideoGenerationOptions<TProviderOptions, TModelSizeByName[TModel]>): Promise<VideoJobResult>; abstract getVideoStatus(jobId: string): Promise<VideoStatusResult>; abstract getVideoUrl(jobId: string): Promise<VideoUrlResult>; protected generateId(): string; }