phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 02:53:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix fireworks run-with-safety template (#766)

Browse source 
Summary:
Fixing issue reported in
https://github.com/meta-llama/llama-stack/pull/755/files#r1915696188

Test Plan:
Re-run the config gen
```
pip install .
python3 llama_stack/scripts/distro_codegen.py
```
This commit is contained in:
Vladimir Ivić 2025-01-14 15:28:55 -08:00 • committed by GitHub
parent 472feea8d4
commit 89e3f81520
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 27 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

22
llama_stack/templates/fireworks/fireworks.py
View file

@ -71,14 +71,6 @@ def get_distribution_template() -> DistributionTemplate:
)
for m in MODEL_ALIASES
]
inference_model = ModelInput(
model_id="${env.INFERENCE_MODEL}",
provider_id="fireworks",
)
safety_model = ModelInput(
model_id="${env.SAFETY_MODEL}",
provider_id="fireworks",
)
embedding_model = ModelInput(
model_id="all-MiniLM-L6-v2",
provider_id="sentence-transformers",
@ -133,6 +125,11 @@ def get_distribution_template() -> DistributionTemplate:
provider_type="inline::llama-guard",
config={},
),
Provider(
provider_id="llama-guard-vision",
provider_type="inline::llama-guard",
config={},
),
Provider(
provider_id="code-scanner",
provider_type="inline::code-scanner",
@ -141,15 +138,18 @@ def get_distribution_template() -> DistributionTemplate:
],
},
default_models=[
inference_model,
safety_model,
*default_models,
embedding_model,
],
default_shields=[
ShieldInput(
shield_id="${env.SAFETY_MODEL}",
shield_id="meta-llama/Llama-Guard-3-8B",
provider_id="llama-guard",
),
ShieldInput(
shield_id="meta-llama/Llama-Guard-3-11B-Vision",
provider_id="llama-guard-vision",
),
ShieldInput(
shield_id="CodeScanner",
provider_id="code-scanner",