feat: --image-type argument overrides value in --config build.yaml (#2179)

closes #2162

# test plan

run `llama stack build --image-name ollama --image-type
<venv/conda/container> --config llama_stack/templates/ollama/build.yaml`
and verify venv | conda | container are built.
This commit is contained in:
Matthew Farrellee 2025-05-16 17:45:41 -04:00 committed by GitHub
parent f02f7b28c1
commit f40693e720
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 3 additions and 1 deletions

View file

@ -206,6 +206,8 @@ def run_stack_build_command(args: argparse.Namespace) -> None:
contents = yaml.safe_load(f)
contents = replace_env_vars(contents)
build_config = BuildConfig(**contents)
if args.image_type:
build_config.image_type = args.image_type
except Exception as e:
cprint(
f"Could not parse config file {args.config}: {e}",

View file

@ -49,7 +49,7 @@ class StackBuild(Subcommand):
type=str,
help="Image Type to use for the build. If not specified, will use the image type from the template config.",
choices=[e.value for e in ImageType],
default=ImageType.CONDA.value,
default=None, # no default so we can detect if a user specified --image-type and override image_type in the config
)
self.parser.add_argument(