def swap_roles(messages):
    for message in messages:
        if message['role'] == 'user':
            message['role'] = 'assistant'
        elif message['role'] == 'assistant':
            message['role'] = 'user'
    return messages
agents = [agent_1, agent_2]
# Kick off the conversation
messages = [{"role": "user", "message": "Hello!"}]
while True:
    for agent in agents:
        messages = agent.chat(messages)
        messages = swap_roles(messages)