Spaces:
Build error
Build error
| from smolagents import Tool | |
| from typing import Any, 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 openrouteservice as ors | |
| from openrouteservice.geocode import pelias_search | |
| import json | |
| # Initialize the client | |
| client = ors.Client(key=os.getenv("ORS_API_KEY")) | |
| if transportation_mode is None: | |
| transportation_mode = "driving-car" | |
| elif transportation_mode == "driving": | |
| transportation_mode = "driving-car" | |
| elif transportation_mode == "walking": | |
| transportation_mode = "foot-walking" | |
| elif transportation_mode == "bicycling": | |
| transportation_mode = "cycling-regular" | |
| elif transportation_mode == "transit": | |
| return "Transit mode is not supported by OpenRouteService. Please choose driving, walking, or bicycling." | |
| try: | |
| # Geocode start location | |
| start_coords = pelias_search(client, start_location) | |
| if not start_coords or not start_coords['features']: | |
| return f"Could not find coordinates for {start_location}" | |
| start_point = start_coords['features'][0]['geometry']['coordinates'] | |
| # Geocode destination location | |
| dest_coords = pelias_search(client, destination_location) | |
| if not dest_coords or not dest_coords['features']: | |
| return f"Could not find coordinates for {destination_location}" | |
| dest_point = dest_coords['features'][0]['geometry']['coordinates'] | |
| # Get directions | |
| routes = client.directions( | |
| coordinates=[start_point, dest_point], | |
| profile=transportation_mode, | |
| format='geojson' | |
| ) | |
| if not routes or 'features' not in routes or not routes['features']: | |
| return "No route found between these places with the required transportation mode." | |
| # Extract duration in seconds and convert to human-readable format | |
| duration_seconds = routes['features'][0]['properties']['segments'][0]['duration'] | |
| hours = int(duration_seconds // 3600) | |
| minutes = int((duration_seconds % 3600) // 60) | |
| if hours > 0: | |
| return f"{hours} hour{'s' if hours != 1 else ''} {minutes} min" | |
| else: | |
| return f"{minutes} min" | |
| except Exception as e: | |
| print(e) | |
| return str(e) |