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

test safety against safety client

Browse source
This commit is contained in:
Ashwin Bharambe 2024-09-20 14:55:00 -07:00
parent 6e0f283f52
commit 9e16b0948b
19 changed files with 1076 additions and 10754 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -6,12 +6,7 @@
from typing import List
from llama_stack.distribution.datatypes import (
Api,
InlineProviderSpec,
ProviderSpec,
remote_provider_spec,
)
from llama_stack.distribution.datatypes import * # noqa: F403
def available_providers() -> List[ProviderSpec]:
@ -28,15 +23,4 @@ def available_providers() -> List[ProviderSpec]:
module="llama_stack.providers.impls.meta_reference.safety",
config_class="llama_stack.providers.impls.meta_reference.safety.SafetyConfig",
),
remote_provider_spec(
api=Api.safety,
adapter=AdapterSpec(
adapter_id="bedrock",
pip_packages=[
"aws-sdk",
],
module="llama_stack.providers.adapters.safety.bedrock",
header_extractor="llama_stack.providers.adapters.safety.bedrock.BedrockSafetyRequestProviderData",
),
),
]