Coverage for .venv/lib/python3.13/site-packages/litellm/proxy/guardrails/guardrail_hooks/alice/__init__.py: 57%

1from typing import TYPE_CHECKING, Final 

2 

3from litellm.types.guardrails import SupportedGuardrailIntegrations 

4 

5from .alice import AliceGuardrail 

6 

7if TYPE_CHECKING: 7 ↛ 8line 7 didn't jump to line 8 because the condition on line 7 was never true

8 from litellm.types.guardrails import Guardrail, LitellmParams 

9 

10 

11def initialize_guardrail(litellm_params: "LitellmParams", guardrail: "Guardrail"): 

12 import litellm 

13 

14 _alice_guardrail_callback: Final = AliceGuardrail( 

15 api_key=litellm_params.api_key, 

16 api_base=litellm_params.api_base, 

17 unreachable_fallback=getattr(litellm_params, "unreachable_fallback", "fail_closed"), 

18 guardrail_name=guardrail.get("guardrail_name", ""), 

19 event_hook=litellm_params.mode, 

20 default_on=litellm_params.default_on, 

21 ) 

22 

23 litellm.logging_callback_manager.add_litellm_callback(_alice_guardrail_callback) 

24 return _alice_guardrail_callback 

25 

26 

27guardrail_initializer_registry: Final = { # mutable-ok: module-level registry, built once and never mutated 

28 SupportedGuardrailIntegrations.ALICE.value: initialize_guardrail, 

29} 

30 

31 

32guardrail_class_registry: Final = { # mutable-ok: module-level registry, built once and never mutated 

33 SupportedGuardrailIntegrations.ALICE.value: AliceGuardrail, 

34}