Princekumar commited on
Commit ·
0af8313
1
Parent(s): ce0a9d0
adding authorized imports pandas, openpyxl and requests
Browse files
app.py
CHANGED
|
@@ -31,6 +31,7 @@ class BasicAgent:
|
|
| 31 |
self.agent = CodeAgent(
|
| 32 |
model=llm_model,
|
| 33 |
tools=agent_tools,
|
|
|
|
| 34 |
)
|
| 35 |
self.agent.system_prompt += "\n" + SYSTEM_PROMPT
|
| 36 |
print("Smolagent initialized successfully")
|
|
|
|
| 31 |
self.agent = CodeAgent(
|
| 32 |
model=llm_model,
|
| 33 |
tools=agent_tools,
|
| 34 |
+
additional_authorized_imports=["pandas", "openpyxl", "requests"],
|
| 35 |
)
|
| 36 |
self.agent.system_prompt += "\n" + SYSTEM_PROMPT
|
| 37 |
print("Smolagent initialized successfully")
|