Spaces:
Configuration error
Configuration error
File size: 830 Bytes
121fcda ccbe781 121fcda ccbe781 121fcda ccbe781 121fcda ccbe781 121fcda ccbe781 121fcda ccbe781 121fcda | 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 | from smolagents import Tool
from typing import Any, Optional
class SimpleTool(Tool):
name = "suggest_menu"
description = "Suggests a menu based on occasion"
inputs = {'occasion': {'type': 'string', 'description': 'The of occasion for the party'}}
output_type = "string"
def forward(self, occasion: str) -> str:
"""
Suggests a menu based on occasion
Args:
occasion: The 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." |