生成任务失败积分回退

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

@@ -1,6 +1,6 @@
 from datetime import datetime
 
-from sqlalchemy import DateTime, Float, ForeignKey, Integer, String, Text
+from sqlalchemy import DateTime, Float, ForeignKey, Index, Integer, String, Text
 from sqlalchemy.orm import Mapped, mapped_column
 
 from app.models.base import Base, TimestampMixin, SoftDeleteMixin
@@ -15,6 +15,12 @@ class ChatGenerationTask(Base, TimestampMixin, SoftDeleteMixin):
     """
 
     __tablename__ = "chat_generation_tasks"
+    __table_args__ = (
+        # 防止前端按钮连点/网络重试时同一个 idempotency_key 并发创建多条任务。
+        # nullable unique 兼容不传 idempotency_key 的普通请求。
+        Index("uq_chat_generation_tasks_user_mode_idempotency", "user_id", "generation_mode", "idempotency_key", unique=True),
+    )
+
 
     id: Mapped[str] = mapped_column(String(32), primary_key=True)
     user_id: Mapped[str] = mapped_column(