From e4d399b1f27d7358f8ed89bead1ea451cea9e8bc Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Wed, 18 Sep 2024 14:20:45 -0700 Subject: [PATCH] fix docker --- llama_stack/cli/stack/build.py | 8 +++++++- llama_stack/distribution/configure_container.sh | 2 +- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/llama_stack/cli/stack/build.py b/llama_stack/cli/stack/build.py index d2c1bf324..f35c6ceab 100644 --- a/llama_stack/cli/stack/build.py +++ b/llama_stack/cli/stack/build.py @@ -118,8 +118,14 @@ class StackBuild(Subcommand): 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}") + ) cprint( - f"You may now run `llama stack configure {build_config.name}` or `llama stack configure {str(build_file_path)}`", + f"You may now run `llama stack configure {configure_name}` or `llama stack configure {str(build_file_path)}`", color="green", ) diff --git a/llama_stack/distribution/configure_container.sh b/llama_stack/distribution/configure_container.sh index 49ef439a9..68009f8b1 100755 --- a/llama_stack/distribution/configure_container.sh +++ b/llama_stack/distribution/configure_container.sh @@ -28,4 +28,4 @@ set -x podman run -it \ -v $host_build_dir:$container_build_dir \ $docker_image \ - llama stack configure --config ./llamastack-build.yaml --output-dir $container_build_dir + llama stack configure ./llamastack-build.yaml --output-dir $container_build_dir