AI创作批量生成任务 main V1 init

video-gen-api/app/models/video_engine.py

@@ -1,4 +1,4 @@
-from sqlalchemy import Boolean, Integer, String
+from sqlalchemy import Boolean, CheckConstraint, Integer, String
 from sqlalchemy.orm import Mapped, mapped_column
 
 from app.models.base import Base, TimestampMixin
@@ -6,6 +6,9 @@ from app.models.base import Base, TimestampMixin
 
 class VideoEngine(Base, TimestampMixin):
     __tablename__ = "video_engines"
+    __table_args__ = (
+        CheckConstraint("max_generation_count BETWEEN 1 AND 5", name="ck_video_engines_max_generation_count"),
+    )
 
     id: Mapped[str] = mapped_column(String(32), primary_key=True)
     name: Mapped[str] = mapped_column(String(64), nullable=False)
@@ -20,6 +23,21 @@ class VideoEngine(Base, TimestampMixin):
     max_image_count: Mapped[int] = mapped_column(Integer, default=2)
     max_video_count: Mapped[int] = mapped_column(Integer, default=0)
     max_audio_count: Mapped[int] = mapped_column(Integer, default=0)
+
+    # 管理后台只配置能力开关与数量上限；本次实际生成数量保存在 ChatGenerationTask.generation_count。
+    multi_generation_enabled: Mapped[bool] = mapped_column(
+        Boolean,
+        default=False,
+        server_default="false",
+        nullable=False,
+    )
+    max_generation_count: Mapped[int] = mapped_column(
+        Integer,
+        default=1,
+        server_default="1",
+        nullable=False,
+    )
+
     supports_first_last_frame: Mapped[bool] = mapped_column(Boolean, default=False)
     supports_universal_reference: Mapped[bool] = mapped_column(Boolean, default=True)
     generate_url: Mapped[str | None] = mapped_column(String(512), nullable=True, default="")