FROM vllm/vllm-openai-rocm:nightly

WORKDIR /workspace

COPY requirements.txt /workspace/requirements.txt
RUN pip install --no-cache-dir --retries 20 --timeout 600 -i https://pypi.tuna.tsinghua.edu.cn/simple -r /workspace/requirements.txt \
 || pip install --no-cache-dir --retries 20 --timeout 600 -i https://mirrors.aliyun.com/pypi/simple -r /workspace/requirements.txt \
 || pip install --no-cache-dir --retries 20 --timeout 600 -r /workspace/requirements.txt

COPY app /workspace/app

ENV PYTHONUNBUFFERED=1

EXPOSE 8001

CMD ["python", "-m", "app.start_openai"]
