Update categories/travel_flight/model.py

categories/travel_flight/model.py

@@ -6,30 +6,17 @@ from pydantic import BaseModel, Field
 
 
 class InformationExtractedFromABillReceipt(BaseModel):
-    """
-    response_schemas = [
-        ResponseSchema(name="place (from)", description="place where flight starts/takes-off"),
-        ResponseSchema(name="date (from)", description="date on which flight starts/takes-off (DD/MM/YYYY)"),
-        ResponseSchema(name="time (from)", description="time at which flight starts/takes-off"),
-        ResponseSchema(name="place (to)", description="place where flight end/lands"),
-        ResponseSchema(name="date (to)", description="date on which flight end/lands (DD/MM/YYYY)"),
-        ResponseSchema(name="time (to)", description="time at which flight end/lands"),
-        ResponseSchema(name="PNR Number", description ="PNR Number of flight"),
-        ResponseSchema(name="Amount", description="cost of flight ticket"),
-        ResponseSchema(name="Billing Date", description="The date the invoice was issued"),
-        ResponseSchema(name="Summary", description = "5-6 words short summary of purchased good(s)"),
-    ))
-    ]"""
+    ''''''
 
-    place_from: str = Field(..., title="place where flight starts/takes-off")
+    place_from: str = Field(..., title="place where journey starts")
     date_from: date = Field(
-        ..., title="date on which flight starts/takes-off (DD/MM/YYYY)"
+        ..., title="date on which journey starts (DD/MM/YYYY)"
     )
-    time_from: time = Field(..., title="time at which flight starts/takes-off")
-    place_to: str = Field(..., title="place where flight end/lands")
-    date_to: date = Field(..., title="date on which flight end/lands (DD/MM/YYYY)")
-    time_to: time = Field(..., title="time at which flight end/lands")
-    uids: str = Field(..., title="PNR Number of flight")
-    total: float = Field(..., title="cost of flight ticket")
+    time_from: time = Field(..., title="time at which journey starts")
+    place_to: str = Field(..., title="place where journey end")
+    date_to: date = Field(..., title="date on which journey end (DD/MM/YYYY)")
+    time_to: time = Field(..., title="time at which journey end")
+    uids: str = Field(..., title="The bill number/invoice number")
+    total: float = Field(..., title="cost of journey ticket")
     issue_date: date = Field(..., title="The date the invoice was issued")
     summary: str = Field(..., title="5-6 words short summary of purchased good(s)")