from datetime import datetime

from pydantic import BaseModel, Field


class TenantCreate(BaseModel):
    name: str = Field(..., max_length=200)
    slug: str = Field(..., max_length=100)


class TenantUpdate(BaseModel):
    name: str | None = Field(None, max_length=200)
    slug: str | None = Field(None, max_length=100)


class TenantResponse(BaseModel):
    id: str
    name: str
    slug: str
    status: str
    created_at: datetime
    updated_at: datetime


class TenantConfigCreate(BaseModel):
    llm_provider: str = Field(..., max_length=50)
    llm_api_key: str
    llm_model: str = Field(..., max_length=100)
    llm_base_url: str | None = Field(None, max_length=500)
    max_tokens_per_month: int | None = None


class TenantConfigResponse(BaseModel):
    id: str
    tenant_id: str
    llm_provider: str
    llm_model: str
    llm_base_url: str | None
    max_tokens_per_month: int
    created_at: datetime
    updated_at: datetime