From 6302a1ee90f00cd4eb56f8a1856d2ab4df98cf2b Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Wed, 18 Sep 2024 23:48:31 -0700 Subject: [PATCH] fix prompt with name args (#80) --- 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 c5dea1db8..dea705628 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(