| 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." |