Update agent.py
Browse files
agent.py
CHANGED
|
@@ -15,12 +15,12 @@ def run_agent(model_id, level, question, file_name):
|
|
| 15 |
|
| 16 |
agent = CodeAgent(
|
| 17 |
add_base_tools=True,
|
| 18 |
-
additional_authorized_imports=["pandas", "numpy"]
|
| 19 |
max_steps=max_steps,
|
| 20 |
model=OpenAIServerModel(model_id=model_id),
|
| 21 |
tools=[DuckDuckGoSearchTool(),
|
| 22 |
WikipediaSearchTool()],
|
| 23 |
-
verbosity_level=2
|
| 24 |
)
|
| 25 |
|
| 26 |
return agent.run(question)
|
|
|
|
| 15 |
|
| 16 |
agent = CodeAgent(
|
| 17 |
add_base_tools=True,
|
| 18 |
+
additional_authorized_imports=["pandas", "numpy"],
|
| 19 |
max_steps=max_steps,
|
| 20 |
model=OpenAIServerModel(model_id=model_id),
|
| 21 |
tools=[DuckDuckGoSearchTool(),
|
| 22 |
WikipediaSearchTool()],
|
| 23 |
+
verbosity_level=2
|
| 24 |
)
|
| 25 |
|
| 26 |
return agent.run(question)
|