Upload tool

agent.json

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

app.py

@@ -1,23 +1,22 @@
-from smolagents import GradioUI, CodeAgent, HfApiModel
-from scripts.web_search import DuckDuckGoSearchTool
-from scripts.visit_webpage import VisitWebpageTool
-from scripts.final_answer import FinalAnswerTool
 
-model = HfApiModel()
-web_search = DuckDuckGoSearchTool()
-visit_webpage = VisitWebpageTool()
-final_answer = FinalAnswerTool()
+from smolagents import GradioUI, {{ class_name }}, {{ agent_dict['model']['class'] }}
 
-agent = CodeAgent(
+{% for tool in tools.values() %}
+from scripts.{{ tool.name }} import {{ tool.__class__.__name__ }}
+{% endfor %}
+
+model = {{ agent_dict['model']['class'] }}()
+
+{% for tool in tools.values() %}
+{{ tool.name }} = {{ tool.__class__.__name__ }}()
+{% endfor %}
+
+agent = {{ class_name }}(
     model=model,
-    tools= [web_search, visit_webpage, final_answer],
-    max_steps=6,
-    verbosity_level=1,
-    grammar=None,
-    planning_interval=None,
-    name=None,
-    description=None,
-    authorized_imports=['datetime', 'queue', 'stat', 'itertools', 'pandas', 'random', 'collections', 'math', 'statistics', 'unicodedata', 're', 'time'],
+    tools=[{% for tool in tools.keys() %}{{ tool }}{% if not loop.last %}, {% endif %}{% endfor %}],
+    {% for attribute_name, value in agent_dict.items() if attribute_name not in ["model", "tools", "prompt_templates"] %}
+    {{ attribute_name }}={{ value }},
+    {% endfor %}
     prompts_path='./prompts.yaml'
 )
 

requirements.txt

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

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"