models.py

# models.py

from pydantic import BaseModel

class InitChatRequest(BaseModel):
    video_id: str

class ChatRequest(BaseModel):
    session_id: str
    query: str

class EndSessionRequest(BaseModel):
    session_id: str