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 1

safety API cleanup part 1

Browse source 
Sample adapter implementation for Bedrock implementation of Guardrails
This commit is contained in:
Ashwin Bharambe 2024-09-20 10:57:26 -07:00 • committed by Xi Yan
parent 32beecb20d
commit 7e40eead4e
7 changed files with 130 additions and 78 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -6,7 +6,12 @@
from typing import List
from llama_stack.distribution.datatypes import Api, InlineProviderSpec, ProviderSpec
from llama_stack.distribution.datatypes import (
Api,
InlineProviderSpec,
ProviderSpec,
remote_provider_spec,
)
def available_providers() -> List[ProviderSpec]:
@ -23,4 +28,15 @@ 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",
),
),
]