The Wurkkos TS10 does not run on AA. It is 14500 only.
DC-Fix can diffuse a beam with a hotspot, or I have seen some lights with pebbled TIR optics that produce a very nice even beam.
If you really like your Tool, I would measure the reflector and see if you can find a pebbled or diffused TIR optic of the same size. Yajiamei is one possible retailer to try. They sell primarily through Aliexpress.
For some perspective, here’s some details from one user about various TIR optics he tried with his Convoy S2+ (the set he used will be too large for the Tool AA)