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