# FROM unsloth/unsloth:stable
FROM nielsrolf/ow-default:v0.7

USER root

WORKDIR /openweights

# Install SSH
# RUN apt-get update && \
#     apt-get install -y openssh-server rsync git-lfs && \
#     mkdir /var/run/sshd
# RUN apt-get update && apt-get install -y --no-install-recommends unison

# # Create a directory for SSH keys
# RUN mkdir -p /root/.ssh && chmod 700 /root/.ssh

# # Update SSH configuration
# RUN echo "PermitRootLogin yes" >> /etc/ssh/sshd_config && \
#     echo "PasswordAuthentication no" >> /etc/ssh/sshd_config && \
#     echo "PubkeyAuthentication yes" >> /etc/ssh/sshd_config

RUN python3 -m pip install --upgrade pip
RUN python3 -m pip install inspect_ai git+https://github.com/UKGovernmentBEIS/inspect_evals
RUN python3 -m pip install vllm huggingface_hub[hf_transfer] hf_transfer supabase python-dotenv fire httpx>=0.24.0 runpod
ENV HF_HUB_ENABLE_HF_TRANSFER=1

RUN git lfs install

COPY README.md .
COPY pyproject.toml .
COPY openweights openweights
COPY entrypoint.sh .
RUN python3 -m pip install -e .

# Add conda to PATH for interactive SSH sessions
RUN echo 'export PATH=/opt/conda/bin:$PATH' >> /root/.bashrc && \
    echo 'export PATH=/opt/conda/bin:$PATH' >> /root/.profile

EXPOSE 22
EXPOSE 8000
EXPOSE 10101

# USER unsloth

ENTRYPOINT ["/openweights/entrypoint.sh"]
