diff --git a/llama_stack/cli/stack/build.py b/llama_stack/cli/stack/build.py index c5dea1db..dea70562 100644 --- a/llama_stack/cli/stack/build.py +++ b/llama_stack/cli/stack/build.py @@ -191,7 +191,13 @@ def _run_stack_build_command(self, args: argparse.Namespace) -> None: return if not args.config and not args.template: - name = prompt("> Enter a name for your Llama Stack (e.g. my-local-stack): ") + if not args.name: + name = prompt( + "> Enter a name for your Llama Stack (e.g. my-local-stack): " + ) + else: + name = args.name + image_type = prompt( "> Enter the image type you want your Llama Stack to be built as (docker or conda): ", validator=Validator.from_callable(