drAbreu commited on
Commit
a0bacd9
·
1 Parent(s): 0858b8a

Adding web tools to agent

Browse files
Files changed (1) hide show
  1. agents/llama_index_agent.py +3 -2
agents/llama_index_agent.py CHANGED
@@ -53,8 +53,9 @@ class GaiaAgent(ReActAgent):
53
  if tools is None:
54
  tools = [
55
  reverse_text_tool,
56
- wikipedia_tool,
57
- tavily_tool
 
58
  ]
59
 
60
  # Use default system prompt if not provided
 
53
  if tools is None:
54
  tools = [
55
  reverse_text_tool,
56
+ wikipedia_tool.load_data,
57
+ wikipedia_tool.search_data,
58
+ tavily_tool.search
59
  ]
60
 
61
  # Use default system prompt if not provided