fix(bedrock/claude_platform): use builtin types for UP006 compliance

Replace List and Dict from typing with builtin list and dict to stay
within the ruff strict-rule budget ceiling for UP006.
This commit is contained in:
Venkat Donavalli 2026-06-24 08:33:29 -05:00 • committed by mateo-berri
parent a2c0cf9a46
commit 72811bcf2a
No known key found for this signature in database

View file

@ -1,4 +1,4 @@
from typing import Any, Dict, List, Optional
from typing import Any, Optional
import litellm
from litellm.llms.anthropic.chat.transformation import AnthropicConfig
@ -27,12 +27,12 @@ class BedrockClaudePlatformConfig(BedrockClaudePlatformMixin, AnthropicConfig):
self,
headers: dict,
model: str,
messages: List[AllMessageValues],
messages: list[AllMessageValues],
optional_params: dict,
litellm_params: dict,
api_key: Optional[str] = None,
api_base: Optional[str] = None,
) -> Dict:
) -> dict:
workspace_id = self._get_workspace_id(optional_params, litellm_params)
if workspace_id is None:
raise litellm.AuthenticationError(
@ -88,7 +88,7 @@ class BedrockClaudePlatformConfig(BedrockClaudePlatformMixin, AnthropicConfig):
def transform_request(
self,
model: str,
messages: List[AllMessageValues],
messages: list[AllMessageValues],
optional_params: dict,
litellm_params: dict,
headers: dict,