phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-16 06:53:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add deprecation_error pointing meta-reference -> inline::llama-guard

Browse source
This commit is contained in:
Ashwin Bharambe 2024-11-07 15:23:15 -08:00
parent fdaec91747
commit 984ba074e1
9 changed files with 84 additions and 74 deletions
Download patch file Download diff file Expand all files Collapse all files

15
llama_stack/providers/registry/safety.py
View file

@ -29,11 +29,18 @@ def available_providers() -> List[ProviderSpec]:
api_dependencies=[
Api.inference,
],
deprecation_warning="Please use the `llama-guard` / `prompt-guard` / `code-scanner` providers instead.",
deprecation_error="""
Provider `meta-reference` for API `safety` does not work with the latest Llama Stack.
- if you are using Llama Guard v3, please use the `inline::llama-guard` provider instead.
- if you are using Prompt Guard, please use the `inline::prompt-guard` provider instead.
- if you are using Code Scanner, please use the `inline::code-scanner` provider instead.
""",
),
InlineProviderSpec(
api=Api.safety,
provider_type="llama-guard",
provider_type="inline::llama-guard",
pip_packages=[],
module="llama_stack.providers.inline.safety.llama_guard",
config_class="llama_stack.providers.inline.safety.llama_guard.LlamaGuardConfig",
@ -43,7 +50,7 @@ def available_providers() -> List[ProviderSpec]:
),
InlineProviderSpec(
api=Api.safety,
provider_type="prompt-guard",
provider_type="inline::prompt-guard",
pip_packages=[
"transformers",
"torch --index-url https://download.pytorch.org/whl/cpu",
@ -56,7 +63,7 @@ def available_providers() -> List[ProviderSpec]:
),
InlineProviderSpec(
api=Api.safety,
provider_type="code-scanner",
provider_type="inline::code-scanner",
pip_packages=[
"codeshield",
],