phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

ruff format

Browse source
This commit is contained in:
Tomer Bin 2025-03-13 08:58:29 +02:00
parent a8f6232eed
commit 6ebf11e4b3
1 changed files with 2 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/guardrails/guardrail_hooks/aim.py
View file

@ -36,12 +36,8 @@ class AimGuardrailMissingSecrets(Exception):
class AimGuardrail(CustomGuardrail): class AimGuardrail(CustomGuardrail):
def __init__( def __init__(self, api_key: Optional[str] = None, api_base: Optional[str] = None, **kwargs):
self, api_key: Optional[str] = None, api_base: Optional[str] = None, **kwargs self.async_handler = get_async_httpx_client(llm_provider=httpxSpecialProvider.GuardrailCallback)
):
self.async_handler = get_async_httpx_client(
llm_provider=httpxSpecialProvider.GuardrailCallback
)
self.api_key = api_key or os.environ.get("AIM_API_KEY") self.api_key = api_key or os.environ.get("AIM_API_KEY")
if not self.api_key: if not self.api_key:
msg = ( msg = (