Manage budgets for Unity AI Gateway
This feature is in Public Preview.
Budgets help you monitor monthly Unity AI Gateway spend using filters applied to billing records. You can scope a budget specifically to the Unity AI Gateway product to track only Unity AI Gateway spending, or include Unity AI Gateway in an account-wide budget.
Unity AI Gateway budgets also track Genie product usage. To set budgets for Genie specifically, see Manage budgets and cost controls for Genie.
For details on creating and monitoring budgets, see Create and monitor budgets.
Requirements
- Unity AI Gateway enabled for your account.
- The billable usage system table enabled for your account. See Enable system tables.
- The Unity AI Gateway Budget Public Preview enabled in the account console. See Manage Databricks previews.
How budgets work with Unity AI Gateway
When creating a budget, select Unity AI Gateway as the resource type to track only Unity AI Gateway spending (pay-per-token and ai_query inference).
Per-user spend thresholds: For budgets scoped to Unity AI Gateway, you can configure per-user thresholds that track individual user spending and send alerts when a user exceeds their threshold.
Limitations
- Unity AI Gateway budgets currently track pay-per-token and ai_query (batch) inference. Provisioned throughput and external-model inference are not currently tracked.