Spaces:
Running
Running
from smolagents import Tool | |
from typing import Optional | |
class SimpleTool(Tool): | |
name = "get_travel_duration" | |
description = "Gets the travel time between two places." | |
inputs = {"start_location":{"type":"string","description":"the place from which you start your ride"},"destination_location":{"type":"string","description":"the place of arrival"},"transportation_mode":{"type":"string","nullable":True,"description":"The transportation mode, in 'driving', 'walking', 'bicycling', or 'transit'. Defaults to 'driving'."}} | |
output_type = "string" | |
def forward(self, start_location: str, destination_location: str, transportation_mode: Optional[str] = None) -> str: | |
"""Gets the travel time between two places. | |
Args: | |
start_location: the place from which you start your ride | |
destination_location: the place of arrival | |
transportation_mode: The transportation mode, in 'driving', 'walking', 'bicycling', or 'transit'. Defaults to 'driving'. | |
""" | |
import os # All imports are placed within the function, to allow for sharing to Hub. | |
import googlemaps | |
from datetime import datetime | |
gmaps = googlemaps.Client(os.getenv("GMAPS_API_KEY")) | |
if transportation_mode is None: | |
transportation_mode = "driving" | |
try: | |
directions_result = gmaps.directions( | |
start_location, | |
destination_location, | |
mode=transportation_mode, | |
departure_time=datetime(2025, 12, 6, 11, 0), # At 11, date far in the future | |
) | |
if len(directions_result) == 0: | |
return "No way found between these places with the required transportation mode." | |
return directions_result[0]["legs"][0]["duration"]["text"] | |
except Exception as e: | |
print(e) | |
return e |