From e91ee754972a35f3866ce182c79bd5f258a5947a Mon Sep 17 00:00:00 2001 From: Nathan Weinberg Date: Tue, 18 Mar 2025 22:39:49 -0400 Subject: [PATCH] feat: add additional logging to llama stack build Partial revert of fa68ded07c5a6469f113b016a335f355a94ed504 this commit ensures users know where their new templates are generated and how to run the newly built distro locally Signed-off-by: Nathan Weinberg --- llama_stack/cli/stack/_build.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/llama_stack/cli/stack/_build.py b/llama_stack/cli/stack/_build.py index 2787a93d5..f3a29b947 100644 --- a/llama_stack/cli/stack/_build.py +++ b/llama_stack/cli/stack/_build.py @@ -19,7 +19,7 @@ import yaml from prompt_toolkit import prompt from prompt_toolkit.completion import WordCompleter from prompt_toolkit.validation import Validator -from termcolor import cprint +from termcolor import colored, cprint from llama_stack.cli.stack.utils import ImageType from llama_stack.cli.table import print_table @@ -389,6 +389,11 @@ def _run_stack_build_command_from_build_config( shutil.copy(path, run_config_file) cprint("Build Successful!", color="green") + cprint("You can find the newly-built template here: " + colored(template_path, "light_blue")) + cprint( + "You can run the new Llama Stack distro via: " + + colored(f"llama stack run {template_path} --image-type {build_config.image_type}", "light_blue") + ) return template_path else: return _generate_run_config(build_config, build_dir, image_name)