File size: 785 Bytes
37947df
0c4fbad
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
from smolagents import tool

@tool
def suggest_menu(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."