from typing import Optional

from pydantic import BaseModel


class QuotaCheckResponse(BaseModel):
    has_quota: bool
    remaining_conversations: Optional[int] = None
    remaining_chapters: Optional[int] = None
    remaining_words: Optional[int] = None
    used_conversations: Optional[int] = None
    used_chapters: Optional[int] = None
    max_conversations: Optional[int] = None
    max_chapters: Optional[int] = None
    message: str