From d2d7cde77e84d06c471099e4783c357b53a9c385 Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Wed, 18 Sep 2024 23:44:14 -0700 Subject: [PATCH] fix prompt with name args --- llama_stack/cli/stack/build.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/llama_stack/cli/stack/build.py b/llama_stack/cli/stack/build.py index 0393fb708..d085f392b 100644 --- a/llama_stack/cli/stack/build.py +++ b/llama_stack/cli/stack/build.py @@ -191,7 +191,13 @@ class StackBuild(Subcommand): 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(