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

1from typing import TYPE_CHECKING, Final 

2 

3from litellm.types.guardrails import SupportedGuardrailIntegrations 

4 

5from .singulr import SingulrGuardrail 

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( 

12 litellm_params: "LitellmParams", 

13 guardrail: "Guardrail", 

14): 

15 import litellm 

16 

17 _cb: Final = SingulrGuardrail( 

18 singulr_api_base=getattr(litellm_params, "singulr_api_base", None) or litellm_params.api_base, 

19 singulr_api_key=getattr(litellm_params, "singulr_api_key", None) or litellm_params.api_key, 

20 singulr_application_id=getattr(litellm_params, "singulr_application_id", None), 

21 singulr_guardrail_id=getattr(litellm_params, "singulr_guardrail_id", None), 

22 block_on_error=getattr(litellm_params, "block_on_error", None), 

23 timeout=litellm_params.timeout, 

24 guardrail_name=guardrail.get( 

25 "guardrail_name", 

26 "", 

27 ), 

28 event_hook=litellm_params.mode, 

29 default_on=litellm_params.default_on, 

30 ) 

31 litellm.logging_callback_manager.add_litellm_callback( 

32 _cb, 

33 ) 

34 

35 return _cb 

36 

37 

38guardrail_initializer_registry: Final = { 

39 SupportedGuardrailIntegrations.SINGULR.value: initialize_guardrail, 

40} 

41 

42guardrail_class_registry: Final = { 

43 SupportedGuardrailIntegrations.SINGULR.value: SingulrGuardrail, 

44}