Coverage for lintro / ai / enums / sanitize_mode.py: 100%

1"""Sanitize mode for prompt injection detection.""" 

2 

3from __future__ import annotations 

4 

5from enum import StrEnum, auto 

6 

7 

8class SanitizeMode(StrEnum): 

9 """Controls how detected prompt injection patterns are handled. 

10 

11 Attributes: 

12 WARN: Log a warning but continue processing (default). 

13 BLOCK: Skip fix generation for files with detected patterns. 

14 OFF: Disable injection pattern detection entirely. 

15 """ 

16 

17 WARN = auto() 

18 BLOCK = auto() 

19 OFF = auto()