From 6ae6ff51d512f64d8b825ca5e0788d12b05d7ea2 Mon Sep 17 00:00:00 2001
From: Alan Jean <afinegameofnil@gmail.com>
Date: Tue, 31 Jan 2023 19:30:12 +0400
Subject: [PATCH] =?UTF-8?q?feat(inference):=20add=20a=20prod=20stage=20to?=
 =?UTF-8?q?=20the=20server's=20dockerfile=20=F0=9F=90=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 docker/inference/Dockerfile.server | 44 ++++++++++++++++++++++--------
 1 file changed, 33 insertions(+), 11 deletions(-)

diff --git a/docker/inference/Dockerfile.server b/docker/inference/Dockerfile.server
index 34d8e0c9..0838a21e 100644
--- a/docker/inference/Dockerfile.server
+++ b/docker/inference/Dockerfile.server
@@ -22,7 +22,7 @@ RUN --mount=type=cache,target=/var/cache/pip \
 
 
 
-FROM python:3.10-alpine3.17 as dev
+FROM python:3.10-alpine3.17 as base-env
 ARG APP_USER
 ARG APP_RELATIVE_PATH
 ARG MODULE
@@ -44,25 +44,47 @@ RUN adduser               \
       --no-create-home    \
       "${APP_USER}"
 
+USER ${APP_USER}
 
 WORKDIR ${APP_ROOT}
 
 
-COPY --chown="${APP_USER}:${APP_USER}" ./oasst-shared ${SHARED_LIBS_BASE}/oasst-shared
-RUN --mount=type=cache,target=/var/cache/pip,from=build \
-    pip install                  \
-      --cache-dir=/var/cache/pip \
-      -e "${SHARED_LIBS_BASE}/oasst-shared"
-
-
-USER ${APP_USER}
-
-
 COPY --chown="${APP_USER}:${APP_USER}" --from=build /build/lib                     ${APP_LIBS}
 COPY --chown="${APP_USER}:${APP_USER}"              ./${APP_RELATIVE_PATH}/main.py .
 
 
+
+FROM base-env as dev
+ARG APP_USER
+
+
+COPY --chown="${APP_USER}:${APP_USER}" ./oasst-shared ${SHARED_LIBS_BASE}/oasst-shared
+
+USER root
+RUN --mount=type=cache,target=/var/cache/pip,from=build \
+    pip install                  \
+      --cache-dir=/var/cache/pip \
+      -e "${SHARED_LIBS_BASE}/oasst-shared"
+USER ${APP_USER}
+
+
 VOLUME [ "${APP_BASE}/lib/oasst-shared" ]
 
 
 CMD uvicorn main:app --reload --host 0.0.0.0 --port "${PORT}"
+
+
+
+FROM base-env as prod
+ARG APP_USER
+
+
+COPY --chown="${APP_USER}:${APP_USER}" ./oasst-shared /tmp/lib/oasst-shared
+RUN --mount=type=cache,target=/var/cache/pip,from=dev \
+    pip install                  \
+      --cache-dir=/var/cache/pip \
+      --target="${APP_LIBS}"     \
+      /tmp/lib/oasst-shared
+
+
+CMD uvicorn main:app --host 0.0.0.0 --port "${PORT}"