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

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"),
+        ),
     )