Was only handling message_update, now also handles message_start, message_end, turn_start, turn_end, agent_start, agent_end.