69a8829750
P0 Critical: presentation isolation (client scoping), storage super_admin fix, template selection in worker, IMAGE_PROVIDERS list fix. P1 High: template layout management UI (delete/filter/bulk), slide-based parsing mode, LLM model listing & connection test, settings persistence to DB (Fernet encryption), logout button. P2 Polish: storage improvements (master deck files, per-client breakdown, bulk delete, hard purge, client selector), image generation error visibility (__image_error__ badge), hamster wheel loading animation. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
37 lines
1.6 KiB
Python
37 lines
1.6 KiB
Python
"""AI usage tracking model for analytics."""
|
|
import uuid
|
|
from datetime import datetime
|
|
from typing import Optional
|
|
|
|
from sqlalchemy import Column, DateTime, ForeignKey, Integer, String
|
|
from sqlmodel import Field, SQLModel
|
|
|
|
from utils.datetime_utils import get_current_utc_datetime
|
|
|
|
|
|
class AIUsageModel(SQLModel, table=True):
|
|
__tablename__ = "ai_usage_logs"
|
|
|
|
id: uuid.UUID = Field(primary_key=True, default_factory=uuid.uuid4)
|
|
user_id: Optional[uuid.UUID] = Field(
|
|
sa_column=Column(ForeignKey("users.id"), nullable=True), default=None
|
|
)
|
|
client_id: Optional[uuid.UUID] = Field(
|
|
sa_column=Column(ForeignKey("clients.id"), nullable=True), default=None
|
|
)
|
|
presentation_id: Optional[uuid.UUID] = Field(
|
|
sa_column=Column(ForeignKey("presentations.id"), nullable=True), default=None
|
|
)
|
|
provider: str = Field(sa_column=Column(String, nullable=False))
|
|
model: str = Field(sa_column=Column(String, nullable=False))
|
|
call_type: str = Field(sa_column=Column(String, nullable=False))
|
|
input_tokens: Optional[int] = Field(sa_column=Column(Integer, nullable=True), default=None)
|
|
output_tokens: Optional[int] = Field(sa_column=Column(Integer, nullable=True), default=None)
|
|
total_tokens: Optional[int] = Field(sa_column=Column(Integer, nullable=True), default=None)
|
|
duration_ms: Optional[int] = Field(sa_column=Column(Integer, nullable=True), default=None)
|
|
error_details: Optional[str] = Field(sa_column=Column(String, nullable=True), default=None)
|
|
created_at: datetime = Field(
|
|
sa_column=Column(
|
|
DateTime(timezone=True), nullable=False, default=get_current_utc_datetime
|
|
),
|
|
)
|