update prompt-shield to reflect latest changes in agentic

This commit is contained in:
Hardik Shah 2024-07-19 18:12:09 -07:00
parent ce0804556b
commit 9c9b834c0f
3 changed files with 45 additions and 5 deletions

View file

@ -16,6 +16,8 @@ class BuiltinShield(Enum):
prompt_guard = "prompt_guard"
code_scanner_guard = "code_scanner_guard"
third_party_shield = "third_party_shield"
injection_shield = "injection_shield"
jailbreak_shield = "jailbreak_shield"
ShieldType = Union[BuiltinShield, str]