Upload agent

agent.json

@@ -1,12 +1,14 @@
 {
     "tools": [
+        "web_search",
+        "suggest_menu",
         "final_answer"
     ],
     "model": {
         "class": "HfApiModel",
         "data": {
-            "last_input_token_count": 2569,
-            "last_output_token_count": 85,
+            "last_input_token_count": null,
+            "last_output_token_count": null,
             "model_id": "Qwen/Qwen2.5-Coder-32B-Instruct",
             "provider": null
         }
@@ -38,6 +40,7 @@
     "name": null,
     "description": null,
     "requirements": [
+        "duckduckgo_search",
         "smolagents"
     ],
     "authorized_imports": [

app.py

@@ -5,6 +5,8 @@ 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.suggest_menu import SimpleTool as SuggestMenu
 from tools.final_answer import FinalAnswerTool as FinalAnswer
 
 
@@ -14,6 +16,8 @@ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
 provider=None,
 )
 
+web_search = WebSearch()
+suggest_menu = SuggestMenu()
 final_answer = FinalAnswer()
 
 
@@ -22,7 +26,7 @@ with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:
 
 agent = CodeAgent(
     model=model,
-    tools=[],
+    tools=[web_search, suggest_menu],
     managed_agents=[],
     max_steps=20,
     verbosity_level=1,

requirements.txt

@@ -1 +1,2 @@
+duckduckgo_search
 smolagents

tools/suggest_menu.py

@@ -0,0 +1,24 @@
+from smolagents import Tool
+from typing import Any, Optional
+
+class SimpleTool(Tool):
+    name = "suggest_menu"
+    description = "Suggests a menu based on the occasion."
+    inputs = {"occasion":{"type":"string","description":"The type of occasion for the party"}}
+    output_type = "string"
+
+    def forward(self, occasion: str) -> str:
+        """
+        Suggests a menu based on the occasion.
+        Args:
+            occasion: The type of occasion for the party
+        """
+
+        if occasion == "casual":
+            return "Pizza, snacks and drinks"
+        elif occasion == "formal":
+            return "3-course dinner with wine and dessert"
+        elif occasion == "superhero":
+            return "Buffet with high-energy and healthy food"
+        else:
+            return "Custom menu for the butler"

tools/web_search.py

@@ -0,0 +1,27 @@
+from typing import Any, Optional
+from smolagents.tools import Tool
+import duckduckgo_search
+
+class DuckDuckGoSearchTool(Tool):
+    name = "web_search"
+    description = "Performs a duckduckgo web search based on your query (think a Google search) then returns the top search results."
+    inputs = {'query': {'type': 'string', 'description': 'The search query to perform.'}}
+    output_type = "string"
+
+    def __init__(self, max_results=10, **kwargs):
+        super().__init__()
+        self.max_results = max_results
+        try:
+            from duckduckgo_search import DDGS
+        except ImportError as e:
+            raise ImportError(
+                "You must install package `duckduckgo_search` to run this tool: for instance run `pip install duckduckgo-search`."
+            ) from e
+        self.ddgs = DDGS(**kwargs)
+
+    def forward(self, query: str) -> str:
+        results = self.ddgs.text(query, max_results=self.max_results)
+        if len(results) == 0:
+            raise Exception("No results found! Try a less restrictive/shorter query.")
+        postprocessed_results = [f"[{result['title']}]({result['href']})\n{result['body']}" for result in results]
+        return "## Search Results\n\n" + "\n\n".join(postprocessed_results)