Crossing the diplomatic line: When do ambassadors get expelled?

There have been calls for New Zealand to expel Russia's ambassador in response to the war in Ukraine. But is that the right diplomatic step to take?