reorder output msg

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

View file

@ -112,22 +112,18 @@ class StackBuild(Subcommand):
to_write = json.loads(json.dumps(build_config.dict(), cls=EnumEncoder))
f.write(yaml.dump(to_write, sort_keys=False))
return_code = build_image(build_config, build_file_path)
if return_code != 0:
return
cprint(
f"Build spec configuration saved at {str(build_file_path)}",
color="blue",
)
configure_name = (
build_config.name
if build_config.image_type == "conda"
else (f"llamastack-{build_config.name}")
)
return_code = build_image(build_config, build_file_path)
if return_code != 0:
return
if build_config.image_type == "conda":
cprint(
f"You can now run `llama stack configure {configure_name}`",
f"You can now run `llama stack configure {build_config.name}`",
color="green",
)

View file

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