enhance AI agents with predictive ML