Update tools/flight_status.py

tools/flight_status.py

@@ -5,10 +5,7 @@ import requests
 class FlightStatus(Tool):
     name = "flight_status"
     description = "Fetches the status of a given flight based on the flight number using AviationStack API."
-    inputs = {
-        'flight': {'type': 'string', 'description': 'Flight Code to fetch the information.'},
-        'api_key': {'type': 'string', 'description': 'The API key for accessing the AviationStack data'}
-    }
+    inputs = {'flight': {'type': 'string', 'description': 'Flight Code to fetch the information.'}}
     output_type = {'flight_info': {'type': 'dict', 'description': 'A dictionary containing the flight status information or an error message'}}
 
     def forward(self) -> dict:
@@ -17,8 +14,7 @@ class FlightStatus(Tool):
         Returns a dictionary containing flight status or error details.
         """
         flight_number = self.inputs.get('flight')
-        #api_key = self.inputs.get('api_key')
-        api_key = ""
+        api_key = "6ffd5cb6e366ed0cf24a1caa38b17c31"
 
         # Ensure flight number and API key are provided
         if not flight_number or not api_key:
@@ -34,20 +30,17 @@ class FlightStatus(Tool):
 
         try:
             # Send a GET request to the AviationStack API
-            #response = requests.get(base_url, params=params)
-            response = 200
+            response = requests.get(base_url, params=params)
             
-            # Check if the request was successful
-            # if response.status_code == 200:
-            #     data = response.json()
+            Check if the request was successful
+            if response.status_code == 200:
+                data = response.json()
 
-            #     # If no flight data is found
-            #     if not data.get("data"):
-            #         return {'error': f"No information available for flight {flight_number}."}
+                # If no flight data is found
+                if not data.get("data"):
+                    return {'error': f"No information available for flight {flight_number}."}
 
-            #     flight_data = data['data'][0]  # Get the first flight result
-            if response == 200:
-                flight_data = {'flight_date': '2025-02-16', 'flight_status': 'scheduled', 'departure': {'airport': 'John F Kennedy International', 'timezone': 'America/New_York', 'iata': 'JFK', 'icao': 'KJFK', 'terminal': '8', 'gate': '4', 'delay': None, 'scheduled': '2025-02-16T19:00:00+00:00', 'estimated': '2025-02-16T19:00:00+00:00', 'actual': None, 'estimated_runway': None, 'actual_runway': None}, 'arrival': {'airport': 'Heathrow', 'timezone': 'Europe/London', 'iata': 'LHR', 'icao': 'EGLL', 'terminal': '3', 'gate': None, 'baggage': None, 'delay': None, 'scheduled': '2025-02-17T06:50:00+00:00', 'estimated': '2025-02-17T06:50:00+00:00', 'actual': None, 'estimated_runway': None, 'actual_runway': None}, 'airline': {'name': 'American Airlines', 'iata': 'AA', 'icao': 'AAL'}, 'flight': {'number': '100', 'iata': 'AA100', 'icao': 'AAL100', 'codeshared': None}, 'aircraft': None, 'live': None}
+                flight_data = data['data'][0]  # Get the first flight result
 
                 # Extract relevant flight information
                 flight_info = {