Spaces:
Runtime error
Runtime error
File size: 4,220 Bytes
7f951c8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 |
import os
from typing import Optional
from smolagents.tools import Tool
import serpapi
from dotenv import load_dotenv
load_dotenv() # Loads variables from .env into environment
class FlightsFinderTool(Tool):
name = "flights_finder"
description = "Find flights using the Google Flights engine."
inputs = {
'departure_airport': {'type': 'string', 'description': 'Departure airport code (IATA)'},
'arrival_airport': {'type': 'string', 'description': 'Arrival airport code (IATA)'},
'outbound_date': {'type': 'string', 'description': 'Outbound date in YYYY-MM-DD format'},
'return_date': {'type': 'string', 'description': 'Return date in YYYY-MM-DD format'},
'adults': {'type': 'integer', 'default': 1, 'nullable': True, 'description': 'Number of adults'},
'children': {'type': 'integer', 'default': 0, 'nullable': True, 'description': 'Number of children'},
}
output_type = "string"
@staticmethod
def find_flight(
departure_airport: Optional[str] = None,
arrival_airport: Optional[str] = None,
date: Optional[str] = None,
adults: Optional[int] = 1,
children: Optional[int] = 0,
) -> str:
"""
Finds the cheapest one-way flight for a given route and date.
Args:
departure_airport (str): Departure airport code (IATA)
arrival_airport (str): Arrival airport code (IATA)
date (str): Flight date in YYYY-MM-DD format
adults (int): Number of adults
children (int): Number of children
infants_in_seat (int): Number of infants in seat
infants_on_lap (int): Number of lap infants
Returns:
str: Formatted string with cheapest flight details
"""
params = {
'api_key': os.getenv("SERPAPI_API_KEY"),
'engine': 'google_flights',
'hl': 'en',
'gl': 'us',
'departure_id': departure_airport,
'arrival_id': arrival_airport,
'outbound_date': date,
'currency': 'USD',
'adults': adults,
'children': children,
'type': 2,
}
try:
search = serpapi.search(params)
flights = search.data.get("best_flights", [])
if not flights:
return "No flights found."
# Find the flight with the lowest price
cheapest = min(flights, key=lambda f: f.get("price", float("inf")))
if not cheapest.get("flights"):
return "No flight segments found."
flight = cheapest["flights"][0]
dep = flight["departure_airport"]
arr = flight["arrival_airport"]
dep_time = dep["time"]
arr_time = arr["time"]
duration = flight["duration"]
airline = flight.get("airline", "Unknown")
price = cheapest["price"]
hours = duration // 60
minutes = duration % 60
duration_str = f"{hours}h {minutes}m"
return (
f"From {dep['id']} at {dep_time} → {arr['id']} at {arr_time} | "
f"Duration: {duration_str}\nAirline: {airline} | Price: ${price}"
)
except Exception as e:
return f"Error occurred: {e}"
def forward(
self,
departure_airport: str,
arrival_airport: str,
outbound_date: str,
return_date: str,
adults: int = 1,
children: int = 0,
) -> str:
outbound = self.find_flight(
departure_airport=departure_airport,
arrival_airport=arrival_airport,
date=outbound_date,
adults=adults,
children=children,
)
inbound = self.find_flight(
departure_airport=arrival_airport,
arrival_airport=departure_airport,
date=return_date,
adults=adults,
children=children,
)
return f"✈️ Outbound Flight:\n{outbound}\n\n🛬 Inbound Flight:\n{inbound}"
def __init__(self, *args, **kwargs):
self.is_initialized = False
|