phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-01 16:24:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

new build msg from template

Browse source
This commit is contained in:
Xi Yan 2024-11-18 22:22:58 -08:00
parent d25cae3995
commit f967a740be
1 changed files with 11 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

12
llama_stack/cli/stack/build.py
View file

@ -8,6 +8,7 @@ import argparse
from llama_stack.cli.subcommand import Subcommand
from llama_stack.distribution.datatypes import * # noqa: F403
import importlib
import os
import shutil
from functools import lru_cache
@ -285,8 +286,17 @@ class StackBuild(Subcommand):
os.makedirs(build_dir, exist_ok=True)
run_config_file = build_dir / f"{build_config.name}-run.yaml"
shutil.copy(template_path, run_config_file)
module_name = f"llama_stack.templates.{template_name}"
module = importlib.import_module(module_name)
distribution_template = module.get_distribution_template()
cprint("Build Successful! Next steps: ", color="green")
env_vars = ", ".join(distribution_template.run_config_env_vars.keys())
cprint(
f"You can now edit {run_config_file} and run `llama stack run {run_config_file}`",
f" 1. Set the environment variables: {env_vars}",
color="green",
)
cprint(
f" 2. `llama stack run {run_config_file}`",
color="green",
)
else: