AI cost control