scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

test(proxy): lock team wildcard deployment resolution for provider-scoped credentials

Browse source
This commit is contained in:
mateo-berri 2026-07-29 11:29:38 -07:00
parent 3639e3619f
commit 1a6d7ff1d3
1 changed files with 34 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
tests/test_litellm/proxy/openai_files_endpoint/test_files_endpoint.py
View file

@ -2778,3 +2778,37 @@ def test_provider_scoped_credentials_never_use_other_teams_deployment():
assert credentials is not None
assert credentials["vertex_project"] == "shared-project"
def test_provider_scoped_credentials_resolve_team_wildcard_deployment():
from litellm.proxy.openai_files_endpoints.common_utils import (
resolve_provider_scoped_credentials,
)
router = Router(
model_list=[
{
"model_name": "team-vertex-wildcard",
"litellm_params": {
"model": "vertex_ai/*",
"vertex_project": "team-wildcard-project",
},
"model_info": {
"id": "team-wildcard-dep",
"team_id": "team-caller",
"team_public_model_name": "vertex_ai/*",
},
}
]
)
credentials = resolve_provider_scoped_credentials(
llm_router=router,
custom_llm_provider="vertex_ai",
user_api_key_dict=UserAPIKeyAuth(
api_key="sk-test", team_id="team-caller", team_models=["vertex_ai/*"]
),
)
assert credentials is not None
assert credentials["vertex_project"] == "team-wildcard-project"