cedwards427 commited on
Commit
28e2271
·
verified ·
1 Parent(s): 8a894b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -7,6 +7,8 @@ import pandas as pd
7
  import json
8
  import csv
9
  import openpyxl
 
 
10
 
11
  from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, tool, OpenAIServerModel
12
 
@@ -23,7 +25,7 @@ class BasicAgent:
23
 
24
  self.agent = CodeAgent(
25
  tools=[DuckDuckGoSearchTool()],
26
- additional_authorized_imports=['requests', 'bs4', 'csv', 'pandas', 'json', 'openpyxl'],
27
  model=model,
28
  max_steps=10,
29
  verbosity_level=2
 
7
  import json
8
  import csv
9
  import openpyxl
10
+ import io
11
+ import exec
12
 
13
  from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, tool, OpenAIServerModel
14
 
 
25
 
26
  self.agent = CodeAgent(
27
  tools=[DuckDuckGoSearchTool()],
28
+ additional_authorized_imports=['requests', 'bs4', 'csv', 'pandas', 'json', 'openpyxl', 'io', 'exec'],
29
  model=model,
30
  max_steps=10,
31
  verbosity_level=2