From 8a18d5ecbf2d5db955bf838685043be6fe90f240 Mon Sep 17 00:00:00 2001
From: Matthew Farrellee <matt@cs.wisc.edu>
Date: Thu, 15 May 2025 10:26:42 -0400
Subject: [PATCH] feat: --image-type argument overrides value in --config
 build.yaml

closes #2162
---
 llama_stack/cli/stack/_build.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/llama_stack/cli/stack/_build.py b/llama_stack/cli/stack/_build.py
index ae4a39ce2..88557e10d 100644
--- a/llama_stack/cli/stack/_build.py
+++ b/llama_stack/cli/stack/_build.py
@@ -203,6 +203,8 @@ def run_stack_build_command(args: argparse.Namespace) -> None:
         with open(args.config) as f:
             try:
                 build_config = BuildConfig(**yaml.safe_load(f))
+                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}",