diff --git a/backend/src/common/base_dto.py b/backend/src/common/base_dto.py index 78b047f0..1832b9e8 100644 --- a/backend/src/common/base_dto.py +++ b/backend/src/common/base_dto.py @@ -68,10 +68,6 @@ class GenerationModelEnum(str, Enum): VTO = "virtual-try-on-001" # Video-Specific Models - VEO_3_1_FAST_GENERATE_001 = "veo-3.1-fast-generate-001" - VEO_3_1_LITE_GENERATE_001 = "veo-3.1-lite-generate-001" - VEO_3_1_GENERATE_001 = "veo-3.1-generate-001" - VEO_3_1_PREVIEW = "veo-3.1-generate-preview" VEO_3_FAST = "veo-3.0-fast-generate-001" VEO_3_QUALITY = "veo-3.0-generate-001" VEO_3_FAST_PREVIEW = "veo-3.0-fast-generate-preview" diff --git a/backend/src/videos/dto/create_veo_dto.py b/backend/src/videos/dto/create_veo_dto.py index 71406be7..0bd7aa40 100644 --- a/backend/src/videos/dto/create_veo_dto.py +++ b/backend/src/videos/dto/create_veo_dto.py @@ -56,7 +56,7 @@ class CreateVeoDto(BaseDto): description="The ID of the workspace for this generation.", ) generation_model: GenerationModelEnum = Field( - default=GenerationModelEnum.VEO_3_1_GENERATE_001, + default=GenerationModelEnum.VEO_3_QUALITY, description="Model used for image generation.", ) aspect_ratio: AspectRatioEnum = Field( @@ -167,17 +167,10 @@ def validate_cross_fields(self) -> "CreateVeoDto": if has_any_references: if ( model != GenerationModelEnum.VEO_2_GENERATE_EXP - and model != GenerationModelEnum.VEO_3_1_PREVIEW - and model != GenerationModelEnum.VEO_3_1_GENERATE_001 - and model != GenerationModelEnum.VEO_3_1_LITE_GENERATE_001 - and model != GenerationModelEnum.VEO_3_1_FAST_GENERATE_001 ): raise ValueError( "Reference images are only supported by the " - f"'{GenerationModelEnum.VEO_3_1_PREVIEW.value}' model, " - f"'{GenerationModelEnum.VEO_3_1_GENERATE_001.value}' model, " - f"'{GenerationModelEnum.VEO_3_1_LITE_GENERATE_001.value}' model, or " - f"'{GenerationModelEnum.VEO_3_1_FAST_GENERATE_001.value}' model.", + f"'{GenerationModelEnum.VEO_2_GENERATE_EXP.value}' model.", ) start_image_present = bool(self.start_image_asset_id) @@ -218,10 +211,6 @@ def validate_video_generation_model( ) -> GenerationModelEnum: """Ensures that only supported generation models for video are used.""" valid_video_ratios = [ - GenerationModelEnum.VEO_3_1_PREVIEW, - GenerationModelEnum.VEO_3_1_GENERATE_001, - GenerationModelEnum.VEO_3_1_LITE_GENERATE_001, - GenerationModelEnum.VEO_3_1_FAST_GENERATE_001, GenerationModelEnum.VEO_3_FAST, GenerationModelEnum.VEO_3_QUALITY, GenerationModelEnum.VEO_3_FAST_PREVIEW,