mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-05 02:41:56 +00:00
fix(bedrock/claude_platform): resolve reportArgumentType type error
Rename _resolve_unsupported_override -> resolve_unsupported_override (used cross-module, so underscore prefix triggered reportPrivateUsage) and accept a Protocol type instead of bare dict so GenericLiteLLMParams satisfies the type checker without a cast.
This commit is contained in:
parent
8bf4c56171
commit
2e55105020
3 changed files with 12 additions and 8 deletions
|
|
@ -1,8 +1,12 @@
|
|||
from typing import Literal, Optional, Tuple
|
||||
from typing import Literal, Optional, Protocol, Tuple
|
||||
|
||||
import litellm
|
||||
from litellm._logging import verbose_logger
|
||||
from litellm.llms.bedrock.base_aws_llm import BaseAWSLLM
|
||||
|
||||
|
||||
class _SupportsGet(Protocol):
|
||||
def get(self, key: str, default: object = None) -> object: ...
|
||||
from litellm.secret_managers.main import get_secret_str
|
||||
|
||||
CLAUDE_PLATFORM_SERVICE_NAME: Literal["aws-external-anthropic"] = (
|
||||
|
|
@ -74,8 +78,8 @@ def filter_claude_platform_request_body(
|
|||
}
|
||||
|
||||
|
||||
def _resolve_unsupported_override(
|
||||
litellm_params: dict,
|
||||
def resolve_unsupported_override(
|
||||
litellm_params: _SupportsGet,
|
||||
) -> Optional[frozenset[str]]:
|
||||
"""Read ``claude_platform_unsupported_params`` from litellm_params.
|
||||
|
||||
|
|
@ -87,7 +91,7 @@ def _resolve_unsupported_override(
|
|||
if raw is None:
|
||||
return None
|
||||
if isinstance(raw, (list, set, frozenset, tuple)):
|
||||
return frozenset(raw)
|
||||
return frozenset(str(item) for item in raw)
|
||||
return None
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -10,7 +10,7 @@ from litellm.types.router import GenericLiteLLMParams
|
|||
|
||||
from .common_utils import (
|
||||
BedrockClaudePlatformMixin,
|
||||
_resolve_unsupported_override,
|
||||
resolve_unsupported_override,
|
||||
filter_claude_platform_request_body,
|
||||
strip_claude_platform_route,
|
||||
)
|
||||
|
|
@ -67,7 +67,7 @@ class BedrockClaudePlatformMessagesConfig(
|
|||
litellm_params: GenericLiteLLMParams,
|
||||
headers: dict,
|
||||
) -> Dict:
|
||||
unsupported_override = _resolve_unsupported_override(litellm_params)
|
||||
unsupported_override = resolve_unsupported_override(litellm_params)
|
||||
anthropic_messages_optional_request_params = (
|
||||
filter_claude_platform_request_body(
|
||||
anthropic_messages_optional_request_params,
|
||||
|
|
|
|||
|
|
@ -7,7 +7,7 @@ from litellm.types.llms.openai import AllMessageValues
|
|||
|
||||
from .common_utils import (
|
||||
BedrockClaudePlatformMixin,
|
||||
_resolve_unsupported_override,
|
||||
resolve_unsupported_override,
|
||||
filter_claude_platform_request_body,
|
||||
)
|
||||
|
||||
|
|
@ -94,7 +94,7 @@ class BedrockClaudePlatformConfig(BedrockClaudePlatformMixin, AnthropicConfig):
|
|||
litellm_params: dict,
|
||||
headers: dict,
|
||||
) -> dict:
|
||||
unsupported_override = _resolve_unsupported_override(litellm_params)
|
||||
unsupported_override = resolve_unsupported_override(litellm_params)
|
||||
optional_params = filter_claude_platform_request_body(
|
||||
optional_params, unsupported_override=unsupported_override
|
||||
)
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue