"""
LuckBot Role Definitions.

Each role defines a specialized system prompt for a Swarm agent.
The role is selected based on the SWARM_AGENT_TYPE environment variable.
"""

from . import trader, developer, quant

ROLES = {
    "Trading": trader,
    "Development": developer,
    "Operations": developer,
    "Research": quant,
    # Aliases
    "trader": trader,
    "developer": developer,
    "quant": quant,
}


def get_role_prompt(agent_type: str) -> str:
    """Get the system prompt for a given agent type/role."""
    role = ROLES.get(agent_type)
    if role:
        return role.SYSTEM_PROMPT
    # Default to trader if unknown type
    return trader.SYSTEM_PROMPT


def get_role_name(agent_type: str) -> str:
    """Get the display name for a given agent type/role."""
    role = ROLES.get(agent_type)
    if role:
        return role.ROLE_NAME
    return "Trader"