From a437b9dfa4e6febe0747b956e965a551d922d6bc Mon Sep 17 00:00:00 2001 From: George Burton Date: Wed, 27 Mar 2024 08:43:27 +0000 Subject: [PATCH 1/2] added EMBEDDING_MODEL to streamlit-app build --- docker-compose.yml | 2 ++ streamlit_app/Dockerfile | 3 +-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/docker-compose.yml b/docker-compose.yml index 178ffe455..ee9d7ad3e 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -43,6 +43,8 @@ services: build: context: . dockerfile: ./streamlit_app/Dockerfile + args: + - EMBEDDING_MODEL=${EMBEDDING_MODEL:-} env_file: - .env ports: diff --git a/streamlit_app/Dockerfile b/streamlit_app/Dockerfile index 62d17adce..f03e3b2c0 100644 --- a/streamlit_app/Dockerfile +++ b/streamlit_app/Dockerfile @@ -27,8 +27,7 @@ COPY --from=builder ${VIRTUAL_ENV} ${VIRTUAL_ENV} ADD redbox/ /app/redbox ADD download_embedder.py /app/ -RUN python download_embedder.py - +RUN python download_embedder.py --model_name ${EMBEDDING_MODEL} ADD streamlit_app/ /app EXPOSE 8501 From 2ef802028a28cfbfe52ff5d6cbc9fa89de7aad04 Mon Sep 17 00:00:00 2001 From: George Burton Date: Wed, 27 Mar 2024 09:00:19 +0000 Subject: [PATCH 2/2] added missing ARG --- streamlit_app/Dockerfile | 1 + 1 file changed, 1 insertion(+) diff --git a/streamlit_app/Dockerfile b/streamlit_app/Dockerfile index f03e3b2c0..9da3f72c0 100644 --- a/streamlit_app/Dockerfile +++ b/streamlit_app/Dockerfile @@ -14,6 +14,7 @@ ADD pyproject.toml poetry.lock ./ RUN --mount=type=cache,target=$POETRY_CACHE_DIR poetry install --no-root --no-ansi --with streamlit-app,ai,ingester --without dev,worker,api --no-root FROM python:3.11-slim-buster as runtime +ARG EMBEDDING_MODEL RUN apt-get update RUN apt-get install -y libgl-dev libmagic-dev inetutils-ping