22 lines
829 B
Python
22 lines
829 B
Python
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)
|