Upload tool

agent.json

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

app.py

@@ -18,6 +18,9 @@ visit_webpage = VisitWebpageTool()
 final_answer = FinalAnswerTool()
 
 
+with open("prompts.yaml", 'r') as stream:
+    prompt_templates = yaml.safe_load(stream)
+
 agent = CodeAgent(
     model=model,
     tools=[web_search, visit_webpage, final_answer],
@@ -27,7 +30,7 @@ agent = CodeAgent(
     planning_interval=None,
     name=None,
     description=None,
-    prompt_templates=yaml.safe_load(Path("prompts.yaml").read_text()
+    prompt_templates=prompt_templates
 )
 
 GradioUI(agent).launch()

requirements.txt

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

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
+import markdownify
 
 class VisitWebpageTool(Tool):
     name = "visit_webpage"