Coverage for .venv/lib/python3.13/site-packages/litellm/proxy/policy_engine/__init__.py: 100%

1""" 

2LiteLLM Policy Engine 

3 

4The Policy Engine allows administrators to define policies that combine guardrails 

5with scoping rules. Policies can target specific teams, API keys, and models using 

6wildcard patterns, and support inheritance from base policies. 

7 

8Configuration structure: 

9- `policies`: Define WHAT guardrails to apply (with inheritance and conditions) 

10- `policy_attachments`: Define WHERE policies apply (teams, keys, models) 

11 

12Example: 

13```yaml 

14policies: 

15 global-baseline: 

16 description: "Base guardrails for all requests" 

17 guardrails: 

18 add: [pii_blocker] 

19 

20 gpt4-safety: 

21 inherit: global-baseline 

22 description: "Extra safety for GPT-4" 

23 guardrails: 

24 add: [toxicity_filter] 

25 condition: 

26 model: "gpt-4.*" # regex pattern 

27 

28policy_attachments: 

29 - policy: global-baseline 

30 scope: "*" 

31 - policy: gpt4-safety 

32 scope: "*" 

33``` 

34""" 

35 

36from litellm.proxy.policy_engine.attachment_registry import ( 

37 AttachmentRegistry, 

38 get_attachment_registry, 

39) 

40from litellm.proxy.policy_engine.condition_evaluator import ConditionEvaluator 

41from litellm.proxy.policy_engine.policy_matcher import PolicyMatcher 

42from litellm.proxy.policy_engine.policy_registry import ( 

43 PolicyRegistry, 

44 get_policy_registry, 

45) 

46from litellm.proxy.policy_engine.policy_resolver import PolicyResolver 

47from litellm.proxy.policy_engine.policy_validator import PolicyValidator 

48 

49__all__ = [ 

50 "AttachmentRegistry", 

51 "ConditionEvaluator", 

52 "PolicyMatcher", 

53 "PolicyRegistry", 

54 "PolicyResolver", 

55 "PolicyValidator", 

56 "get_attachment_registry", 

57 "get_policy_registry", 

58]