phspoden commited on
Commit
7570cb2
·
1 Parent(s): 1229e72

Fixed issue in get_random_name tool and added parameter 'nationality'.

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -27,11 +27,11 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
27
  def get_random_name(nationality: str) -> str:
28
  """A tool that retrieves a random name
29
  Args:
30
- nationality: A string representing a valid nationality (e.g., 'Mexican').
31
  """
32
  try:
33
  url = 'https://api.fbi.gov/wanted/v1/list'
34
- params = {'field_offices': 'philadelphia'} # todo make params a parameter
35
  response = requests.get(url, params= params)
36
 
37
  if response.status_code == 200:
@@ -40,7 +40,7 @@ def get_random_name(nationality: str) -> str:
40
  nr_of_items = len(data['items'])
41
  random_int = random.randint(0,nr_of_items-1)
42
  # get the title and return it
43
- return data['items'][str(random_int)]["title"]
44
  else:
45
  return f'Error - get_random_name() -> Response status code was not good ({response.status_code})'
46
 
 
27
  def get_random_name(nationality: str) -> str:
28
  """A tool that retrieves a random name
29
  Args:
30
+ nationality: A string representing a valid nationality (e.g., 'Mexican' or 'American').
31
  """
32
  try:
33
  url = 'https://api.fbi.gov/wanted/v1/list'
34
+ params = {'race': nationality}
35
  response = requests.get(url, params= params)
36
 
37
  if response.status_code == 200:
 
40
  nr_of_items = len(data['items'])
41
  random_int = random.randint(0,nr_of_items-1)
42
  # get the title and return it
43
+ return data['items'][random_int]["title"]
44
  else:
45
  return f'Error - get_random_name() -> Response status code was not good ({response.status_code})'
46