ds_drone_bot/api/models/requests.py

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

class BaseRequest(BaseModel):
    pass

class ChatMessage(BaseModel):
    role: str  # "human" or "ai"
    content: str


class ChatRequest(BaseModel):
    query: str
    history: List[ChatMessage] = []
    customer_metadata: Dict = None  # Customer info: name, email, phone, user_id


class SurveyRequest(BaseModel):
    booking_form_input: Dict