feat(proxy): Assign default budget to auto-generated JWT teams #14514
+105
−3
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Relevant issues
Fixes #13581
Pre-Submission checklist
Please complete all items before asking a LiteLLM maintainer to review your PR
tests/litellm/
directory, Adding at least 1 test is a hard requirement - see detailsmake test-unit
Type
🆕 New Feature
Changes
This PR introduces a system to automatically assign a default budget to new teams created via the JWT
team_id_upsert
flow.create_team_with_defaults
function inlitellm/proxy/auth/auth_checks.py
to handle the creation of new teams and the application of default settings.max_budget
value from thelitellm.default_team_settings
list in the proxy configuration and applies it to the new team during creation._get_team_db_check
function has been refactored to use this new function, ensuring the logic is clean and reusable.Usage
To enable this feature, an administrator must configure
default_team_settings
in theirconfig.yaml
.Example
config.yaml
:Screenshot