From 336003cc3b9ae12e2f2b6c25ddeb54f9e752a5ef Mon Sep 17 00:00:00 2001 From: aiedwardyi <41576951+aiedwardyi@users.noreply.github.com> Date: Wed, 26 Aug 2026 12:38:57 +0900 Subject: [PATCH] chore: format Bedrock guardrail --- .../guardrail_hooks/bedrock_guardrails.py | 17 +++++++---------- 1 file changed, 7 insertions(+), 10 deletions(-) diff --git a/litellm/proxy/guardrails/guardrail_hooks/bedrock_guardrails.py b/litellm/proxy/guardrails/guardrail_hooks/bedrock_guardrails.py index 56cfb87dc6f..18a259b1eb3 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/bedrock_guardrails.py +++ b/litellm/proxy/guardrails/guardrail_hooks/bedrock_guardrails.py @@ -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