LucaR84 commited on
Commit
b9a38d4
·
verified ·
1 Parent(s): 8edca5e

Update app.py

Browse files

Forgot to add the pokemon locations tool to the list of the tools

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,7 +10,7 @@ from Gradio_UI import GradioUI
10
 
11
  # Below is an example of a tool that does nothing. Amaze us with your creativity !
12
  @tool
13
- def pokemon_locations(pokemon_name: str) -> dict:
14
 
15
  """A tool that retrieves the areas where Pokémon can be found.
16
  Args:
@@ -215,7 +215,7 @@ with open("prompts.yaml", 'r') as stream:
215
 
216
  agent = CodeAgent(
217
  model=model,
218
- tools=[get_pokemon_detail, image_generation_tool, get_current_time_in_timezone, final_answer], ## add your tools here (don't remove final answer)
219
  max_steps=6,
220
  verbosity_level=1,
221
  grammar=None,
 
10
 
11
  # Below is an example of a tool that does nothing. Amaze us with your creativity !
12
  @tool
13
+ def get_pokemon_locations(pokemon_name: str) -> dict:
14
 
15
  """A tool that retrieves the areas where Pokémon can be found.
16
  Args:
 
215
 
216
  agent = CodeAgent(
217
  model=model,
218
+ tools=[get_pokemon_locations, get_pokemon_detail, image_generation_tool, get_current_time_in_timezone, final_answer], ## add your tools here (don't remove final answer)
219
  max_steps=6,
220
  verbosity_level=1,
221
  grammar=None,