forked from phoenix-oss/llama-stack-mirror
34ab7a3b6c
Lint check in main branch is failing. This fixes the lint check after we moved to ruff in https://github.com/meta-llama/llama-stack/pull/921. We need to move to a `ruff.toml` file as well as fixing and ignoring some additional checks. Signed-off-by: Yuan Tang <terrytangyuan@gmail.com>
17 lines
567 B
Markdown
17 lines
567 B
Markdown
## Safety Guardrails
|
|
|
|
Safety is a critical component of any AI application. Llama Stack provides a Shield system that can be applied at multiple touchpoints:
|
|
|
|
```python
|
|
# Register a safety shield
|
|
shield_id = "content_safety"
|
|
client.shields.register(shield_id=shield_id, provider_shield_id="llama-guard-basic")
|
|
|
|
# Run content through shield
|
|
response = client.safety.run_shield(
|
|
shield_id=shield_id, messages=[{"role": "user", "content": "User message here"}]
|
|
)
|
|
|
|
if response.violation:
|
|
print(f"Safety violation detected: {response.violation.user_message}")
|
|
```
|