# api/models/schemas.py
from pydantic import BaseModel
from typing import Optional, Dict


class ChatRequest(BaseModel):
    message: str
    session_id: Optional[str] = None
    stream: bool = False


class ApprovalRequest(BaseModel):
    decision: str  # "approve" or "reject"
    reason: Optional[str] = None


class ChatResponse(BaseModel):
    response: str
    session_id: str
    has_interrupt: bool = False
    interrupt_type: Optional[str] = None
    interrupt_data: Optional[Dict] = None