o ۷i@svdZddlmZddlmZddlmZmZmZGddde eZ GdddeZ Gd d d eZ Gd d d eZ d S)z OpenAI-compatible protocol definitions for image generation. This module provides Pydantic models that follow the OpenAI DALL-E API specification for text-to-image generation, with vllm-omni specific extensions. )Enum)Any) BaseModelFieldfield_validatorc@seZdZdZdZdZdS)ResponseFormatzImage response formatb64_jsonurlN)__name__ __module__ __qualname____doc__B64_JSONURLrrb/home/ubuntu/vllm_env/lib/python3.10/site-packages/vllm_omni/entrypoints/openai/protocol/images.pyrsrc@seZdZUdZedddZeed<edddZedBed <ed d d d d Z e ed<edddZ edBed<ee j ddZe ed<edddZedBed<ededdZededdZedddZedBed<edd ddd Ze dBed<edddd d ZedBed!<edddd"d ZedBed#<edd$dZe dBed%<edd&dZedBed'<edd(dZeeefdBed)<ed*d+dZedBed,<ed*d-dZ edBed.<dS)/ImageGenerationRequestz OpenAI DALL-E compatible image generation request. Follows the OpenAI Images API specification with vllm-omni extensions for advanced diffusion parameters. .z(Text description of the desired image(s) descriptionpromptNzBModel to use (optional, uses server's configured model if omitted)defaultrmodel zNumber of images to generate)rgelernz[Image dimensions in WIDTHxHEIGHT format (e.g., '1024x1024', uses model defaults if omitted)sizezFormat of the returned imageresponse_formatzUser identifier for trackingusercCs*|durdSt|trd|vrtd|S)zValidate size parameter. Accepts any string in 'WIDTHxHEIGHT' format (e.g., '1024x1024', '512x768'). No restrictions on specific dimensions - models can handle arbitrary sizes. Nxz9size must be in format 'WIDTHxHEIGHT' (e.g., '1024x1024')) isinstancestr ValueErrorclsvrrr validate_size/s z$ImageGenerationRequest.validate_sizecCs$|dur|tjkrtd||S)z6Validate response format - only b64_json is supported.Nz3Only 'b64_json' response format is supported, got: )rrr$r%rrrvalidate_response_format>sz/ImageGenerationRequest.validate_response_formatz*Text describing what to avoid in the imagenegative_promptzINumber of diffusion sampling steps (uses model defaults if not specified)num_inference_stepsgg4@zEClassifier-free guidance scale (uses model defaults if not specified)guidance_scalezJTrue CFG scale (model-specific parameter, may be ignored if not supported)true_cfg_scalezRandom seed for reproducibilityseedz\Device for the seeded torch.Generator (e.g. 'cpu', 'cuda'). Defaults to the runner's device.generator_devicezOptional LoRA adapter for this request. Expected shape: {name/path/scale/int_id}. Field names are flexible (e.g. name|lora_name|adapter, path|lora_path|local_path, scale|lora_scale, int_id|lora_int_id).loraFzEnable VAE slicingvae_use_slicingzEnable VAE tilingvae_use_tiling)!r r r r rrr#__annotations__rrintrrrrr r classmethodr(r)r*r,r-floatr.r/r0r1dictrr2boolr3rrrrrsd     rc@sZeZdZUdZedddZedBed<edddZedBed<edddZ edBed <dS) ImageDatazSingle generated image dataNzBase64-encoded PNG imagerrzImage URL (not implemented)r z2Revised prompt (OpenAI compatibility, always null)revised_prompt) r r r r rrr#r4r r;rrrrr:qs r:c@sfeZdZUdZedddZeed<edddZe e ed<edd dZ e ed <edd dZ e ed <dS) ImageGenerationResponsezj OpenAI DALL-E compatible image generation response. Returns generated images with metadata. .z/Unix timestamp of when the generation completedrcreatedzArray of generated imagesdataNz)The output format of the image generation output_formatzThe size of the image generatedr)r r r r rr=r5r4r>listr:r?r#rrrrrr<ys r<N)r enumrtypingrpydanticrrrr#rrr:r<rrrrs  Z