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

1from .enkryptai import EnkryptAIGuardrails 

2 

3__all__ = ["EnkryptAIGuardrails"] 

4 

5 

6from typing import TYPE_CHECKING, Final 

7 

8from litellm.types.guardrails import SupportedGuardrailIntegrations 

9 

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

11 from litellm.types.guardrails import Guardrail, LitellmParams 

12 

13 

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

15 import litellm 

16 

17 _enkryptai_callback: Final = EnkryptAIGuardrails( 

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

19 api_key=litellm_params.api_key, 

20 api_base=litellm_params.api_base, 

21 policy_name=litellm_params.policy_name, 

22 deployment_name=litellm_params.deployment_name, 

23 detectors=litellm_params.detectors, 

24 block_on_violation=litellm_params.block_on_violation, 

25 event_hook=litellm_params.mode, 

26 default_on=litellm_params.default_on, 

27 ) 

28 litellm.logging_callback_manager.add_litellm_callback(_enkryptai_callback) 

29 

30 return _enkryptai_callback 

31 

32 

33guardrail_initializer_registry: Final = { 

34 SupportedGuardrailIntegrations.ENKRYPTAI.value: initialize_guardrail, 

35} 

36 

37 

38guardrail_class_registry: Final = { 

39 SupportedGuardrailIntegrations.ENKRYPTAI.value: EnkryptAIGuardrails, 

40}