cedwards427 commited on
Commit
55d6fe8
·
verified ·
1 Parent(s): 4ad74af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -4,6 +4,9 @@ import requests
4
  import bs4
5
  import inspect
6
  import pandas as pd
 
 
 
7
 
8
  from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, tool, OpenAIServerModel
9
 
@@ -20,7 +23,7 @@ class BasicAgent:
20
 
21
  self.agent = CodeAgent(
22
  tools=[DuckDuckGoSearchTool()],
23
- additional_authorized_imports=['requests', 'bs4'],
24
  model=model,
25
  max_steps=10,
26
  verbosity_level=2
 
4
  import bs4
5
  import inspect
6
  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
 
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