from __future__ import annotations

from decimal import Decimal

from sqlalchemy import Boolean, Index, Numeric, String
from sqlalchemy.orm import Mapped, mapped_column

from app.models.base import Base, TimestampMixin


class LlmBillingPolicyModel(Base, TimestampMixin):
    __tablename__ = "llm_billing_policies"
    __table_args__ = (Index("uq_llm_billing_policies_scene", "scene_code", unique=True),)

    id: Mapped[str] = mapped_column(String(32), primary_key=True)
    scene_code: Mapped[str] = mapped_column(String(64), nullable=False)
    scene_name: Mapped[str] = mapped_column(String(128), nullable=False)
    pre_deduct_credits: Mapped[Decimal] = mapped_column(Numeric(20, 2), nullable=False)
    is_active: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True, server_default="true")
    version: Mapped[int] = mapped_column(nullable=False, default=1, server_default="1")
    created_by: Mapped[str | None] = mapped_column(String(32), nullable=True)
    updated_by: Mapped[str | None] = mapped_column(String(32), nullable=True)