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

reorder output msg

Browse source
This commit is contained in:
Xi Yan 2024-09-26 23:28:42 -07:00
parent 3b807912d2
commit ecd17ce9e9
2 changed files with 14 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -112,24 +112,20 @@ class StackBuild(Subcommand):
to_write = json.loads(json.dumps(build_config.dict(), cls=EnumEncoder)) to_write = json.loads(json.dumps(build_config.dict(), cls=EnumEncoder))
f.write(yaml.dump(to_write, sort_keys=False)) f.write(yaml.dump(to_write, sort_keys=False))
return_code = build_image(build_config, build_file_path)
if return_code != 0:
return
cprint( cprint(
f"Build spec configuration saved at {str(build_file_path)}", f"Build spec configuration saved at {str(build_file_path)}",
color="blue", color="blue",
) )
configure_name = ( return_code = build_image(build_config, build_file_path)
build_config.name if return_code != 0:
if build_config.image_type == "conda" return
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 {build_config.name}`",
color="green", 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

9
llama_stack/cli/stack/configure.py
View file

@ -142,7 +142,8 @@ class StackConfigure(Subcommand):
color="blue", color="blue",
) )
cprint( if build_config.image_type == "conda":
f"You can now run `llama stack run {image_name} --port PORT`", cprint(
color="green", f"You can now run `llama stack run {image_name} --port PORT`",
) color="green",
)