diff --git a/video-gen-api/app/api/v1/generation_ai.py b/video-gen-api/app/api/v1/generation_ai.py index 17842a22..6ebbca95 100644 --- a/video-gen-api/app/api/v1/generation_ai.py +++ b/video-gen-api/app/api/v1/generation_ai.py @@ -90,6 +90,8 @@ router = APIRouter( "supported_durations": [4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15], "max_duration": 15, "priority": 10, + "supports_first_last_frame": False, + "supports_universal_reference": False, } ], } diff --git a/video-gen-api/app/schemas/generation_ai.py b/video-gen-api/app/schemas/generation_ai.py index f18d2629..208c6c7e 100644 --- a/video-gen-api/app/schemas/generation_ai.py +++ b/video-gen-api/app/schemas/generation_ai.py @@ -193,6 +193,8 @@ class GenerationAIVideoEngineOptionOut(BaseModel): priority: int = Field(0, description="引擎优先级,数值越大越优先") max_image_count: int | None = Field(None, description="最大图片数量") max_video_count: int | None = Field(None, description="最大视频数量") + supports_first_last_frame: bool = Field(False, description="是否支持首帧和最后一帧") + supports_universal_reference: bool = Field(False, description="是否支持通用参考") class GenerationAIEngineGroupOut(BaseModel): diff --git a/video-gen-api/app/services/generation_ai_service.py b/video-gen-api/app/services/generation_ai_service.py index 4f3228df..0b4a25ab 100644 --- a/video-gen-api/app/services/generation_ai_service.py +++ b/video-gen-api/app/services/generation_ai_service.py @@ -203,6 +203,8 @@ async def list_generation_ai_engine_options(db: AsyncSession) -> GenerationAIEng priority=engine.priority or 0, max_image_count=engine.max_image_count, max_video_count=engine.max_video_count, + supports_first_last_frame=engine.supports_first_last_frame, + supports_universal_reference=engine.supports_universal_reference, ) for engine in video_result.scalars().all() ]