phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-03 14:32:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat add sambanova safety adaptor

Browse source
This commit is contained in:
jhpiedrahitao 2025-03-25 14:56:49 -05:00
parent 8783dd8162
commit ed0d5a4df1
10 changed files with 165 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

14
llama_stack/templates/sambanova/sambanova.py
View file

@ -34,7 +34,7 @@ def get_distribution_template() -> DistributionTemplate:
providers = {
"inference": ["remote::sambanova", "inline::sentence-transformers"],
"vector_io": ["inline::faiss", "remote::chromadb", "remote::pgvector"],
"safety": ["inline::llama-guard"],
"safety": ["remote::sambanova"],
"agents": ["inline::meta-reference"],
"telemetry": ["inline::meta-reference"],
"tool_runtime": [
@ -115,7 +115,7 @@ def get_distribution_template() -> DistributionTemplate:
return DistributionTemplate(
name=name,
distro_type="self_hosted",
description="Use SambaNova for running LLM inference",
description="Use SambaNova for running LLM inference and safety",
container_image=None,
template_path=Path(__file__).parent / "doc_template.md",
providers=providers,
@ -127,7 +127,15 @@ def get_distribution_template() -> DistributionTemplate:
"vector_io": vector_io_providers,
},
default_models=default_models + [embedding_model],
default_shields=[ShieldInput(shield_id="meta-llama/Llama-Guard-3-8B")],
default_shields=[
ShieldInput(
shield_id="meta-llama/Llama-Guard-3-8B", provider_shield_id="sambanova/Meta-Llama-Guard-3-8B"
),
ShieldInput(
shield_id="sambanova/Meta-Llama-Guard-3-8B",
provider_shield_id="sambanova/Meta-Llama-Guard-3-8B",
),
],
default_tool_groups=default_tool_groups,
),
},