Spaces:
Sleeping
Sleeping
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 coding day."}} | |
output_type = "string" | |
def forward(self, occasion: str) -> str: | |
""" | |
Suggests a menu based on the occasion. | |
Args: | |
occasion: The type of occasion for the coding day. | |
""" | |
if occasion == "vibe": | |
return "Pizza, snacks, and drinks." | |
elif occasion == "bit stressed": | |
return "some normal food." | |
elif occasion == "deadline": | |
return "Buffet with high-energy and healthy food." | |
else: | |
return "Custom menu for the butler." |