Upload agent

agent.json

@@ -1,12 +1,13 @@
 {
     "tools": [
+        "suggest_menu",
         "final_answer"
     ],
     "model": {
         "class": "HfApiModel",
         "data": {
-            "last_input_token_count": 2558,
-            "last_output_token_count": 90,
+            "last_input_token_count": 2432,
+            "last_output_token_count": 451,
             "model_id": "Qwen/Qwen2.5-Coder-32B-Instruct",
             "provider": null
         }

app.py

@@ -5,6 +5,7 @@ from smolagents import GradioUI, CodeAgent, HfApiModel
 # Get current directory path
 CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
 
+from tools.suggest_menu import SimpleTool as SuggestMenu
 from tools.final_answer import FinalAnswerTool as FinalAnswer
 
 
@@ -14,6 +15,7 @@ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
 provider=None,
 )
 
+suggest_menu = SuggestMenu()
 final_answer = FinalAnswer()
 
 
@@ -22,7 +24,7 @@ with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:
 
 agent = CodeAgent(
     model=model,
-    tools=[],
+    tools=[suggest_menu],
     managed_agents=[],
     max_steps=6,
     verbosity_level=1,

tools/suggest_menu.py

@@ -0,0 +1,23 @@
+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."