Separate AI module