bstraehle commited on
Commit
d08687f
·
verified ·
1 Parent(s): dc9ea5d

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
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)