phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 15:23:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update output msg

Browse source
This commit is contained in:
Xi Yan 2024-10-02 12:56:16 -07:00
parent 7fadfa7435
commit 5d3403202e
1 changed files with 18 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -114,10 +114,10 @@ class StackBuild(Subcommand):
# save build.yaml spec for building same distribution again # save build.yaml spec for building same distribution again
if build_config.image_type == ImageType.docker.value: if build_config.image_type == ImageType.docker.value:
# docker needs build file to be in the llama-stack repo dir to be able to copy over to the image # docker needs build file to be in the llama-stack repo dir to be able to copy over to the image
llama_stack_path = Path(os.path.abspath(__file__)).parent.parent.parent.parent llama_stack_path = Path(
build_dir = ( os.path.abspath(__file__)
llama_stack_path / "tmp/configs/" ).parent.parent.parent.parent
) build_dir = llama_stack_path / "tmp/configs/"
else: else:
build_dir = DISTRIBS_BASE_DIR / f"llamastack-{build_config.name}" build_dir = DISTRIBS_BASE_DIR / f"llamastack-{build_config.name}"
@ -137,10 +137,16 @@ class StackBuild(Subcommand):
if build_config.image_type == "conda" if build_config.image_type == "conda"
else (f"llamastack-{build_config.name}") else (f"llamastack-{build_config.name}")
) )
if build_config.image_type == "conda":
cprint( cprint(
f"You can now run `llama stack configure {configure_name}`", f"You can now run `llama stack configure {configure_name}`",
color="green", color="green",
) )
else:
cprint(
f"You can now run `llama stack run {build_config.name}` or `llama stack configure {configure_name} to re-configure docker run config.`",
color="green",
)
def _run_template_list_cmd(self, args: argparse.Namespace) -> None: def _run_template_list_cmd(self, args: argparse.Namespace) -> None:
import json import json
@ -212,7 +218,10 @@ class StackBuild(Subcommand):
if args.name: if args.name:
maybe_build_config = self._get_build_config_from_name(args) maybe_build_config = self._get_build_config_from_name(args)
if maybe_build_config: if maybe_build_config:
cprint(f"Building from existing build config for {args.name} in {str(maybe_build_config)}...", "green") cprint(
f"Building from existing build config for {args.name} in {str(maybe_build_config)}...",
"green",
)
with open(maybe_build_config, "r") as f: with open(maybe_build_config, "r") as f:
build_config = BuildConfig(**yaml.safe_load(f)) build_config = BuildConfig(**yaml.safe_load(f))
self._run_stack_build_command_from_build_config(build_config) self._run_stack_build_command_from_build_config(build_config)