phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-04 20:14:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

adding safety adapter for Together

Browse source
This commit is contained in:
Yogish Baliga 2024-09-20 09:35:01 -07:00
parent e617273d8c
commit 9f3300df25
6 changed files with 243 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -7,6 +7,7 @@
from typing import List
from llama_stack.distribution.datatypes import * # noqa: F403
from llama_stack.distribution.datatypes import Api, InlineProviderSpec, ProviderSpec, remote_provider_spec, AdapterSpec
def available_providers() -> List[ProviderSpec]:
@ -30,6 +31,20 @@ def available_providers() -> List[ProviderSpec]:
pip_packages=[],
module="llama_stack.providers.adapters.safety.sample",
config_class="llama_stack.providers.adapters.safety.sample.SampleConfig",
)
),
remote_provider_spec(
api=Api.safety,
adapter=AdapterSpec(
adapter_id="together",
pip_packages=[
"together",
],
module="llama_stack.providers.adapters.safety.together",
config_class="llama_stack.providers.adapters.safety.together.TogetherSafetyConfig",
),
),
]