m-ric HF staff commited on
Commit
20cb02f
·
verified ·
1 Parent(s): 4da9eaa

Update tool.py

Browse files
Files changed (1) hide show
  1. tool.py +3 -3
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"},"transportation":{"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: 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: The transportation mode, in 'driving', 'walking', 'bicycling', or 'transit'.
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