AI automation for infrastructure management