File size: 1,260 Bytes
d733596 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 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." |