mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-11 19:56:03 +00:00
1c557b87cb
In CI, git is in detached HEAD state, so git rev-parse returns 'HEAD'. Use GITHUB_REF and GITHUB_BASE_REF to properly detect release branches in GitHub Actions workflows. This ensures pre-commit hooks in CI can correctly identify when running against release branches and set the appropriate UV index URLs.
53 lines
1.8 KiB
Bash
Executable file
53 lines
1.8 KiB
Bash
Executable file
#!/bin/bash
|
|
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the terms described in the LICENSE file in
|
|
# the root directory of this source tree.
|
|
|
|
set -euo pipefail
|
|
|
|
# Detect current branch and target branch
|
|
# In GitHub Actions, use GITHUB_REF/GITHUB_BASE_REF
|
|
if [[ -n "${GITHUB_REF:-}" ]]; then
|
|
BRANCH="${GITHUB_REF#refs/heads/}"
|
|
else
|
|
BRANCH=$(git rev-parse --abbrev-ref HEAD 2>/dev/null || echo "")
|
|
fi
|
|
|
|
# For PRs, check the target branch
|
|
if [[ -n "${GITHUB_BASE_REF:-}" ]]; then
|
|
TARGET_BRANCH="${GITHUB_BASE_REF}"
|
|
else
|
|
TARGET_BRANCH=$(git rev-parse --abbrev-ref HEAD@{upstream} 2>/dev/null | sed 's|origin/||' || echo "")
|
|
fi
|
|
|
|
echo "[uv-run-with-index] Current branch: '$BRANCH'" >&2
|
|
echo "[uv-run-with-index] Target branch: '$TARGET_BRANCH'" >&2
|
|
echo "[uv-run-with-index] PWD: $PWD" >&2
|
|
echo "[uv-run-with-index] Command: uv $*" >&2
|
|
|
|
# Check if on a release branch or targeting one, or LLAMA_STACK_RELEASE_MODE is set
|
|
IS_RELEASE=false
|
|
if [[ "$BRANCH" =~ ^release-[0-9]+\.[0-9]+\.x$ ]]; then
|
|
echo "[uv-run-with-index] ✓ On release branch: $BRANCH" >&2
|
|
IS_RELEASE=true
|
|
elif [[ "$TARGET_BRANCH" =~ ^release-[0-9]+\.[0-9]+\.x$ ]]; then
|
|
echo "[uv-run-with-index] ✓ Targeting release branch: $TARGET_BRANCH" >&2
|
|
IS_RELEASE=true
|
|
elif [[ "${LLAMA_STACK_RELEASE_MODE:-}" == "true" ]]; then
|
|
echo "[uv-run-with-index] ✓ LLAMA_STACK_RELEASE_MODE=true" >&2
|
|
IS_RELEASE=true
|
|
else
|
|
echo "[uv-run-with-index] ✗ Not a release branch" >&2
|
|
fi
|
|
|
|
# On release branches, use test.pypi as extra index for RC versions
|
|
if [[ "$IS_RELEASE" == "true" ]]; then
|
|
echo "[uv-run-with-index] Setting UV_EXTRA_INDEX_URL=https://test.pypi.org/simple/" >&2
|
|
export UV_EXTRA_INDEX_URL="https://test.pypi.org/simple/"
|
|
export UV_INDEX_STRATEGY="unsafe-best-match"
|
|
fi
|
|
|
|
# Run uv with all arguments passed through
|
|
exec uv "$@"
|