Upload agent

agent.json

@@ -1,6 +1,7 @@
 {
     "class": "CodeAgent",
     "tools": [
+        "suggest_menu",
         "final_answer"
     ],
     "model": {

app.py

@@ -5,6 +5,7 @@ from smolagents import GradioUI, CodeAgent, InferenceClientModel
 # 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
 
 
@@ -13,6 +14,7 @@ model = InferenceClientModel(
 model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
 )
 
+suggest_menu = SuggestMenu()
 final_answer = FinalAnswer()
 
 
@@ -21,7 +23,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=20,
     verbosity_level=1,

tools/suggest_menu.py

@@ -0,0 +1,27 @@
+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. Allowed values are: - "casual": Menu for casual party. - "formal": Menu for formal party. - "superhero": Menu for superhero party. - "custom": Custom menu.'}}
+    output_type = "string"
+
+    def forward(self, occasion: str) -> str:
+        """
+        Suggests a menu based on the occasion.
+        Args:
+            occasion (str): The type of occasion for the party. Allowed values are:
+                            - "casual": Menu for casual party.
+                            - "formal": Menu for formal party.
+                            - "superhero": Menu for superhero party.
+                            - "custom": Custom menu.
+        """
+        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."