diff --git a/llama_stack/distribution/build_conda_env.sh b/llama_stack/distribution/build_conda_env.sh index 3fea09ad5..ff9c26e5e 100755 --- a/llama_stack/distribution/build_conda_env.sh +++ b/llama_stack/distribution/build_conda_env.sh @@ -9,6 +9,9 @@ LLAMA_MODELS_DIR=${LLAMA_MODELS_DIR:-} LLAMA_STACK_DIR=${LLAMA_STACK_DIR:-} TEST_PYPI_VERSION=${TEST_PYPI_VERSION:-} +# This timeout (in seconds) is necessary when installing PyTorch via uv since it's likely to time out +# Reference: https://github.com/astral-sh/uv/pull/1694 +UV_HTTP_TIMEOUT=${UV_HTTP_TIMEOUT:-500} if [ -n "$LLAMA_STACK_DIR" ]; then echo "Using llama-stack-dir=$LLAMA_STACK_DIR" diff --git a/llama_stack/distribution/build_container.sh b/llama_stack/distribution/build_container.sh index 8cad13eec..4101cec44 100755 --- a/llama_stack/distribution/build_container.sh +++ b/llama_stack/distribution/build_container.sh @@ -11,6 +11,9 @@ LLAMA_STACK_DIR=${LLAMA_STACK_DIR:-} TEST_PYPI_VERSION=${TEST_PYPI_VERSION:-} PYPI_VERSION=${PYPI_VERSION:-} BUILD_PLATFORM=${BUILD_PLATFORM:-} +# This timeout (in seconds) is necessary when installing PyTorch via uv since it's likely to time out +# Reference: https://github.com/astral-sh/uv/pull/1694 +UV_HTTP_TIMEOUT=${UV_HTTP_TIMEOUT:-500} # mounting is not supported by docker buildx, so we use COPY instead USE_COPY_NOT_MOUNT=${USE_COPY_NOT_MOUNT:-} diff --git a/llama_stack/distribution/build_venv.sh b/llama_stack/distribution/build_venv.sh index 970c4144f..3166c07f6 100755 --- a/llama_stack/distribution/build_venv.sh +++ b/llama_stack/distribution/build_venv.sh @@ -12,6 +12,9 @@ LLAMA_MODELS_DIR=${LLAMA_MODELS_DIR:-} LLAMA_STACK_DIR=${LLAMA_STACK_DIR:-} TEST_PYPI_VERSION=${TEST_PYPI_VERSION:-} +# This timeout (in seconds) is necessary when installing PyTorch via uv since it's likely to time out +# Reference: https://github.com/astral-sh/uv/pull/1694 +UV_HTTP_TIMEOUT=${UV_HTTP_TIMEOUT:-500} if [ -n "$LLAMA_STACK_DIR" ]; then echo "Using llama-stack-dir=$LLAMA_STACK_DIR"