Use uv in Dockerfile

Author: abrookins

.github/workflows/python-tests.yml

@@ -21,11 +21,12 @@ jobs:
     - name: Install pre-commit
       run: |
         python -m pip install --upgrade pip
-        pip install pre-commit
+        pip install uv
+        uv sync --only-dev
 
     - name: Run pre-commit
       run: |
-        pre-commit run --all-files
+        uv run pre-commit run --all-files
 
   test:
     needs: lint
@@ -56,10 +57,10 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install -e .[dev]
+        uv sync --all-extras
 
     - name: Run tests
       run: |
-        pytest --run-api-tests
+        uv run pytest --run-api-tests
       env:
         OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}

Dockerfile

@@ -1,7 +1,10 @@
-FROM python:3.12-slim
+FROM ghcr.io/astral-sh/uv:python3.12-bookworm-slim
 
 WORKDIR /app
 
+ENV UV_COMPILE_BYTECODE=1
+ENV UV_LINK_MODE=copy
+
 # Install system dependencies including build tools
 RUN apt-get update && apt-get install -y \
     curl \
@@ -10,12 +13,19 @@ RUN apt-get update && apt-get install -y \
     g++ \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy project files
-COPY pyproject.toml README.md ./
-COPY agent_memory_server ./agent_memory_server
+RUN --mount=type=cache,target=/root/.cache/uv \
+    --mount=type=bind,source=uv.lock,target=uv.lock \
+    --mount=type=bind,source=pyproject.toml,target=pyproject.toml \
+    uv sync --frozen --no-install-project --no-dev
+
+ADD . /app
+RUN --mount=type=cache,target=/root/.cache/uv \
+    uv sync --frozen --no-dev
+
+ENV PATH="/app/.venv/bin:$PATH"
+
+ENTRYPOINT []
 
-# Install Python dependencies
-RUN pip install --no-cache-dir -e .
 
 # Run the API server
 CMD ["python", "-m", "agent_memory_server.main"]