from pydantic import BaseModel
from typing import Optional


class SaveChatLogRequest(BaseModel):
    session_id: str
    log: str
    sender: str
    daily: Optional[int] = None
    category: str
    chat_session_id: Optional[int] = None


class GetChatLogQuery(BaseModel):
    category: Optional[str] = None
    daily: Optional[int] = None


class CreateSessionRequest(BaseModel):
    session_id: str
    category: str
    title: Optional[str] = None
    daily: Optional[int] = None


class UpdateSessionTitleRequest(BaseModel):
    title: str