Solasta got it right. After you surprise attack something the rest of your party also get a turn while the enemy have the surprised condition. After that you roll initiative normally. BG3 needs to do it the same way.
I thought even that was powerful. I liked how Pathfinder handled it with “half-turn” surprise turns.
