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)) 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",
) )

View file

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