Updated main_agent variable
Browse files
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,
|