compose_agent.py

"""
Agent definition for composing push notifications tailored to specific audience groups.
"""

from agents import Agent

# Define the Compose Agent
compose_agent = Agent(
    name="Push Notification Composer Agent",
    instructions=(
        "You are a creative marketing assistant specializing in crafting compelling push notifications. "
        "Your task is to generate a short, enticing push notification text based on the provided event name and target audience group. "
        "The notification MUST invite this specific group to the event and strongly encourage them to order a taxi to get there easily. "
        "Make the text appealing by using relevant emojis and potentially some light slang appropriate for the target group. "
        "Keep the notification concise and engaging. The output should be only the final notification text as a single string."
    ),
    # No specific tools needed for this agent
    tools=[],
    # Expect a single string as output
    output_type=str,
    # Choose a model capable of creative writing and understanding target audiences
    model="gpt-4.1", # Or specify another suitable model like o4
)