Upload tool

agent.json

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

app.py

@@ -1,4 +1,5 @@
 
+import yaml
 from smolagents import GradioUI, CodeAgent, HfApiModel
 
 from tools.web_search import DuckDuckGoSearchTool
@@ -16,6 +17,10 @@ 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],
@@ -25,7 +30,7 @@ agent = CodeAgent(
     planning_interval=None,
     name=None,
     description=None,
-    prompts_path='./prompts.yaml'
+    prompt_templates=prompt_templates
 )
 
 GradioUI(agent).launch()

requirements.txt

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

tools/visit_webpage.py

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