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

1from typing import TYPE_CHECKING, Final 

2 

3from litellm.types.guardrails import SupportedGuardrailIntegrations 

4 

5from .qualifire import QualifireGuardrail 

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 _qualifire_callback: Final = QualifireGuardrail( 

15 api_key=litellm_params.api_key, 

16 api_base=litellm_params.api_base, 

17 evaluation_id=getattr(litellm_params, "evaluation_id", None), 

18 prompt_injections=getattr(litellm_params, "prompt_injections", None), 

19 hallucinations_check=getattr(litellm_params, "hallucinations_check", None), 

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

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

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

23 tool_selection_quality_check=getattr(litellm_params, "tool_selection_quality_check", None), 

24 assertions=getattr(litellm_params, "assertions", None), 

25 on_flagged=getattr(litellm_params, "on_flagged", "block"), 

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

27 event_hook=litellm_params.mode, 

28 default_on=litellm_params.default_on, 

29 ) 

30 

31 litellm.logging_callback_manager.add_litellm_callback(_qualifire_callback) 

32 

33 return _qualifire_callback 

34 

35 

36guardrail_initializer_registry: Final = { 

37 SupportedGuardrailIntegrations.QUALIFIRE.value: initialize_guardrail, 

38} 

39 

40 

41guardrail_class_registry: Final = { 

42 SupportedGuardrailIntegrations.QUALIFIRE.value: QualifireGuardrail, 

43}