File size: 769 Bytes
f00294f
 
28d8897
f00294f
 
6779bd2
 
 
f00294f
 
6779bd2
f00294f
 
78a1bf0
 
 
 
 
3ad3f59
 
 
 
78a1bf0
3ad3f59
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
import warnings

from .llm_judge_entity_recognition_guardrail import RestrictedTermsJudge

try:
    from .presidio_entity_recognition_guardrail import (
        PresidioEntityRecognitionGuardrail,
    )
except ImportError:
    warnings.warn(
        "Presidio is not installed. If you want to use `PresidioEntityRecognitionGuardrail`, you can install the required packages using `pip install -e .[presidio]`"
    )

from .regex_entity_recognition_guardrail import RegexEntityRecognitionGuardrail
from .transformers_entity_recognition_guardrail import (
    TransformersEntityRecognitionGuardrail,
)

__all__ = [
    "PresidioEntityRecognitionGuardrail",
    "RegexEntityRecognitionGuardrail",
    "TransformersEntityRecognitionGuardrail",
    "RestrictedTermsJudge",
]