From 53721e91ad39b37e7cfa496870d15f122202c090 Mon Sep 17 00:00:00 2001 From: Yuan Tang Date: Wed, 29 Jan 2025 12:36:52 -0500 Subject: [PATCH] Fix validator of "container" image type (#901) This was missed in https://github.com/meta-llama/llama-stack/pull/802 somehow. --- llama_stack/cli/stack/_build.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/llama_stack/cli/stack/_build.py b/llama_stack/cli/stack/_build.py index 99c3a7ffc..8fde31779 100644 --- a/llama_stack/cli/stack/_build.py +++ b/llama_stack/cli/stack/_build.py @@ -74,7 +74,7 @@ def run_stack_build_command( build_config.image_type = args.image_type else: cprint( - f"Please specify a image-type (docker | conda | venv) for {args.template}", + f"Please specify a image-type (container | conda | venv) for {args.template}", color="red", ) return @@ -95,10 +95,10 @@ def run_stack_build_command( ) image_type = prompt( - "> Enter the image type you want your Llama Stack to be built as (docker or conda or venv): ", + "> Enter the image type you want your Llama Stack to be built as (container or conda or venv): ", validator=Validator.from_callable( - lambda x: x in ["docker", "conda", "venv"], - error_message="Invalid image type, please enter conda or docker or venv", + lambda x: x in ["container", "conda", "venv"], + error_message="Invalid image type, please enter conda or container or venv", ), default="conda", ) @@ -259,7 +259,7 @@ def _run_stack_build_command_from_build_config( else: if not image_name: raise ValueError( - "Please specify an image name when building a docker image without a template" + "Please specify an image name when building a container image without a template" ) elif build_config.image_type == ImageType.conda.value: if not image_name: