Fix api dying (#353)

Adds a startup script to use in dockerfile instead of a list style CMD --------- Co-authored-by: Faisal Alquaddoomi <[email protected]>
monarch-initiative · Sep 29, 2023 · 3481eaf · 3481eaf
1 parent bc16562
commit 3481eaf
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/backend/Dockerfile b/backend/Dockerfile
@@ -29,7 +29,7 @@ RUN apt-get update -y && \
 
 # copy project requirement files here to ensure they will be cached.
 WORKDIR $PYSETUP_PATH
-COPY . /opt/monarch-app
+COPY . /opt/monarch-app/backend
 
 # install runtime deps - uses $POETRY_VIRTUALENVS_IN_PROJECT internally
 RUN curl -sSL https://install.python-poetry.org | python3 && \
@@ -41,12 +41,12 @@ RUN curl -sSL https://install.python-poetry.org | python3 && \
 
 FROM builder-base as production
 ENV SKLEARN_ALLOW_DEPRECATED_SKLEARN_PACKAGE_INSTALL=True
-ENV UVICORN_WORKERS=8
+# ENV UVICORN_WORKERS=8
 WORKDIR /opt/monarch-app/backend
 
 RUN poetry install
 
-CMD ["/opt/poetry/bin/poetry", "run", "uvicorn", "src.monarch_py.api.main:app", "--host", "0.0.0.0", "--port", "8000", "--workers", "${UVICORN_WORKERS}"]
+CMD /opt/monarch-app/backend/start_api.sh
 
 ### Image URLs for reference ###
 #

diff --git a/backend/start_api.sh b/backend/start_api.sh
@@ -0,0 +1,7 @@
+#!/usr/bin/bash 
+
+# check for uvicorn workers env var, default to 1
+UVICORN_WORKERS=${UVICORN_WORKERS:-1}
+
+# Start the API
+/opt/poetry/bin/poetry run uvicorn src.monarch_py.api.main:app --host 0.0.0.0 --port 8000 --workers ${UVICORN_WORKERS}