phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-17 05:42:37 +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", "transformers",
"torch --index-url https://download.pytorch.org/whl/cpu", "torch --index-url https://download.pytorch.org/whl/cpu",
], ],
module="llama_stack.providers.inline.meta_reference.safety", module="llama_stack.providers.inline.safety.meta_reference",
config_class="llama_stack.providers.inline.meta_reference.safety.SafetyConfig", config_class="llama_stack.providers.inline.safety.meta_reference.SafetyConfig",
api_dependencies=[ api_dependencies=[
Api.inference, Api.inference,
], ],
@ -54,8 +54,8 @@ def available_providers() -> List[ProviderSpec]:
pip_packages=[ pip_packages=[
"codeshield", "codeshield",
], ],
module="llama_stack.providers.inline.meta_reference.codeshield", module="llama_stack.providers.inline.safety.meta_reference",
config_class="llama_stack.providers.inline.meta_reference.codeshield.CodeShieldConfig", config_class="llama_stack.providers.inline.safety.meta_reference.CodeShieldConfig",
api_dependencies=[], api_dependencies=[],
), ),
] ]

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

@ -8,7 +8,7 @@ import pytest
import pytest_asyncio import pytest_asyncio
from llama_stack.distribution.datatypes import Api, Provider 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, LlamaGuardShieldConfig,
SafetyConfig, SafetyConfig,
) )