tools/suggest_menu.py

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