Update categories/accomodation/model.py

categories/accomodation/model.py

@@ -6,24 +6,17 @@ 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")
+    
+    # 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")
+    # 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="cost of journey ticket converted from the given currency to INR")
+    # booking_platform: str = Field(..., title="The booking platform")
+    issue_date: date = Field(..., title="The date the invoice was issued")
+    summary: str = Field(..., title="5-6 words short summary of purchased good(s)")
+
+       
+