fix aporia typo

This commit is contained in:
Ishaan Jaff 2024-08-19 18:34:45 -07:00
parent e7fa2cf0aa
commit ca9c81470b
5 changed files with 24 additions and 24 deletions

View file

@ -113,11 +113,11 @@ def init_guardrails_v2(all_guardrails: dict):
# Init guardrail CustomLoggerClass
if litellm_params["guardrail"] == "aporia":
from litellm.proxy.enterprise.enterprise_hooks.aporio_ai import (
_ENTERPRISE_Aporio,
from litellm.proxy.enterprise.enterprise_hooks.aporia_ai import (
_ENTERPRISE_Aporia,
)
_aporia_callback = _ENTERPRISE_Aporio(
_aporia_callback = _ENTERPRISE_Aporia(
api_base=litellm_params["api_base"],
api_key=litellm_params["api_key"],
guardrail_name=guardrail["guardrail_name"],