Added clean up to build step of the Dockerfile

Author: m-messer

Dockerfile

@@ -9,29 +9,35 @@ ENV POETRY_NO_INTERACTION=1 \
 
 COPY pyproject.toml poetry.lock ./
 
+# Install dependencies and clean up in same layer
 RUN --mount=type=cache,target=$POETRY_CACHE_DIR \
-    poetry install --without dev --no-root
+    poetry install --without dev --no-root && \
+    # Remove unnecessary files from venv to reduce size
+    find /app/.venv -name "*.pyc" -delete && \
+    find /app/.venv -name "__pycache__" -type d -exec rm -rf {} + && \
+    find /app/.venv -name "*.pyo" -delete && \
+    # Remove test files and documentation
+    find /app/.venv -path "*/tests/*" -delete && \
+    find /app/.venv -path "*/test/*" -delete && \
+    find /app/.venv -name "*.md" -delete && \
+    find /app/.venv -name "*.txt" -delete
 
 FROM ghcr.io/lambda-feedback/evaluation-function-base/python:3.12
 
 ENV VIRTUAL_ENV=/app/.venv \
     PATH="/app/.venv/bin:$PATH"
 
+# Copy the cleaned virtual environment
 COPY --from=builder ${VIRTUAL_ENV} ${VIRTUAL_ENV}
 
-# Precompile python files for faster startup
-RUN python -m compileall -q .
-
-# Copy the evaluation function to the app directory
+# Copy evaluation function first (smaller, changes more often)
 COPY evaluation_function ./evaluation_function
 
-# Command to start the evaluation function with
-ENV FUNCTION_COMMAND="python"
-
-# Args to start the evaluation function with
-ENV FUNCTION_ARGS="-m,evaluation_function.main"
-
-# The transport to use for the RPC server
-ENV FUNCTION_RPC_TRANSPORT="ipc"
+# Precompile python files for faster startup (do this last)
+RUN python -m compileall -q .
 
-ENV LOG_LEVEL="debug"
+# Environment variables
+ENV FUNCTION_COMMAND="python" \
+    FUNCTION_ARGS="-m,evaluation_function.main" \
+    FUNCTION_RPC_TRANSPORT="ipc" \
+    LOG_LEVEL="debug"