Upload tool

app.py

@@ -0,0 +1,6 @@
+from smolagents import launch_gradio_demo
+from tool import SimpleTool
+
+tool = SimpleTool()
+
+launch_gradio_demo(tool)

requirements.txt

@@ -0,0 +1,5 @@
+agents_package
+os
+googlemaps
+smolagents
+datetime

tool.py

@@ -0,0 +1,32 @@
+from smolagents import Tool
+
+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"}}
+    output_type = "string"
+
+    def forward(self, start_location: str, destination_location: str, departure_time: Optional[int] = None) -> str:
+        """Gets the travel time in car between two places.
+    
+        Args:
+            start_location: the place from which you start your ride
+            destination_location: the place of arrival
+            departure_time: the departure time, provide only a `datetime.datetime` if you want to specify this
+        """
+        import googlemaps
+        import os
+
+        gmaps = googlemaps.Client(os.getenv("GMAPS_API_KEY"))
+
+        if departure_time is None:
+            from datetime import datetime
+            monday = datetime(2025, 1, 6, 11, 0)
+
+        directions_result = gmaps.directions(
+            start_location,
+            destination_location,
+            mode="transit",
+            departure_time=departure_time
+        )
+        return directions_result[0]["legs"][0]["duration"]["text"]