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

8 statements  

« prev     ^ index     » next       coverage.py v7.15.2, created at 2026-10-10 12:01 +0000

1from typing import Final 

2 

3import litellm 

4from litellm.types.guardrails import Guardrail, LitellmParams 

5 

6 

7def initialize_guardrail(litellm_params: LitellmParams, guardrail: Guardrail): 

8 from litellm.proxy.guardrails.guardrail_hooks.tool_policy.tool_policy_guardrail import ( 

9 ToolPolicyGuardrail, 

10 ) 

11 

12 _callback: Final = ToolPolicyGuardrail( 

13 guardrail_name=guardrail.get("guardrail_name", "tool_policy"), 

14 event_hook=litellm_params.mode, 

15 default_on=litellm_params.default_on, 

16 ) 

17 litellm.logging_callback_manager.add_litellm_callback(_callback) 

18 return _callback