qq1023 commited on
Commit
50b87bc
1 Parent(s): af746d0

Update categories/travel_flight/model.py

Browse files
Files changed (1) hide show
  1. categories/travel_flight/model.py +3 -1
categories/travel_flight/model.py CHANGED
@@ -16,7 +16,8 @@ class InformationExtractedFromABillReceipt(BaseModel):
16
  ResponseSchema(name="time (to)", description="time at which flight end/lands"),
17
  ResponseSchema(name="PNR Number", description ="PNR Number of flight"),
18
  ResponseSchema(name="Amount", description="cost of flight ticket"),
19
- ResponseSchema(name="Billing Date", description="The date the invoice was issued")
 
20
  ))
21
  ]"""
22
 
@@ -31,3 +32,4 @@ class InformationExtractedFromABillReceipt(BaseModel):
31
  uids: str = Field(..., title="PNR Number of flight")
32
  total: float = Field(..., title="cost of flight ticket")
33
  issue_date: date = Field(..., title="The date the invoice was issued")
 
 
16
  ResponseSchema(name="time (to)", description="time at which flight end/lands"),
17
  ResponseSchema(name="PNR Number", description ="PNR Number of flight"),
18
  ResponseSchema(name="Amount", description="cost of flight ticket"),
19
+ ResponseSchema(name="Billing Date", description="The date the invoice was issued"),
20
+ ResponseSchema(name="Summary", description = "5-6 words short summary of purchased good(s)"),
21
  ))
22
  ]"""
23
 
 
32
  uids: str = Field(..., title="PNR Number of flight")
33
  total: float = Field(..., title="cost of flight ticket")
34
  issue_date: date = Field(..., title="The date the invoice was issued")
35
+ summary: str = Field(..., title="5-6 words short summary of purchased good(s)")