Upload 3 files

app.py

@@ -1,14 +1,12 @@
-# app.py
 import os
 import gradio as gr
 import requests
-import inspect
 import pandas as pd
 
 from smolagents import CodeAgent
-from smolagents.tools.search import DuckDuckGoSearchTool
-from smolagents.models import HfApiModel
-from smolagents.sandboxes import E2BSandbox
+from smolagents import DuckDuckGoSearchTool
+from smolagents.models import ApiModel as HfApiModel
+from e2b_code_interpreter import Sandbox as E2BSandbox
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
@@ -16,8 +14,26 @@ class BasicAgent:
     def __init__(self):
         model = HfApiModel(model="mistralai/Mistral-7B-Instruct-v0.2")
         tools = [DuckDuckGoSearchTool()]
-        sandbox = E2BSandbox(allowed_imports=["math", "datetime"], trusted=True)
-        self.agent = CodeAgent(tools=tools, model=model, sandbox=sandbox)
+        sandbox = E2BSandbox(
+            template=None,
+            timeout=60,
+            metadata=None,
+            envs=None,
+            secure=True,
+            api_key=None,
+            domain=None,
+            debug=False,
+            sandbox_id=None,
+            request_timeout=30.0,
+            proxy=None
+        )
+        self.agent = CodeAgent(
+            tools=tools,
+            model=model,
+            executor_type="e2b",
+            executor_kwargs={"sandbox": sandbox},
+            additional_authorized_imports=["math", "datetime"]
+        )
 
     def __call__(self, question: str) -> str:
         return self.agent.run(question).strip()

requirements.txt

@@ -1,4 +1,5 @@
 gradio==5.25.2
 requests
 pandas
-https://github.com/huggingface/agents/archive/refs/heads/main.zip
+smolagents[e2b]
+https://github.com/huggingface/agents/archive/refs/heads/main.zip