Revert "Revert "AI update""

This reverts commit 554b7b330e16bd10ff42edef9dd2545297fb08c8.