If AK is still a projected 2nd round pick, then unless he has some assurances (bronny style) that a team (maybe a couple of teams) that is a good fit and has a plan for him is going to pick him he should come back.
The NIL money is better staying than getting drafted by a random team and getting stuck on a 2-way or G-league deal.
If a team tells him he's likely going to stick, then he should go. Bird in the hand and all.
AK would have a better team around him this season and a better shot at a 3rd championship. He would also linked this generation of players to the championship DNA as the only starter left from those teams plus the 3 guys who were backups (Ball, Stewart, Ross). Culture win.