Spaces:
Running
Running
Update tool.py
Browse files
tool.py
CHANGED
|
@@ -4,16 +4,16 @@ from typing import Optional
|
|
| 4 |
class SimpleTool(Tool):
|
| 5 |
name = "get_travel_duration"
|
| 6 |
description = "Gets the travel time in car between two places."
|
| 7 |
-
inputs = {"start_location":{"type":"string","description":"the place from which you start your ride"},"destination_location":{"type":"string","description":"the place of arrival"},"
|
| 8 |
output_type = "string"
|
| 9 |
|
| 10 |
-
def forward(self, start_location: str, destination_location: str,
|
| 11 |
"""Gets the travel time in car between two places.
|
| 12 |
|
| 13 |
Args:
|
| 14 |
start_location: the place from which you start your ride
|
| 15 |
destination_location: the place of arrival
|
| 16 |
-
|
| 17 |
"""
|
| 18 |
import googlemaps
|
| 19 |
import os
|
|
|
|
| 4 |
class SimpleTool(Tool):
|
| 5 |
name = "get_travel_duration"
|
| 6 |
description = "Gets the travel time in car between two places."
|
| 7 |
+
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'.}}
|
| 8 |
output_type = "string"
|
| 9 |
|
| 10 |
+
def forward(self, start_location: str, destination_location: str, transportation_mode: Optional[string] = None) -> str:
|
| 11 |
"""Gets the travel time in car between two places.
|
| 12 |
|
| 13 |
Args:
|
| 14 |
start_location: the place from which you start your ride
|
| 15 |
destination_location: the place of arrival
|
| 16 |
+
transportation_mode: The transportation mode, in 'driving', 'walking', 'bicycling', or 'transit'.
|
| 17 |
"""
|
| 18 |
import googlemaps
|
| 19 |
import os
|