1

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

@@ -0,0 +1,21 @@
+from datetime import datetime
+
+from sqlalchemy import DateTime, ForeignKey, Integer, String
+from sqlalchemy.orm import Mapped, mapped_column
+
+from app.models.base import Base, TimestampMixin
+
+
+class TokenUsage(Base, TimestampMixin):
+    __tablename__ = "token_usage"
+
+    id: Mapped[str] = mapped_column(String(32), primary_key=True)
+    model_config_id: Mapped[str | None] = mapped_column(
+        String(32), ForeignKey("model_configs.id", ondelete="SET NULL"), nullable=True
+    )
+    user_id: Mapped[str | None] = mapped_column(
+        String(32), ForeignKey("users.id", ondelete="SET NULL"), nullable=True, index=True
+    )
+    input_tokens: Mapped[int] = mapped_column(Integer, default=0)
+    output_tokens: Mapped[int] = mapped_column(Integer, default=0)
+    total_tokens: Mapped[int] = mapped_column(Integer, default=0)