Lezyne has a variety of ebike compatible lights and I believe also has a compatibility chart of which models work with which ebike motor brands.

I'd suggest checking with them for recommendations