爆款开头API开发完成文件追加

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

@@ -21,6 +21,8 @@ from app.models.chat_provider_call_log import ChatProviderCallLog
 from app.models.generated_resource import GeneratedResource
 from app.models.user_resource_month_stat import UserResourceMonthStat
 from app.models.user_resource_total_stat import UserResourceTotalStat
+from app.models.module_generation_project import ModuleGenerationProject
+from app.models.module_generation_step import ModuleGenerationStep
 from app.models.user_oauth import UserOAuth
 from app.models.user_oauth_account import UserOAuthAccount
 from app.models.user_oauth_app import UserOAuthApp
@@ -34,5 +36,6 @@ __all__ = [
     "MenuConfig", "RechargePackage", "OperationLog",
     "ChatGenerationTask", "ChatGenerationTaskEvent", "ChatProviderCallLog",
     "GeneratedResource", "UserResourceMonthStat", "UserResourceTotalStat",
+    "ModuleGenerationProject", "ModuleGenerationStep",
     "UserOAuth", "UserOAuthAccount", "UserOAuthApp",
 ]

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

@@ -1,6 +1,6 @@
 from datetime import datetime
 
-from sqlalchemy import DateTime, Float, ForeignKey, Index, Integer, String, Text
+from sqlalchemy import DateTime, Float, ForeignKey, Index, Integer, String, Text, text
 from sqlalchemy.orm import Mapped, mapped_column
 
 from app.models.base import Base, TimestampMixin, SoftDeleteMixin
@@ -18,7 +18,14 @@ class ChatGenerationTask(Base, TimestampMixin, SoftDeleteMixin):
     __table_args__ = (
         # 防止前端按钮连点/网络重试时同一个 idempotency_key 并发创建多条任务。
         # nullable unique 兼容不传 idempotency_key 的普通请求。
-        Index("uq_chat_generation_tasks_user_mode_idempotency", "user_id", "generation_mode", "idempotency_key", unique=True),
+        Index(
+            "uq_chat_generation_tasks_user_mode_idempotency",
+            "user_id",
+            "generation_mode",
+            "idempotency_key",
+            unique=True,
+            postgresql_where=text("deleted_at IS NULL AND idempotency_key IS NOT NULL"),
+        ),
     )