borisyich commited on
Commit
7e0b4a4
·
verified ·
1 Parent(s): f3be6f5

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +12 -5
agent.py CHANGED
@@ -44,14 +44,21 @@ class BasicAgent:
44
 
45
  # Instantiate Agent
46
  self.agent = CodeAgent(tools=[
 
 
 
47
  FinalAnswerTool()
48
  ],
49
  model=model,
50
- managed_agents=[
51
- web_agent,
52
- python_agent
53
- ],
54
- additional_authorized_imports=["requests"],
 
 
 
 
55
  add_base_tools=True,
56
  )
57
 
 
44
 
45
  # Instantiate Agent
46
  self.agent = CodeAgent(tools=[
47
+ DuckDuckGoSearchTool(),
48
+ VisitWebpageTool(),
49
+ WikipediaSearchTool(),
50
  FinalAnswerTool()
51
  ],
52
  model=model,
53
+ additional_authorized_imports=[
54
+ "json",
55
+ "pandas",
56
+ "numpy",
57
+ "requests",
58
+ "time",
59
+ "datetime",
60
+ "re"
61
+ ],
62
  add_base_tools=True,
63
  )
64