mohammadreza pakzadian commited on
Commit
43c2f23
·
1 Parent(s): 6442393

add additional_authorized_imports to code agent

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -32,7 +32,8 @@ class BasicAgent:
32
  tools=[visit_webpage, web_search, final_answer, youtube_transcript],
33
  max_steps=5,
34
  verbosity_level=1,
35
- add_base_tools=True
 
36
  )
37
  def __call__(self, question: str) -> str:
38
  print(f"Agent received question (first 50 chars): {question}")
 
32
  tools=[visit_webpage, web_search, final_answer, youtube_transcript],
33
  max_steps=5,
34
  verbosity_level=1,
35
+ add_base_tools=True,
36
+ additional_authorized_imports=['random', 'stat', 'time', 'collections', 'itertools', 'queue', 're', 'datetime', 'statistics', 'math', 'unicodedata', 'csv', 'pandas']
37
  )
38
  def __call__(self, question: str) -> str:
39
  print(f"Agent received question (first 50 chars): {question}")