Spaces:
Running
Running
param-bharat
commited on
Commit
·
2a93403
1
Parent(s):
993988f
feat: add secrets detection to package and update imports
Browse files
guardrails_genie/guardrails/__init__.py
CHANGED
@@ -1,13 +1,14 @@
|
|
1 |
-
from .
|
2 |
-
PromptInjectionClassifierGuardrail,
|
3 |
-
PromptInjectionSurveyGuardrail,
|
4 |
-
)
|
5 |
-
from .entity_recognition import (
|
6 |
PresidioEntityRecognitionGuardrail,
|
7 |
RegexEntityRecognitionGuardrail,
|
8 |
TransformersEntityRecognitionGuardrail,
|
9 |
RestrictedTermsJudge,
|
10 |
)
|
|
|
|
|
|
|
|
|
|
|
11 |
from .manager import GuardrailManager
|
12 |
|
13 |
__all__ = [
|
@@ -18,4 +19,5 @@ __all__ = [
|
|
18 |
"TransformersEntityRecognitionGuardrail",
|
19 |
"RestrictedTermsJudge",
|
20 |
"GuardrailManager",
|
|
|
21 |
]
|
|
|
1 |
+
from guardrails_genie.guardrails.entity_recognition import (
|
|
|
|
|
|
|
|
|
2 |
PresidioEntityRecognitionGuardrail,
|
3 |
RegexEntityRecognitionGuardrail,
|
4 |
TransformersEntityRecognitionGuardrail,
|
5 |
RestrictedTermsJudge,
|
6 |
)
|
7 |
+
from guardrails_genie.guardrails.injection import (
|
8 |
+
PromptInjectionClassifierGuardrail,
|
9 |
+
PromptInjectionSurveyGuardrail,
|
10 |
+
)
|
11 |
+
from guardrails_genie.guardrails.secrets_detection import SecretsDetectionGuardrail
|
12 |
from .manager import GuardrailManager
|
13 |
|
14 |
__all__ = [
|
|
|
19 |
"TransformersEntityRecognitionGuardrail",
|
20 |
"RestrictedTermsJudge",
|
21 |
"GuardrailManager",
|
22 |
+
"SecretsDetectionGuardrail",
|
23 |
]
|