Have a look at my earlier post #30 which lists the Cane Creek 40 series parts, but they will be expensive.
Biggest issue at the moment is availability.
If the bearings are ok, but its the cables going through the headset that are your issue, I'd just re route outside and through the frame like the 2020 version
There's a thread here for another AT owner wanting to do the same.
I have a 2021 Cube AT which came with the cables routed through the headset/spacers. I’m putting on a Superstars headset and need to re-route the cables like the 2020 model. Where can I get the side plates with the holes in and also the backing plates. I know ebikeprints do the outers but...
www.emtbforums.com