Create tools.py

tools.py

@@ -0,0 +1,144 @@
+# tools.py
+# Additional tools for Alfred the Gala Agent
+
+from smolagents import tool
+import datetime
+
+
+@tool
+def get_current_time() -> str:
+    """
+    Gets the current date and time.
+    Use this when you need to know what time it is for scheduling or time-related questions.
+    
+    Returns:
+        The current date and time as a formatted string
+    """
+    now = datetime.datetime.now()
+    return f"Current date and time: {now.strftime('%A, %B %d, %Y at %I:%M %p')}"
+
+
+@tool
+def calculate_party_budget(
+    num_guests: int,
+    cost_per_guest: float,
+    venue_cost: float,
+    entertainment_cost: float
+) -> str:
+    """
+    Calculate the total budget needed for the gala party.
+    
+    Args:
+        num_guests: Number of guests attending
+        cost_per_guest: Cost per guest (food, drinks, favors)
+        venue_cost: Cost of the venue rental
+        entertainment_cost: Cost of entertainment (DJ, dancers, etc.)
+    
+    Returns:
+        A breakdown of the party budget
+    """
+    guest_total = num_guests * cost_per_guest
+    total = guest_total + venue_cost + entertainment_cost
+    
+    return f"""
+🎉 GALA BUDGET BREAKDOWN 🎉
+━━━━━━━━━━━━━━━━━━━━━━━━━
+Guests: {num_guests} × ${cost_per_guest:.2f} = ${guest_total:.2f}
+Venue: ${venue_cost:.2f}
+Entertainment: ${entertainment_cost:.2f}
+━━━━━━━━━━━━━━━━━━━━━━━━━
+TOTAL: ${total:.2f}
+"""
+
+
+@tool
+def suggest_seating_arrangement(guest_names: str, table_size: int) -> str:
+    """
+    Suggest a seating arrangement for guests.
+    
+    Args:
+        guest_names: Comma-separated list of guest names
+        table_size: Number of seats per table
+    
+    Returns:
+        A suggested seating arrangement
+    """
+    guests = [name.strip() for name in guest_names.split(",")]
+    num_guests = len(guests)
+    num_tables = (num_guests + table_size - 1) // table_size
+    
+    arrangement = "🪑 SEATING ARRANGEMENT 🪑\n"
+    arrangement += "━" * 30 + "\n"
+    
+    for table_num in range(num_tables):
+        start_idx = table_num * table_size
+        end_idx = min(start_idx + table_size, num_guests)
+        table_guests = guests[start_idx:end_idx]
+        
+        arrangement += f"\n📍 Table {table_num + 1}:\n"
+        for guest in table_guests:
+            arrangement += f"   • {guest}\n"
+    
+    return arrangement
+
+
+@tool
+def dietary_check(dietary_requirement: str) -> str:
+    """
+    Check common dietary requirements and suggest menu options.
+    
+    Args:
+        dietary_requirement: The dietary requirement to check (e.g., vegetarian, vegan, gluten-free, halal, kosher)
+    
+    Returns:
+        Menu suggestions for the dietary requirement
+    """
+    menus = {
+        "vegetarian": """
+🥗 VEGETARIAN MENU OPTIONS:
+• Stuffed bell peppers with quinoa
+• Mushroom risotto
+• Caprese salad with fresh mozzarella
+• Vegetable lasagna
+• Spinach and ricotta ravioli
+""",
+        "vegan": """
+🌱 VEGAN MENU OPTIONS:
+• Grilled vegetable platter
+• Coconut curry with tofu
+• Mediterranean mezze board
+• Mushroom and walnut "meatballs"
+• Fresh fruit tart with coconut cream
+""",
+        "gluten-free": """
+🌾 GLUTEN-FREE MENU OPTIONS:
+• Grilled salmon with herb butter
+• Rice paper spring rolls
+• Quinoa stuffed tomatoes
+• Flourless chocolate cake
+• Fresh fruit platter
+""",
+        "halal": """
+🍖 HALAL MENU OPTIONS:
+• Lamb kofta with yogurt sauce
+• Chicken shawarma
+• Beef kebabs with rice pilaf
+• Baklava dessert
+• Fresh salads
+""",
+        "kosher": """
+✡️ KOSHER MENU OPTIONS:
+• Roasted chicken with vegetables
+• Gefilte fish appetizer
+• Matzo ball soup
+• Brisket with roasted potatoes
+• Honey cake dessert
+"""
+    }
+    
+    requirement_lower = dietary_requirement.lower()
+    
+    if requirement_lower in menus:
+        return menus[requirement_lower]
+    else:
+        return f"I don't have specific menu suggestions for '{dietary_requirement}'. Please consult with the catering team for custom options."