obol-stack/Dockerfile.worker at main · ObolNetwork/obol-stack · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
FROM nvidia/cuda:12.4.1-runtime-ubuntu22.04

ENV DEBIAN_FRONTEND=noninteractive

RUN apt-get update && apt-get install -y --no-install-recommends \
    ca-certificates \
    curl \
    git \
    python3 \
    python3-pip \
    python3-venv \
    && rm -rf /var/lib/apt/lists/*

# Install uv for running the autoresearch repo environment.
RUN curl -LsSf https://astral.sh/uv/install.sh | sh \
    && mv /root/.local/bin/uv /usr/local/bin/uv

WORKDIR /app

COPY internal/embed/skills/autoresearch-worker/scripts/worker_api.py /app/worker_api.py

ENV DATA_DIR=/data \
    AUTORESEARCH_REPO=/data/autoresearch \
    EXPERIMENT_TIMEOUT_SECONDS=300 \
    TRAIN_COMMAND="uv run train.py"

RUN useradd -m -s /bin/bash worker && mkdir -p /data && chown worker:worker /data

USER worker
VOLUME ["/data"]
EXPOSE 8080

ENTRYPOINT ["python3", "/app/worker_api.py", "serve", "--repo", "/data/autoresearch", "--data-dir", "/data", "--host", "0.0.0.0", "--port", "8080"]