1

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

@@ -5,6 +5,7 @@ from datetime import datetime
 from sqlalchemy import DateTime, ForeignKey, Index, String, Text, text
 from sqlalchemy.orm import Mapped, mapped_column
 
+from app.enums.common import ModuleGenerationFlowVersionEnum
 from app.models.base import Base, SoftDeleteMixin, TimestampMixin
 
 
@@ -36,6 +37,12 @@ class ModuleGenerationProject(Base, TimestampMixin, SoftDeleteMixin):
         String(32), ForeignKey("users.id", ondelete="CASCADE"), index=True, nullable=False
     )
     module: Mapped[str] = mapped_column(String(64), index=True, nullable=False)
+    flow_version: Mapped[str] = mapped_column(
+        String(16),
+        nullable=False,
+        default=ModuleGenerationFlowVersionEnum.V1.value,
+        server_default=ModuleGenerationFlowVersionEnum.V1.value,
+    )
     title: Mapped[str | None] = mapped_column(String(160), nullable=True)
     status: Mapped[str] = mapped_column(String(32), default="pending", index=True)
     current_step_code: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)