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

move the meta_reference safety dir also

Browse source
This commit is contained in:
Ashwin Bharambe 2024-11-07 13:02:13 -08:00
parent ec603a435a
commit be6fe0fd77
8 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

0
llama_stack/providers/inline/meta_reference/safety/__init__.py → llama_stack/providers/inline/safety/meta_reference/__init__.py
View file

0
llama_stack/providers/inline/meta_reference/safety/base.py → llama_stack/providers/inline/safety/meta_reference/base.py
View file

0
llama_stack/providers/inline/meta_reference/safety/config.py → llama_stack/providers/inline/safety/meta_reference/config.py
View file

0
llama_stack/providers/inline/meta_reference/safety/llama_guard.py → llama_stack/providers/inline/safety/meta_reference/llama_guard.py
View file

0
llama_stack/providers/inline/meta_reference/safety/prompt_guard.py → llama_stack/providers/inline/safety/meta_reference/prompt_guard.py
View file

0
llama_stack/providers/inline/meta_reference/safety/safety.py → llama_stack/providers/inline/safety/meta_reference/safety.py
View file

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

@ -24,8 +24,8 @@ def available_providers() -> List[ProviderSpec]:
"transformers",
"torch --index-url https://download.pytorch.org/whl/cpu",
],
module="llama_stack.providers.inline.meta_reference.safety",
config_class="llama_stack.providers.inline.meta_reference.safety.SafetyConfig",
module="llama_stack.providers.inline.safety.meta_reference",
config_class="llama_stack.providers.inline.safety.meta_reference.SafetyConfig",
api_dependencies=[
Api.inference,
],
@ -54,8 +54,8 @@ def available_providers() -> List[ProviderSpec]:
pip_packages=[
"codeshield",
],
module="llama_stack.providers.inline.meta_reference.codeshield",
config_class="llama_stack.providers.inline.meta_reference.codeshield.CodeShieldConfig",
module="llama_stack.providers.inline.safety.meta_reference",
config_class="llama_stack.providers.inline.safety.meta_reference.CodeShieldConfig",
api_dependencies=[],
),
]

2
llama_stack/providers/tests/safety/fixtures.py
View file

@ -8,7 +8,7 @@ import pytest
import pytest_asyncio
from llama_stack.distribution.datatypes import Api, Provider
from llama_stack.providers.inline.meta_reference.safety import (
from llama_stack.providers.inline.safety.meta_reference import (
LlamaGuardShieldConfig,
SafetyConfig,
)