a bg@s~ddlZddlmZddlmZmZmZddlmZddl m Z m Z ddl m Z eeZGddde ZGd d d e ZdS) N)datetime)DictOptionalType)CallbackManagerForToolRun) BaseModelField)AmadeusBaseToolc@sneZdZUdZeddZeed<eddZeed<eddZ eed<ed dZ eed <ed d d Z e ed<dS)FlightSearchSchemaz(Schema for the AmadeusFlightSearch tool.zx The three letter International Air Transport Association (IATA) Location Identifier for the search's origin airport. ) descriptionoriginLocationCodez} The three letter International Air Transport Association (IATA) Location Identifier for the search's destination airport. destinationLocationCodea The earliest departure datetime from the origin airport for the flight search in the following format: "YYYY-MM-DDTHH:MM:SS", where "T" separates the date and time components. For example: "2023-06-09T10:30:00" represents June 9th, 2023, at 10:30 AM. departureDateTimeEarliestz The latest departure datetime from the origin airport for the flight search in the following format: "YYYY-MM-DDTHH:MM:SS", where "T" separates the date and time components. For example: "2023-06-09T10:30:00" represents June 9th, 2023, at 10:30 AM. departureDateTimeLatestz6The specific page number of flight results to retrieve)defaultr page_numberN) __name__ __module__ __qualname____doc__rr str__annotations__r rrrintrr}/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/tools/amadeus/flight_search.pyr s"   r c @sZeZdZUdZdZeed<dZeed<eZ e eed<d eeeee e e ed d d ZdS) AmadeusFlightSearchz}ztd|WYd}~n d}~00d} |j} t|d} t|d} | | krtddgSd} z | j j j ||| ddd } Wn,|y}zt |WYd}~n d}~00g}| dur| jD]}i}i|d <|d d |d d <|d d }| jd d|}i|d d <||d d <g}|dddD]Z}i}|d|d<|d|d<|d|d<|d}| jd d|}||d<||qVg|d<||d<||qt|D]:\}}t|ddddd}|| kr||q|d| }|| }|||S)Nr) ResponseErrorzDUnable to import amadeus, please install with `pip install amadeus`. z%Y-%m-%dT%H:%M:%Sz Error: Earliest and latest departure dates need to be the same date. If you're trying to search for round-trip flights, call this function for the outbound flight first, and then call again for the return flight. z%Y-%m-%dr)r r Z departureDateZadultsZpricetotalcurrencyZ dictionariesZ currenciesZ itinerariessegmentsZ departureZarrivalnumberZ flightNumberZ carrierCodeZcarrierscarrierat)Zamadeusr! ImportErrorclientdtstrptimedateloggererrorZshoppingZflight_offers_searchgetstrftimeprintdataresultappend enumeratepop)selfr r rrrrr!eZRESULTS_PER_PAGEr*ZearliestDepartureZlatestDepartureresponser/outputZofferZ itineraryr$r%segmentZflightr'indexZofferDepartureZ startIndexZendIndexrrr_runAsv               zAmadeusFlightSearch._run)rN)rrrrrrrr r rrrrrlistr>rrrrr6s   r)loggingrr+typingrrrZlangchain_core.callbacksrZpydanticrrZ&langchain_community.tools.amadeus.baser getLoggerrr.r rrrrrs    )