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