From 25adc83de8329c4bedff470d9d38287e7e6bcbf0 Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Tue, 17 Sep 2024 19:56:58 -0700 Subject: [PATCH] Fix for safety --- docs/cli_reference.md | 2 +- .../meta_reference/safety/shields/contrib/third_party_shield.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/cli_reference.md b/docs/cli_reference.md index f66c1e4cb..56fa3736e 100644 --- a/docs/cli_reference.md +++ b/docs/cli_reference.md @@ -500,7 +500,7 @@ You know what's even more hilarious? People like you who think they can just Goo Similarly you can test safety (if you configured llama-guard and/or prompt-guard shields) by: ``` -python -m llama_stack.safety.client localhost 5000 +python -m llama_stack.apis.safety.client localhost 5000 ``` You can find more example scripts with client SDKs to talk with the Llama Stack server in our [llama-stack-apps](https://github.com/meta-llama/llama-stack-apps/tree/main/sdk_examples) repo. diff --git a/llama_stack/providers/impls/meta_reference/safety/shields/contrib/third_party_shield.py b/llama_stack/providers/impls/meta_reference/safety/shields/contrib/third_party_shield.py index 9aa8adea8..cc652ae63 100644 --- a/llama_stack/providers/impls/meta_reference/safety/shields/contrib/third_party_shield.py +++ b/llama_stack/providers/impls/meta_reference/safety/shields/contrib/third_party_shield.py @@ -8,7 +8,7 @@ from typing import List from llama_models.llama3.api.datatypes import Message -from llama_stack.safety.meta_reference.shields.base import ( +from llama_stack.providers.impls.meta_reference.safety.shields.base import ( OnViolationAction, ShieldBase, ShieldResponse,