Upload agent

agent.json

@@ -1,13 +1,17 @@
 {
     "tools": [
+        "web_search",
+        "visit_webpage",
         "suggest_menu",
+        "catering_service_tool",
+        "superhero_party_theme_generator",
         "final_answer"
     ],
     "model": {
         "class": "HfApiModel",
         "data": {
-            "last_input_token_count": 3050,
-            "last_output_token_count": 46,
+            "last_input_token_count": 8557,
+            "last_output_token_count": 471,
             "model_id": "Qwen/Qwen2.5-Coder-32B-Instruct",
             "provider": null
         }
@@ -32,27 +36,30 @@
             "post_messages": "Based on the above, please provide an answer to the following user task:\n{{task}}"
         }
     },
-    "max_steps": 20,
-    "verbosity_level": 1,
+    "max_steps": 10,
+    "verbosity_level": 2,
     "grammar": null,
     "planning_interval": null,
     "name": null,
     "description": null,
     "requirements": [
-        "smolagents"
+        "duckduckgo_search",
+        "requests",
+        "smolagents",
+        "markdownify"
     ],
     "authorized_imports": [
-        "random",
-        "itertools",
-        "statistics",
-        "collections",
-        "queue",
-        "time",
         "unicodedata",
-        "datetime",
+        "collections",
         "stat",
+        "statistics",
+        "re",
+        "datetime",
         "math",
-        "re"
+        "queue",
+        "random",
+        "itertools",
+        "time"
     ],
     "executor_type": "local",
     "executor_kwargs": {},

app.py

@@ -5,7 +5,11 @@ from smolagents import GradioUI, CodeAgent, HfApiModel
 # Get current directory path
 CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
 
+from tools.web_search import DuckDuckGoSearchTool as WebSearch
+from tools.visit_webpage import VisitWebpageTool as VisitWebpage
 from tools.suggest_menu import SimpleTool as SuggestMenu
+from tools.catering_service_tool import SimpleTool as CateringServiceTool
+from tools.superhero_party_theme_generator import SuperheroPartyThemeTool as SuperheroPartyThemeGenerator
 from tools.final_answer import FinalAnswerTool as FinalAnswer
 
 
@@ -15,7 +19,11 @@ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
 provider=None,
 )
 
+web_search = WebSearch()
+visit_webpage = VisitWebpage()
 suggest_menu = SuggestMenu()
+catering_service_tool = CateringServiceTool()
+superhero_party_theme_generator = SuperheroPartyThemeGenerator()
 final_answer = FinalAnswer()
 
 
@@ -24,10 +32,10 @@ with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:
 
 agent = CodeAgent(
     model=model,
-    tools=[suggest_menu],
+    tools=[web_search, visit_webpage, suggest_menu, catering_service_tool, superhero_party_theme_generator],
     managed_agents=[],
-    max_steps=20,
-    verbosity_level=1,
+    max_steps=10,
+    verbosity_level=2,
     grammar=None,
     planning_interval=None,
     name=None,

requirements.txt

@@ -1 +1,4 @@
+duckduckgo_search
+requests
 smolagents
+markdownify

tools/superhero_party_theme_generator.py

@@ -6,17 +6,17 @@ class SuperheroPartyThemeTool(Tool):
     description = """
     This tool suggests creative superhero-themed party ideas based on a category.
     It returns a unique party theme idea."""
-    inputs = {'category': {'type': 'string', 'description': "The type of superhero party (e.g., 'classic heroes', 'villain masquerade', 'futuristic gotham')."}}
+    inputs = {'category': {'type': 'string', 'description': "The type of superhero party (e.g., 'classic heroes', 'villain masquerade', 'futuristic Gotham')."}}
     output_type = "string"
 
     def forward(self, category: str):
         themes = {
             "classic heroes": "Justice League Gala: Guests come dressed as their favorite DC heroes with themed cocktails like 'The Kryptonite Punch'.",
             "villain masquerade": "Gotham Rogues' Ball: A mysterious masquerade where guests dress as classic Batman villains.",
-            "futuristic gotham": "Neo-Gotham Night: A cyberpunk-style party inspired by Batman Beyond, with neon decorations and futuristic gadgets."
+            "futuristic Gotham": "Neo-Gotham Night: A cyberpunk-style party inspired by Batman Beyond, with neon decorations and futuristic gadgets."
         }
 
-        return themes.get(category.lower(), "Themed party idea not found. Try 'classic heroes', 'villain masquerade', or 'futuristic gotham'.")
+        return themes.get(category.lower(), "Themed party idea not found. Try 'classic heroes', 'villain masquerade', or 'futuristic Gotham'.")
 
     def __init__(self, *args, **kwargs):
         self.is_initialized = False

tools/visit_webpage.py

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