phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

Make sure TEST_PYPI_VERSION is used in docker builds

Browse source
This commit is contained in:
Ashwin Bharambe 2024-11-11 18:40:13 -08:00
parent 285cd26fb2
commit 343458479d
1 changed files with 10 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

11
llama_stack/distribution/build_container.sh
View file

@ -78,7 +78,16 @@ if [ -n "$LLAMA_STACK_DIR" ]; then
# rebuild. This is just for development convenience.
add_to_docker "RUN pip install --no-cache -e $stack_mount"
else
add_to_docker "RUN pip install --no-cache llama-stack"
if [ -n "$TEST_PYPI_VERSION" ]; then
# these packages are damaged in test-pypi, so install them first
add_to_docker "RUN pip install fastapi libcst"
add_to_docker <<EOF
RUN pip install --no-cache --extra-index-url https://test.pypi.org/simple/ \
llama-models==$TEST_PYPI_VERSION llama-stack==$TEST_PYPI_VERSION
EOF
else
add_to_docker "RUN pip install --no-cache llama-stack"
fi
fi
if [ -n "$LLAMA_MODELS_DIR" ]; then