models/responses.py

from typing import List, Optional
from pydantic import BaseModel, Field

class HotelImage(BaseModel):
    url: str

class HotelAmenity(BaseModel):
    name: str

class HotelData(BaseModel):
    name: str
    rating: Optional[float] = None
    images: List[str] = []
    amenities: List[str] = []
    booking_link: Optional[str] = None

class HotelResponse(BaseModel):
    destination: str
    hotel_name: str
    data: Optional[HotelData] = None
    error: Optional[str] = None

class HotelSearchResponse(BaseModel):
    results: List[HotelResponse]
    status: str = "success"
    count: int