1、增加调用 AI 视频生成能力和虚拟素材库管理的对外api

2、增加后台apikkey管理 3、增加apikey单独的模型定价 4、增加apikey调用情况 5、完善所有数据的注释增加
2026-08-06 13:13:28 +08:00
parent a55d4d649c
commit 0c511f3451
102 changed files with 13986 additions and 41 deletions
@@ -0,0 +1,61 @@
+from sqlalchemy import Float, ForeignKey, Index, Integer, String, Text, text
+from sqlalchemy.orm import Mapped, mapped_column
+
+from app.models.base import Base, TimestampMixin
+
+
+class ApiUsageLog(Base, TimestampMixin):
+    """API 调用详细消耗记录表。
+
+    记录每次 API 请求的完整消费信息，包括：
+    - 扣除金额和退回金额
+    - 模型详情（名称、分辨率、时长等）
+    - 对应的生成任务 ID
+    - 操作类型（扣除/退回）
+    """
+
+    __tablename__ = "api_usage_logs"
+    __table_args__ = (
+        Index("idx_api_usage_logs_api_key_created", "api_key_id", "created_at"),
+        Index("idx_api_usage_logs_task_id", "api_generation_task_id"),
+        Index("idx_api_usage_logs_action", "price_action"),
+    )
+
+    id: Mapped[str] = mapped_column(String(32), primary_key=True)
+    api_key_id: Mapped[str] = mapped_column(
+        String(32), ForeignKey("api_keys.id", ondelete="CASCADE"), nullable=False, index=True
+    )
+    api_generation_task_id: Mapped[str | None] = mapped_column(
+        String(32), ForeignKey("api_generation_tasks.id", ondelete="SET NULL"), nullable=True
+    )
+
+    # === 操作类型 ===
+    price_action: Mapped[str] = mapped_column(String(16), nullable=False, comment="deduct=扣除, refund=退回")
+
+    # === 请求信息 ===
+    request_type: Mapped[str] = mapped_column(String(32), nullable=False, comment="video_create|image_generate")
+    model_name: Mapped[str] = mapped_column(String(128), nullable=False)
+    gen_type: Mapped[str] = mapped_column(String(16), nullable=False)
+    resolution: Mapped[str | None] = mapped_column(String(16), nullable=True)
+    duration: Mapped[int | None] = mapped_column(Integer, nullable=True)
+
+    # === 金额信息 ===
+    credits_cost: Mapped[float] = mapped_column(Float, default=0.0, server_default="0.0", comment="实际扣除金额")
+    refund_amount: Mapped[float] = mapped_column(Float, default=0.0, server_default="0.0", comment="退回金额")
+    quota_before: Mapped[float | None] = mapped_column(Float, nullable=True, comment="操作前配额余额")
+    quota_after: Mapped[float | None] = mapped_column(Float, nullable=True, comment="操作后配额余额")
+
+    # === Token 用量 ===
+    tokens_used: Mapped[int] = mapped_column(Integer, default=0, server_default="0")
+    request_duration_ms: Mapped[int] = mapped_column(Integer, default=0, server_default="0", comment="端到端耗时")
+
+    # === 价格明细（JSON） ===
+    price_detail_json: Mapped[str | None] = mapped_column(Text, nullable=True, comment="价格计算明细JSON")
+
+    # === 结果 ===
+    status: Mapped[str] = mapped_column(String(32), nullable=False, comment="success|failed")
+    error_message: Mapped[str | None] = mapped_column(Text, nullable=True)
+    error_code: Mapped[str | None] = mapped_column(String(64), nullable=True)
+
+    # === 调试 ===
+    request_payload_json: Mapped[str | None] = mapped_column(Text, nullable=True, comment="原始请求快照")