Removing class structure

tools/cheapest_flight.py

@@ -1,109 +1,79 @@
-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']}
-        """
-        )
+import http.client
+import pandas as pd
+import numpy as np
+import json
+from smolagents.tools import Tool
+
+
+
+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