@tanstack/ai
Version:
Type-safe TypeScript AI SDK for streaming chat, tool calling, agents, structured outputs, and multimodal generation.
146 lines (145 loc) • 6.88 kB
TypeScript
import { ModelInputModalitiesByName, VideoGenerationOptions, VideoJobResult, VideoStatusResult, VideoUrlResult } from '../../types.js';
/**
* Structured description of the durations a video model accepts.
*
* Tagged union so the same shape can express discrete enums (OpenAI Sora,
* Veo), continuous ranges, mixed shapes, and models with no duration field.
* Consumed by `VideoAdapter.availableDurations()`.
*
* @experimental Video generation is an experimental feature and may change.
*/
export type DurationOptions<T extends string | number | undefined> = {
kind: 'discrete';
values: ReadonlyArray<NonNullable<T>>;
} | {
kind: 'range';
min: number;
max: number;
step?: number;
unit: 'seconds';
} | {
kind: 'mixed';
values: ReadonlyArray<NonNullable<T>>;
range?: {
min: number;
max: number;
step?: number;
};
} | {
kind: 'none';
};
/**
* 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
* - TModelInputModalitiesByName: Map from model name to the non-text prompt
* modalities it accepts (constrains the `prompt` part types at compile time)
* - TModelDurationByName: Map from model name to its supported duration
* union. Defaults to `Record<string, number>` so adapters that haven't
* declared a map keep today's `duration?: number` typing.
*/
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 | undefined> = Record<string, string>, TModelInputModalitiesByName extends ModelInputModalitiesByName = ModelInputModalitiesByName, TModelDurationByName extends Record<string, string | number | undefined> = Record<string, number>> {
/** 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;
modelInputModalitiesByName: TModelInputModalitiesByName;
modelDurationByName: TModelDurationByName;
};
/**
* 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], TModelDurationByName[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>;
/**
* Describe the durations this adapter's model accepts. Returns a tagged
* union so consumers can render UI / coerce input without provider-specific
* knowledge.
*/
availableDurations: () => DurationOptions<TModelDurationByName[TModel]>;
/**
* Coerce a raw seconds value to the closest valid duration for this model.
* Returns `undefined` for models with no duration field.
*/
snapDuration: (seconds: number) => TModelDurationByName[TModel] | undefined;
}
/**
* A VideoAdapter with any/unknown type parameters.
* Useful as a constraint in generic functions and interfaces.
*/
export type AnyVideoAdapter = VideoAdapter<any, any, 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 | undefined> = Record<string, string>, TModelInputModalitiesByName extends ModelInputModalitiesByName = ModelInputModalitiesByName, TModelDurationByName extends Record<string, string | number | undefined> = Record<string, number>> implements VideoAdapter<TModel, TProviderOptions, TModelProviderOptionsByName, TModelSizeByName, TModelInputModalitiesByName, TModelDurationByName> {
readonly kind: "video";
abstract readonly name: string;
readonly model: TModel;
'~types': {
providerOptions: TProviderOptions;
modelProviderOptionsByName: TModelProviderOptionsByName;
modelSizeByName: TModelSizeByName;
modelInputModalitiesByName: TModelInputModalitiesByName;
modelDurationByName: TModelDurationByName;
};
protected config: VideoAdapterConfig;
constructor(config: VideoAdapterConfig | undefined, model: TModel);
abstract createVideoJob(options: VideoGenerationOptions<TProviderOptions, TModelSizeByName[TModel], TModelDurationByName[TModel]>): Promise<VideoJobResult>;
abstract getVideoStatus(jobId: string): Promise<VideoStatusResult>;
abstract getVideoUrl(jobId: string): Promise<VideoUrlResult>;
/**
* Default implementation returns `{ kind: 'none' }`. Adapters that have
* declared their per-model duration map should override this.
*/
availableDurations(): DurationOptions<TModelDurationByName[TModel]>;
/**
* Default implementation returns `undefined`. Adapters that have declared
* their per-model duration map should override.
*/
snapDuration(_seconds: number): TModelDurationByName[TModel] | undefined;
protected generateId(): string;
}