I have used SRAM GX AXS 12 speed on the Vado 5.0 for a couple years now and it has been good. It was a little bit of faff to get everything properly aligned and shifting but it works fine. I am probably eventually going to swap over to the new XT DI2 wireless stuff because I like Shimano better but I don't know if Zirbel will do a shifter for it and their SRAM Shifter while expensive with the BlipBox is really really nice and super minimalist.

On your current bike I would leave it as is but your next e-bike I think it is well worth it assuming you are good with charging as compared to Di2 (wired) the battery doesn't quite last as long as I would want and doesn't hold charge for longer periods (which is fair and fine). I cannot speak to the new wireless Di2 in terms of longevity but knowing Shimano they have been testing it a lot and doing their best to make it ready to go whereas SRAM likes to get things out faster and let the public do a little more R+D for them but AXS is pretty dialed at this point.