madsc13nt1st commited on
Commit
37d820d
·
verified ·
1 Parent(s): 4212b91

Updated main_agent variable

Browse files
Files changed (1) hide show
  1. agent.py +2 -1
agent.py CHANGED
@@ -27,8 +27,9 @@ calculator_agent = CodeAgent(
27
  )
28
 
29
  main_agent = CodeAgent(
30
- tools=[search_tool, wiki_tool, rag_tool],
31
  model=model,
 
32
  description = system_prompt,
33
  managed_agents = [calculator_agent],
34
  max_steps=10,
 
27
  )
28
 
29
  main_agent = CodeAgent(
30
+ tools=[search_tool, wiki_tool, rag_tool, transcriber_tool, visit_webpage_tool],
31
  model=model,
32
+ additional_authorized_imports = ['requests', 'bs4', 'whisper', 'langchain_community', 'langchain'],
33
  description = system_prompt,
34
  managed_agents = [calculator_agent],
35
  max_steps=10,