from pydantic import BaseModel
from typing import List, Optional

class ChatRequest(BaseModel):
    user_id: str
    stage: str
    message: str
    history: Optional[List[List[str]]] = []

class ChatResponse(BaseModel):
    response_text: str
    source_chunks: List[str]
    debug: Optional[dict] = None