Configure LWAI Prediction Query stage