phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-28 16:51:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: add api.llama provider, llama-guard-4 model

Browse source
This commit is contained in:
Ashwin Bharambe 2025-04-29 09:56:46 -07:00
parent 934446ddb4
commit 38b580db02
21 changed files with 1526 additions and 47 deletions
Download patch file Download diff file Expand all files Collapse all files

9
llama_stack/cli/model/remove.py
View file

@ -42,11 +42,12 @@ class ModelRemove(Subcommand):
)
def _run_model_remove_cmd(self, args: argparse.Namespace) -> None:
from .safety_models import prompt_guard_model_sku
from .safety_models import prompt_guard_model_sku_map
prompt_guard = prompt_guard_model_sku()
if args.model == prompt_guard.model_id:
model = prompt_guard
prompt_guard_model_map = prompt_guard_model_sku_map()
if args.model in prompt_guard_model_map.keys():
model = prompt_guard_model_map[args.model]
else:
model = resolve_model(args.model)