Adding an association in an action based on argument - Ash Framework