added authorized imports
Browse files
app.py
CHANGED
|
@@ -32,6 +32,7 @@ class BasicAgent:
|
|
| 32 |
def __init__(self):
|
| 33 |
self.agent = CodeAgent(tools=[web_search_tool, wiki_search_tool, web_page_tool, python_interpreter],
|
| 34 |
model=InferenceClientModel(),
|
|
|
|
| 35 |
)
|
| 36 |
print("BasicAgent initialized.")
|
| 37 |
def __call__(self, question: str) -> str:
|
|
|
|
| 32 |
def __init__(self):
|
| 33 |
self.agent = CodeAgent(tools=[web_search_tool, wiki_search_tool, web_page_tool, python_interpreter],
|
| 34 |
model=InferenceClientModel(),
|
| 35 |
+
additional_authorized_imports=['numpy', 'pandas', 're']
|
| 36 |
)
|
| 37 |
print("BasicAgent initialized.")
|
| 38 |
def __call__(self, question: str) -> str:
|