phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-11 21:48:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update prompt-shield to reflect latest changes in agentic

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

2
toolchain/safety/shields/__init__.py
View file

@ -11,7 +11,7 @@ from .base import ( # noqa: F401
from .code_scanner import CodeScannerShield # noqa: F401
from .contrib.third_party_shield import ThirdPartyShield # noqa: F401
from .llama_guard import LlamaGuardShield # noqa: F401
from .prompt_guard import PromptGuardShield # noqa: F401
from .prompt_guard import InjectionShield, JailbreakShield, PromptGuardShield # noqa: F401
from .shield_runner import SafetyException, ShieldRunnerMixin # noqa: F401
transformers.logging.set_verbosity_error()