From 16ae95cb09aae0a8f2f95328782cdeb896ca221d Mon Sep 17 00:00:00 2001
From: Ashwin Bharambe <ashwin.bharambe@gmail.com>
Date: Fri, 31 Oct 2025 12:19:38 -0700
Subject: [PATCH] fix: add UV build args to integration-tests.sh docker build

Only adding the UV_EXTRA_INDEX_URL and UV_INDEX_STRATEGY build args
for release branch support, without other unrelated changes.
---
 scripts/integration-tests.sh | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/scripts/integration-tests.sh b/scripts/integration-tests.sh
index 506ac12e0..985952167 100755
--- a/scripts/integration-tests.sh
+++ b/scripts/integration-tests.sh
@@ -279,6 +279,16 @@ if [[ "$STACK_CONFIG" == *"docker:"* && "$COLLECT_ONLY" == false ]]; then
         --build-arg "LLAMA_STACK_DIR=/workspace"
     )
 
+    # Pass UV index configuration for release branches
+    if [[ -n "${UV_EXTRA_INDEX_URL:-}" ]]; then
+        echo "Adding UV_EXTRA_INDEX_URL to docker build: $UV_EXTRA_INDEX_URL"
+        build_cmd+=(--build-arg "UV_EXTRA_INDEX_URL=$UV_EXTRA_INDEX_URL")
+    fi
+    if [[ -n "${UV_INDEX_STRATEGY:-}" ]]; then
+        echo "Adding UV_INDEX_STRATEGY to docker build: $UV_INDEX_STRATEGY"
+        build_cmd+=(--build-arg "UV_INDEX_STRATEGY=$UV_INDEX_STRATEGY")
+    fi
+
     if ! "${build_cmd[@]}"; then
         echo "❌ Failed to build Docker image"
         exit 1