Upload tool

agent.json

@@ -36,16 +36,16 @@
     "description": null,
     "authorized_imports": [
         "datetime",
+        "stat",
+        "unicodedata",
         "pandas",
-        "math",
         "statistics",
+        "time",
         "queue",
-        "collections",
-        "itertools",
         "re",
-        "time",
-        "stat",
+        "itertools",
+        "math",
         "random",
-        "unicodedata"
+        "collections"
     ]
 }

app.py

@@ -1,5 +1,6 @@
 
 import yaml
+from pathlib import Path
 from smolagents import GradioUI, CodeAgent, HfApiModel
 
 from tools.web_search import DuckDuckGoSearchTool
@@ -17,10 +18,6 @@ visit_webpage = VisitWebpageTool()
 final_answer = FinalAnswerTool()
 
 
-prompt_templates = yaml.safe_load(
-    importlib.resources.files("smolagents.prompts").joinpath("code_agent.yaml").read_text()
-)
-
 agent = CodeAgent(
     model=model,
     tools=[web_search, visit_webpage, final_answer],
@@ -30,7 +27,7 @@ agent = CodeAgent(
     planning_interval=None,
     name=None,
     description=None,
-    prompt_templates=prompt_templates
+    prompt_templates=yaml.safe_load(Path("prompts.yaml").read_text()
 )
 
 GradioUI(agent).launch()

requirements.txt

@@ -1,5 +1,5 @@
-duckduckgo_search
-requests
 pandas
-markdownify
 smolagents
+markdownify
+requests
+duckduckgo_search

tools/visit_webpage.py

@@ -1,8 +1,8 @@
 from typing import Any, Optional
 from smolagents.tools import Tool
 import markdownify
-import requests
 import smolagents
+import requests
 
 class VisitWebpageTool(Tool):
     name = "visit_webpage"