From 12042b26d44923d08439a5815b2ea64b3df95c75 Mon Sep 17 00:00:00 2001
From: Ashwin Bharambe <ashwin.bharambe@gmail.com>
Date: Thu, 30 Oct 2025 21:41:45 -0700
Subject: [PATCH] fix(ci): replace unused LLAMA_STACK_CLIENT_DIR with direct
 install

The LLAMA_STACK_CLIENT_DIR variable was used by the old `llama stack build`
command but is no longer used after switching to `llama stack list-deps`.

Replace with direct `uv pip install --force-reinstall` when targeting
release branches to ensure the matching client version is installed.
---
 .github/actions/setup-test-environment/action.yml | 13 +++----------
 1 file changed, 3 insertions(+), 10 deletions(-)

diff --git a/.github/actions/setup-test-environment/action.yml b/.github/actions/setup-test-environment/action.yml
index 27d0943fe..992b25803 100644
--- a/.github/actions/setup-test-environment/action.yml
+++ b/.github/actions/setup-test-environment/action.yml
@@ -54,23 +54,16 @@ runs:
             # Check if the branch exists in the client repo
             if git ls-remote --exit-code --heads https://github.com/llamastack/llama-stack-client-python.git "$TARGET_BRANCH" > /dev/null 2>&1; then
               echo "Installing llama-stack-client-python from matching branch: $TARGET_BRANCH"
-              export LLAMA_STACK_CLIENT_DIR=git+https://github.com/llamastack/llama-stack-client-python.git@$TARGET_BRANCH
+              uv pip install --force-reinstall git+https://github.com/llamastack/llama-stack-client-python.git@$TARGET_BRANCH
             else
               echo "::error::Branch $TARGET_BRANCH not found in llama-stack-client-python repository"
               echo "::error::Please create the matching release branch in llama-stack-client-python before testing"
               exit 1
             fi
-          else
-            echo "Installing latest llama-stack-client-python from main branch"
-            export LLAMA_STACK_CLIENT_DIR=git+https://github.com/llamastack/llama-stack-client-python.git@main
           fi
-        elif [ "${{ inputs.client-version }}" = "published" ]; then
-          echo "Installing published llama-stack-client-python from PyPI"
-          unset LLAMA_STACK_CLIENT_DIR
-        else
-          echo "Invalid client-version: ${{ inputs.client-version }}"
-          exit 1
+          # For main branch, client is already installed by setup-runner
         fi
+        # For published version, client is already installed by setup-runner
 
         echo "Building Llama Stack"