tools/suggest_menu.py

from smolagents import Tool
from typing import Any, Optional

class SimpleTool(Tool):
    name = "suggest_menu"
    description = "Suggest a menu for a given 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:
        """
        Suggest a menu for a given 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.

        Returns:
            str: A suggested menu for the occasion.
        """
        # Define a dictionary of occasions and their corresponding menus
        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."