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 .injection import (
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
  ]