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
|