Update tool.py

tool.py

@@ -4,10 +4,10 @@ from typing import Optional
 class SimpleTool(Tool):
     name = "get_travel_duration"
     description = "Gets the travel time in car 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"},"departure_time":{"type":"integer","nullable":True,"description":"the departure time, provide only a `datetime.datetime` if you want to specify this"}}
+    inputs = {"start_location":{"type":"string","description":"the place from which you start your ride"},"destination_location":{"type":"string","description":"the place of arrival"},"transportation":{"type":"string","nullable":True,"description":"The transportation type, in 'driving', 'walking', 'bicycling', or 'transit'.}}
     output_type = "string"
 
-    def forward(self, start_location: str, destination_location: str, departure_time: Optional[int] = None) -> str:
+    def forward(self, start_location: str, destination_location: str, transportation: Optional[string] = None) -> str:
         """Gets the travel time in car between two places.
     
         Args:
@@ -20,15 +20,14 @@ class SimpleTool(Tool):
 
         gmaps = googlemaps.Client(os.getenv("GMAPS_API_KEY"))
 
-        if departure_time is None:
-            from datetime import datetime
-            monday = datetime(2025, 1, 6, 11, 0)
+        if transportation is None:
+            transportation = "transit"
         try:
             directions_result = gmaps.directions(
                 start_location,
                 destination_location,
-                mode="transit",
-                departure_time=departure_time
+                mode=transportation,
+                departure_time=datetime(2025, 1, 6, 11, 0), # Start on a Monday morning
             )
             return directions_result[0]["legs"][0]["duration"]["text"]
         except Exception as e: