scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 115

chore: format Bedrock guardrail

Browse source
This commit is contained in:
aiedwardyi 2026-08-26 12:38:57 +09:00
parent 34ae769590
commit 336003cc3b
No known key found for this signature in database
1 changed files with 7 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
litellm/proxy/guardrails/guardrail_hooks/bedrock_guardrails.py
View file

@ -782,8 +782,7 @@ class BedrockGuardrail(CustomGuardrail, BaseAWSLLM):
if matched_deployments:
return matched_deployments
if any(
isinstance(deployment, Mapping)
and (deployment.get("model_info") or {}).get("allow_fail_open") is True
isinstance(deployment, Mapping) and (deployment.get("model_info") or {}).get("allow_fail_open") is True
for deployment in allowed_deployments
):
return candidate_deployments or allowed_deployments
@ -825,11 +824,7 @@ class BedrockGuardrail(CustomGuardrail, BaseAWSLLM):
deployment_model: Final[object] = (
params.get("model") if isinstance(params, Mapping) else getattr(params, "model", None)
)
return (
BedrockGuardrail._resolve_model_provider(deployment_model)
if isinstance(deployment_model, str)
else None
)
return BedrockGuardrail._resolve_model_provider(deployment_model) if isinstance(deployment_model, str) else None
@staticmethod
def _router_allows_bedrock(
@ -1251,9 +1246,11 @@ class BedrockGuardrail(CustomGuardrail, BaseAWSLLM):
)
if router_allows_bedrock is not None:
return api_key if router_allows_bedrock else None
return api_key if all(
provider in ("bedrock", "bedrock_converse") for provider in providers
) else None
return (
api_key
if all(provider in ("bedrock", "bedrock_converse") for provider in providers)
else None
)
except Exception: # noqa: BLE001 # fall back to the sync compatibility path
pass