I haven’t tore down my tool AA to check yet but most of these AA or 14500 lights use the PAM2803 boost ic circuit with a FET.
With a 14500 it runs in direct drive with the components in series with the led and feedback resistor limiting current. In my experience the PAM2803 is usually what goes bad first but I have also seen the FET go bad too. You might try replacing it if that’s what is used on the driver.
I get mine from Arrow with free shipping. https://www.arrow.com/en/products/pam2803aaf095/diodes-incorporated