phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-04 04:04:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

build + run image seems to work

Browse source
This commit is contained in:
Ashwin Bharambe 2024-08-27 06:12:19 -07:00
parent 6f83187809
commit a4af9675ac
2 changed files with 24 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

22
llama_toolchain/distribution/build_image.sh
View file

@ -1,20 +1,22 @@
#!/bin/bash
if [ "$#" -ne 4 ]; then
echo "Usage: $0 <image_name> <base_image> <pip_dependencies> <entrypoint_command>"
echo "Example: $0 my-fastapi-app python:3.9-slim 'fastapi uvicorn' 'python3 -m llama_toolchain.distribution.server --port 8000'"
if [ "$#" -ne 3 ]; then
echo "Usage: $0 <image_name> <base_image> <pip_dependencies>
echo "Example: $0 my-fastapi-app python:3.9-slim 'fastapi uvicorn'
exit 1
fi
IMAGE_NAME=$1
BASE_IMAGE=$2
PIP_DEPENDENCIES=$3
ENTRYPOINT_COMMAND=$4
set -euo pipefail
PORT=8001
SCRIPT_DIR=$(dirname "$(readlink -f "$0")")
SOURCE_DIR=$(dirname $(dirname "$SCRIPT_DIR"))
SOURCE_DIR=$(dirname $(dirname $(dirname "$SCRIPT_DIR")))
echo $SOURCE_DIR
TEMP_DIR=$(mktemp -d)
echo "Created temporary directory: $TEMP_DIR"
@ -22,12 +24,14 @@ echo "Created temporary directory: $TEMP_DIR"
cat <<EOF >"$TEMP_DIR/Dockerfile"
FROM $BASE_IMAGE
WORKDIR /app
COPY llama_toolchain /app
COPY llama-stack/llama_toolchain /app/llama_toolchain
COPY llama-models/models /app/llama_models
RUN pip install --no-cache-dir $PIP_DEPENDENCIES
RUN pip install $PIP_DEPENDENCIES
EXPOSE $PORT
ENTRYPOINT ["python3", "-m", "llama_toolchain.distribution.server", "--port", "$PORT"]
EXPOSE 8000
CMD $ENTRYPOINT_COMMAND
EOF
echo "Dockerfile created successfully in $TEMP_DIR/Dockerfile"