from pydantic import BaseModel

class UsageStats(BaseModel):
    total_runs: int
    total_tokens: int
    total_input_tokens: int
    total_output_tokens: int
    total_cost: float

class UsageByEntity(BaseModel):
    entity_id: str | None = None
    stats: UsageStats

class FinancialReport(BaseModel):
    summary: UsageStats
    by_user: list[UsageByEntity] | None = None
    by_project: list[UsageByEntity] | None = None