models.py

from pydantic import BaseModel
from typing import List, Optional

class FlightRequest(BaseModel):
    origin: str
    destination: str
    departure_date: str
    return_date: Optional[str]
    adults: int
    children: Optional[int]
    max_results: Optional[int] = 10

class HotelRequest(BaseModel):
    city_code: str
    ratings: List[str] = ["3", "4", "5"]