From 77e4008e7b8952a73917daf2f92ba656b50e065d Mon Sep 17 00:00:00 2001 From: = Date: Thu, 18 Sep 2025 17:13:55 -0700 Subject: [PATCH] lint fix --- litellm/proxy/guardrails/guardrail_hooks/bedrock_guardrails.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/guardrails/guardrail_hooks/bedrock_guardrails.py b/litellm/proxy/guardrails/guardrail_hooks/bedrock_guardrails.py index 19c26234287..5813a6079dd 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/bedrock_guardrails.py +++ b/litellm/proxy/guardrails/guardrail_hooks/bedrock_guardrails.py @@ -24,7 +24,6 @@ from litellm._logging import verbose_proxy_logger from litellm.caching import DualCache from litellm.integrations.custom_guardrail import ( CustomGuardrail, - log_guardrail_information, ) from litellm.llms.bedrock.base_aws_llm import BaseAWSLLM from litellm.llms.custom_httpx.http_handler import ( @@ -111,6 +110,7 @@ class BedrockGuardrail(CustomGuardrail, BaseAWSLLM): ) self.guardrailIdentifier = guardrailIdentifier self.guardrailVersion = guardrailVersion + self.guardrail_provider = "bedrock" # store kwargs as optional_params self.optional_params = kwargs @@ -372,6 +372,7 @@ class BedrockGuardrail(CustomGuardrail, BaseAWSLLM): # Add guardrail information to request trace ######################################################### self.add_standard_logging_guardrail_information_to_request_data( + guardrail_provider=self.guardrail_provider, guardrail_json_response=response.json(), request_data=request_data or {}, guardrail_status=self._get_bedrock_guardrail_response_status(