How can we set AIU usage limit/restriction

Hello Team,

Is there a way to limit the amount of AIU spent in a given month.

For example, let's say we get our 20K AIU monthly, and are using AI features that uses Agents and Agent Teams with premium model. In this case, can we restrict 20K AI units to be spent and after that amount is spent, the AI features are disabled or not to be used.

We want to explore this option to avoid additional cost being incurred and have a limit to how much AIU are being spent every month.

If this is not possible, I want to confirm on a question: