diff --git a/scripts/setup_workload_identity.sh b/scripts/setup_workload_identity.sh
index 2a854f74..94c3e1e9 100755
--- a/scripts/setup_workload_identity.sh
+++ b/scripts/setup_workload_identity.sh
@@ -456,6 +456,7 @@ if command -v gh &> /dev/null; then
     gh variable set GOOGLE_CLOUD_PROJECT --body "${GOOGLE_CLOUD_PROJECT}" --repo "${GITHUB_REPO}"
     gh variable set GOOGLE_CLOUD_LOCATION --body "${GOOGLE_CLOUD_LOCATION}" --repo "${GITHUB_REPO}"
     gh variable set SERVICE_ACCOUNT_EMAIL --body "${SERVICE_ACCOUNT_EMAIL}" --repo "${GITHUB_REPO}"
+    gh variable set GOOGLE_GENAI_USE_VERTEXAI --body "true" --repo "${GITHUB_REPO}"
 
     print_success "GitHub variables have been set automatically!"
 else
@@ -474,6 +475,10 @@ else
     echo "☁️ Variable Name: SERVICE_ACCOUNT_EMAIL"
     echo "   Variable Value: ${SERVICE_ACCOUNT_EMAIL}"
     echo ""
+    echo "🤖 Variable Name: GOOGLE_GENAI_USE_VERTEXAI"
+    echo "   Variable Value: true"
+    echo "   (Set to 'true' to use Vertex AI, or use GOOGLE_GENAI_USE_GCA for Code Assist)"
+    echo ""
 fi
 
 print_success "Setup completed successfully! 🚀"