o ۷i@sdZddlmZddlmZddlmZmZmZGddde eZ GdddeZ Gd d d eZ Gd d d eZ Gd ddeZdS)z OpenAI-compatible protocol definitions for video generation. This module provides Pydantic models for a video generation endpoint that mirrors the OpenAI Images API shape, with vllm-omni extensions for diffusion video models (e.g., Wan2.2). )Enum)Any) BaseModelFieldfield_validatorc@seZdZdZdZdZdS)VideoResponseFormatzVideo response format.b64_jsonurlN)__name__ __module__ __qualname____doc__B64_JSONURLrrb/home/ubuntu/vllm_env/lib/python3.10/site-packages/vllm_omni/entrypoints/openai/protocol/videos.pyrsrc@szeZdZUdZeddddZedBed<eddddZedBed<eddd dZ edBed <eddd dZ edBed <dS) VideoParamsz-Optional block for video-specific parameters.NVideo width in pixelsdefaultge descriptionwidthVideo height in pixelsheightzNumber of frames num_frames"Frames per second for output videofps) r r r r rrint__annotations__rrrrrrrrs rc@seZdZUdZedddZeed<edddZedBed <ed d d d d Z e ed<edddZ e eBdBed<edddZ edBed<ee jddZe ed<edddZedBed<edddZedBed<edddZedBed<edddZedBed<edd ddZe dBed <edd d!dZe dBed"<edd d#dZe dBed$<edd d%dZe dBed&<edd'dZedBed(<edd d)d*d Ze dBed+<edd,d-d.d ZedBed/<edd,d-d0d ZedBed1<edd,d2d3d ZedBed4<edd5dZ edBed6<edd,d-d7d Z!edBed8<edd9dZ"e dBed:<edd;dZ#e$ee%fdBed<<e&de'd=d>Z(e&de'd?d@Z)e&de'dAdBZ*dS)CVideoGenerationRequestzz OpenAI-style video generation request. Follows the OpenAI Images API conventions with extensions for video. .z(Text description of the desired video(s)rpromptNzBModel to use (optional, uses server's configured model if omitted)rrmodelrzNumber of videos to generate)rrlernzrrrvalidate_response_formatsz/VideoGenerationRequest.validate_response_formatcCsD|durdSt|tr|stdt|St|tr|Std)Nz,seconds must be an integer or numeric string)r;r<isdigitr=rr>rrrvalidate_secondss  z'VideoGenerationRequest.validate_seconds)+r r r r rr#r<r r%r(rr)r*rrr+r,r-boolr.rr/rrrrr0r2r3floatr4r5r6r7r8r9dictrr classmethodrArBrDrrrrr!!s    r!c@sZeZdZUdZedddZedBed<edddZedBed<edddZ edBed <dS) VideoDatazSingle generated video data.NzBase64-encoded MP4 videor$rzVideo URL (not implemented)r z2Revised prompt (OpenAI compatibility, always null)revised_prompt) r r r r rrr<r r rJrrrrrIs rIc@s>eZdZUdZedddZeed<edddZe e ed<dS) VideoGenerationResponsez'OpenAI-style video generation response..z/Unix timestamp of when the generation completedr"createdzArray of generated videosdataN) r r r r rrLrr rMlistrIrrrrrKs rKN)r enumrtypingrpydanticrrrr<rrr!rIrKrrrrs   {