Upload cheapest_flight.py

tools/cheapest_flight.py

@@ -0,0 +1,109 @@
+import http.client
+import pandas as pd
+import numpy as np
+import json
+from smolagents.tools import Tool
+
+class GetCheapestFlightTool(Tool):
+
+    name = "cheapest_flight_tool"
+    description = "Return the cheapest flight"
+    inputs = {
+        'departureId': {'type': 'string', 'description': 'The ID in which the airport of the departure location is.'},
+        'arrivalId': {'type': 'string', 'description': 'The ID in which the airport of the arrival location is.'},
+        'departureDate': {'type': 'string', 'description': 'the date to look for the cheapest flight.'},  }
+    output_type = "dict"
+
+    def get_flight(departureId, arrivalId, departureDate):
+
+        conn = http.client.HTTPSConnection("booking-com15.p.rapidapi.com")
+
+        headers = {
+            'x-rapidapi-key': "RAPIDAPI_KEY", #Replace with your own RapidAPI key
+            'x-rapidapi-host': "booking-com15.p.rapidapi.com"
+        }
+
+        conn.request("GET", f"/api/v1/flights/searchFlights?fromId={departureId}.AIRPORT&toId={arrivalId}.AIRPORT&departDate={departureDate}&currency_code=BRL", headers=headers)
+
+        res = conn.getresponse()
+        data = res.read()
+        json_data = json.loads(data.decode("utf-8"))
+
+        try:
+            # Get all flight offers with validation
+            if not json_data or 'data' not in json_data or 'flightOffers' not in json_data['data']:
+                return None
+            flight_offers = json_data['data']['flightOffers']
+
+            if not flight_offers:
+                return None
+
+            # Initialize variables to track cheapest flight
+            cheapest_price = float('inf')
+            cheapest_flight = None
+
+            # Loop through each flight offer
+            for offer in flight_offers:
+                try:
+                    # Validate price breakdown structure
+                    if ('priceBreakdown' not in offer or 
+                        'total' not in offer['priceBreakdown'] or 
+                        'units' not in offer['priceBreakdown']['total']):
+                        print('a estrutura está correta')
+                        continue
+
+                    # Get total price in units
+                    price = float(offer['priceBreakdown']['total']['units'])
+
+                    # Add nanos (decimal part) if present
+                    if 'nanos' in offer['priceBreakdown']['total']:
+                        price += float(offer['priceBreakdown']['total']['nanos']) / 1000000000
+
+                    # Update cheapest if current price is lower
+                    if price < cheapest_price:
+                        cheapest_price = price
+                        cheapest_flight = offer
+
+                except (ValueError, TypeError):
+                    continue
+
+            # Extract relevant info from cheapest flight
+            if cheapest_flight:
+                try:
+                    result = {
+                        'price': f"{cheapest_flight['priceBreakdown']['total'].get('currencyCode', 'USD')} {cheapest_price:.2f}",
+                        'departure': {
+                            'from': cheapest_flight['segments'][0]['departureAirport'].get('cityName', 'Unknown'),
+                            'time': cheapest_flight['segments'][0].get('departureTime', 'Unknown')
+                        },
+                        'arrival': {
+                            'to': cheapest_flight['segments'][0]['arrivalAirport'].get('cityName', 'Unknown'),
+                            'time': cheapest_flight['segments'][0].get('arrivalTime', 'Unknown')
+                        },
+                        'airline': cheapest_flight['segments'][0]['legs'][0]['carriersData'][0].get('name', 'Unknown')
+                    }
+                    return result
+                except (KeyError, IndexError):
+                    return None
+            return None
+
+        except Exception:
+            return None
+
+if __name__ == "__main__":
+
+    tool = GetCheapestFlightTool()
+
+    departureId = "BOM"
+    arrivalId = "DEL"
+    departureDate = '2025-03-16'
+
+    cheapest = tool.get_flight(departureId, arrivalId, departureDate)
+    print(f"""
+        Cheapest Flight Details:
+        Price: {cheapest['price']}
+        From: {cheapest['departure']['from']} at {cheapest['departure']['time']}
+        To: {cheapest['arrival']['to']} at {cheapest['arrival']['time']}
+        Airline: {cheapest['airline']}
+        """
+        )