File size: 1,140 Bytes
d5dc432 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from typing import Any, Optional
from smolagents.tools import Tool
class SuperheroPartyThemeTool(Tool):
name = "superhero_party_theme_generator"
description = """This tool suggests creative superhero-themed party ideas based on a category.
It returns a unique party theme idea."""
inputs = {'category': {'type': 'string', 'description': 'The type of superhero party (e.g)'}}
output_type = "string"
def forward(self, category:str):
themes = {
"classic heroes": "Justice League Gala: Guests come dressed as their favorite DC heroes with themed cocktails like 'The Kryptonite Punch'.",
"villain masquerade": "Gotham Rogues' Ball: A mysterious masquerade where guests dress as classic Batman villains.",
"futuristic Gotham": "Neo-Gotham Night: A cyberpunk-style party inspired by Batman Beyond, with neon decorations and futuristic gadgets."
}
return themes.get(category.lower(), "Themed party idea not found. Try 'classic heroes', 'villain masquerade', or 'futuristic Gotham'.")
def __init__(self, *args, **kwargs):
self.is_initialized = False
|