Swap mob ai

I need to figure out how I would make a vex basically do the same thing as a wolf, I was able to get a wolf to ride a vex but the issue there is that the wolf cant control how the vex moves