How do I use a Union type as an action argument? - Ash Framework