scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-05 02:41:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 49

style(bedrock/claude_platform): reformat with black + add coverage tests

Browse source 
Black and ruff format disagree on line wrapping; CI uses black. Reformat
common_utils.py with black. Add messages-path override test to cover
resolve_unsupported_override call in messages_transformation.py.
This commit is contained in:
Venkat Donavalli 2026-06-24 09:10:01 -05:00 • committed by mateo-berri
parent 40e44ed4d6
commit 1acef7d880
No known key found for this signature in database
2 changed files with 47 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
litellm/llms/bedrock/claude_platform/common_utils.py
View file

@ -10,7 +10,9 @@ class _SupportsGet(Protocol):
def get(self, key: str, default: object = None) -> object: ...
CLAUDE_PLATFORM_SERVICE_NAME: Literal["aws-external-anthropic"] = "aws-external-anthropic"
CLAUDE_PLATFORM_SERVICE_NAME: Literal["aws-external-anthropic"] = (
"aws-external-anthropic"
)
CLAUDE_PLATFORM_BEDROCK_ROUTE = "claude_platform/"
# Auth/routing params consumed by validate_environment / sign_request that
@ -55,7 +57,9 @@ def filter_claude_platform_request_body(
Filters a copy so callers' ``sign_request`` still sees ``aws_region_name``.
"""
unsupported = (
unsupported_override if unsupported_override is not None else CLAUDE_PLATFORM_ON_AWS_UNSUPPORTED_REQUEST_PARAMS
unsupported_override
if unsupported_override is not None
else CLAUDE_PLATFORM_ON_AWS_UNSUPPORTED_REQUEST_PARAMS
)
dropped_unsupported = [k for k in params if k in unsupported]
if dropped_unsupported:
@ -69,7 +73,9 @@ def filter_claude_platform_request_body(
return {
k: v
for k, v in params.items()
if k not in CLAUDE_PLATFORM_ON_AWS_NON_REQUEST_PARAMS and k not in unsupported and not k.startswith("aws_")
if k not in CLAUDE_PLATFORM_ON_AWS_NON_REQUEST_PARAMS
and k not in unsupported
and not k.startswith("aws_")
}
@ -108,10 +114,14 @@ class BedrockClaudePlatformMixin(BaseAWSLLM):
or litellm_params.get("anthropic-workspace-id")
)
if workspace_id is None:
workspace_id = optional_params.get("anthropic_workspace_id") or litellm_params.get("anthropic_workspace_id")
workspace_id = optional_params.get(
"anthropic_workspace_id"
) or litellm_params.get("anthropic_workspace_id")
if workspace_id is not None:
return str(workspace_id)
return get_secret_str("ANTHROPIC_AWS_WORKSPACE_ID") or get_secret_str("ANTHROPIC_WORKSPACE_ID")
return get_secret_str("ANTHROPIC_AWS_WORKSPACE_ID") or get_secret_str(
"ANTHROPIC_WORKSPACE_ID"
)
def _get_required_aws_region_name(self, optional_params: dict) -> str:
aws_region_name = (
@ -149,7 +159,9 @@ class BedrockClaudePlatformMixin(BaseAWSLLM):
)
if api_base is None:
aws_region_name = self._get_required_aws_region_name(optional_params)
api_base = f"https://{CLAUDE_PLATFORM_SERVICE_NAME}.{aws_region_name}.api.aws"
api_base = (
f"https://{CLAUDE_PLATFORM_SERVICE_NAME}.{aws_region_name}.api.aws"
)
if not api_base.endswith("/v1/messages"):
api_base = f"{api_base.rstrip('/')}/v1/messages"
return api_base

29
tests/test_litellm/llms/bedrock/test_claude_platform_provider.py
View file

@ -505,6 +505,35 @@ def test_claude_platform_unsupported_override_ignores_invalid_type():
assert request_body["max_tokens"] == 10
def test_claude_platform_messages_unsupported_override_allows_context_management():
"""
Messages-path: operators can pass claude_platform_unsupported_params=[]
in litellm_params to allow context_management through.
"""
import litellm
from litellm.types.utils import LlmProviders
config = litellm.ProviderConfigManager.get_provider_anthropic_messages_config(
model="claude_platform/claude-sonnet-4-6",
provider=LlmProviders.BEDROCK,
)
assert config is not None
request_body = config.transform_anthropic_messages_request(
model="claude_platform/claude-sonnet-4-6",
messages=[{"role": "user", "content": "hello"}],
anthropic_messages_optional_request_params={
"context_management": {"edits": [{"type": "clear_tool_uses_20250919"}]},
"max_tokens": 10,
},
litellm_params={"claude_platform_unsupported_params": []},
headers={},
)
assert "context_management" in request_body
assert request_body["max_tokens"] == 10
def test_chat_completion_claude_platform_sigv4_body_has_no_auth_params():
"""
End-to-end (mocked transport): a config-driven SigV4 call with