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: cover missing-model no-op and sagemaker pinning branches

Browse source
This commit is contained in:
Devin AI 2026-07-11 11:27:06 +00:00
parent 4e3cb7ac71
commit 3835d3e728
1 changed files with 23 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
tests/test_litellm/test_router_pin_deployment_credentials.py
View file

@ -80,6 +80,29 @@ def test_disabled_keeps_caller_override():
assert _merged(deployment, kwargs)["aws_role_name"] == CALLER_ROLE
def test_missing_model_is_noop():
litellm.pin_deployment_credentials = True
deployment = {"model_name": "claude", "litellm_params": {"aws_role_name": CONFIG_ROLE}}
kwargs = {"aws_role_name": CALLER_ROLE}
Router._pin_deployment_aws_credentials(deployment=deployment, kwargs=kwargs)
assert kwargs["aws_role_name"] == CALLER_ROLE
def test_sagemaker_provider_is_pinned():
litellm.pin_deployment_credentials = True
deployment = {
"model_name": "sm",
"litellm_params": {"model": "sagemaker_chat/my-endpoint", "aws_role_name": CONFIG_ROLE},
}
kwargs = {"aws_role_name": CALLER_ROLE}
Router._pin_deployment_aws_credentials(deployment=deployment, kwargs=kwargs)
assert "aws_role_name" not in kwargs
def test_non_aws_provider_untouched():
litellm.pin_deployment_credentials = True
deployment = {