Victoria Shevchenko commited on
Commit
616dd97
·
1 Parent(s): 7d6269f

add extract_entities to available tools

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -107,7 +107,7 @@ with open("prompts.yaml", 'r') as stream:
107
 
108
  agent = CodeAgent(
109
  model=model,
110
- tools=[get_current_time_in_timezone, final_answer], ## add your tools here (don't remove final answer)
111
  max_steps=6,
112
  verbosity_level=1,
113
  grammar=None,
 
107
 
108
  agent = CodeAgent(
109
  model=model,
110
+ tools=[get_current_time_in_timezone, extract_entities, final_answer], ## add your tools here (don't remove final answer)
111
  max_steps=6,
112
  verbosity_level=1,
113
  grammar=None,