From 1c46d80c1300e7206dd192d8cccd129a09f8d993 Mon Sep 17 00:00:00 2001 From: Yuan Tang Date: Mon, 3 Feb 2025 03:34:25 -0500 Subject: [PATCH] Configurable and unit Signed-off-by: Yuan Tang --- llama_stack/distribution/build_conda_env.sh | 4 ++-- llama_stack/distribution/build_container.sh | 4 ++-- llama_stack/distribution/build_venv.sh | 4 ++-- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/llama_stack/distribution/build_conda_env.sh b/llama_stack/distribution/build_conda_env.sh index bb4f1b923..ff9c26e5e 100755 --- a/llama_stack/distribution/build_conda_env.sh +++ b/llama_stack/distribution/build_conda_env.sh @@ -9,9 +9,9 @@ LLAMA_MODELS_DIR=${LLAMA_MODELS_DIR:-} LLAMA_STACK_DIR=${LLAMA_STACK_DIR:-} TEST_PYPI_VERSION=${TEST_PYPI_VERSION:-} -# This is necessary when installing PyTorch via uv since it's likely to time out +# 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=500 +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 5c9f31ac4..4101cec44 100755 --- a/llama_stack/distribution/build_container.sh +++ b/llama_stack/distribution/build_container.sh @@ -11,9 +11,9 @@ LLAMA_STACK_DIR=${LLAMA_STACK_DIR:-} TEST_PYPI_VERSION=${TEST_PYPI_VERSION:-} PYPI_VERSION=${PYPI_VERSION:-} BUILD_PLATFORM=${BUILD_PLATFORM:-} -# This is necessary when installing PyTorch via uv since it's likely to time out +# 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=500 +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 eedbbd9e8..3166c07f6 100755 --- a/llama_stack/distribution/build_venv.sh +++ b/llama_stack/distribution/build_venv.sh @@ -12,9 +12,9 @@ LLAMA_MODELS_DIR=${LLAMA_MODELS_DIR:-} LLAMA_STACK_DIR=${LLAMA_STACK_DIR:-} TEST_PYPI_VERSION=${TEST_PYPI_VERSION:-} -# This is necessary when installing PyTorch via uv since it's likely to time out +# 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=500 +UV_HTTP_TIMEOUT=${UV_HTTP_TIMEOUT:-500} if [ -n "$LLAMA_STACK_DIR" ]; then echo "Using llama-stack-dir=$LLAMA_STACK_DIR"