Update doc templates for running safety on self-hosted templates (#874)

This commit is contained in:
Hardik Shah 2025-01-24 11:28:20 -08:00 committed by GitHub
parent eaba6a550a
commit 2cebb24d3a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
12 changed files with 140 additions and 37 deletions

View file

@ -18,10 +18,16 @@ from llama_stack.templates.template import DistributionTemplate, RunConfigSettin
def get_distribution_template() -> DistributionTemplate:
providers = {
"inference": ["remote::sambanova"],
"memory": ["inline::faiss", "remote::chromadb", "remote::pgvector"],
"vector_io": ["inline::faiss", "remote::chromadb", "remote::pgvector"],
"safety": ["inline::llama-guard"],
"agents": ["inline::meta-reference"],
"telemetry": ["inline::meta-reference"],
"tool_runtime": [
"remote::brave-search",
"remote::tavily-search",
"inline::code-interpreter",
"inline::rag-runtime",
],
}
inference_provider = Provider(