categories/accomodation/model.py

from __future__ import annotations

from datetime import datetime

from pydantic import BaseModel, Field


class InformationExtractedFromABillReceipt(BaseModel):
    """
    1. Hotel Name: [Hotel Name]
    2. Address: [Hotel Address]
    3. Bill number/Invoice number: [Bill Number]
    4. booking ID / Confirmation ID / Booking #: [Booking ID]
    5. Check-in Date and Time: [Check-in Date Time]
    6. Check-out Date and Time: [Check-out Date Time]
    7. Total Amount: [Total Amount Charged]
    8. Booking platform: [Booking Platform]
    9. Bill date: [Bill Date]
    """

    hotel_name: str = Field(..., title="The name of the hotel")
    address: str = Field(..., title="The address of the hotel")
    uids: str = Field(..., title="The bill number/invoice number")
    booking_id: str = Field(..., title="The booking ID/confirmation ID/booking number")
    check_in_date_time: datetime = Field(..., title="The check-in date and time")
    check_out_date_time: datetime = Field(..., title="The check-out date and time")
    total: float = Field(..., title="The total amount charged")
    booking_platform: str = Field(..., title="The booking platform")
    bill_date: datetime = Field(..., title="The bill date")