phspoden commited on
Commit
7a3127c
·
1 Parent(s): d730727

Added parameter page to get_random_name tool

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -24,14 +24,15 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
24
 
25
 
26
  @tool
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 = {'nationality': nationality}
35
  response = requests.get(url, params= params)
36
 
37
  if response.status_code == 200:
 
24
 
25
 
26
  @tool
27
+ def get_random_name(nationality: str, page: int) -> 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
+ page: An integer representing the paginated page number (e.g., 1). Starting value: 1.
32
  """
33
  try:
34
  url = 'https://api.fbi.gov/wanted/v1/list'
35
+ params = {'nationality': nationality, 'page': page}
36
  response = requests.get(url, params= params)
37
 
38
  if response.status_code == 200: