AI Budget Control