From 6ebf11e4b301d05925032682a37ec6f250e1c343 Mon Sep 17 00:00:00 2001 From: Tomer Bin Date: Thu, 13 Mar 2025 08:58:29 +0200 Subject: [PATCH] ruff format --- litellm/proxy/guardrails/guardrail_hooks/aim.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/litellm/proxy/guardrails/guardrail_hooks/aim.py b/litellm/proxy/guardrails/guardrail_hooks/aim.py index 920d9161497..f062189b536 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/aim.py +++ b/litellm/proxy/guardrails/guardrail_hooks/aim.py @@ -36,12 +36,8 @@ class AimGuardrailMissingSecrets(Exception): class AimGuardrail(CustomGuardrail): - def __init__( - self, api_key: Optional[str] = None, api_base: Optional[str] = None, **kwargs - ): - self.async_handler = get_async_httpx_client( - llm_provider=httpxSpecialProvider.GuardrailCallback - ) + def __init__(self, api_key: Optional[str] = None, api_base: Optional[str] = None, **kwargs): + self.async_handler = get_async_httpx_client(llm_provider=httpxSpecialProvider.GuardrailCallback) self.api_key = api_key or os.environ.get("AIM_API_KEY") if not self.api_key: msg = (