#基础镜像请自行联系昆仑芯R200厂商技术支持人员获取
ARG BASE_IMAGE=kunlunxin-xtrt-llm:0.5.3.2
FROM $BASE_IMAGE

ARG PYPI_MIRROR=https://pypi.tuna.tsinghua.edu.cn/simple
RUN pip config set global.index-url $PYPI_MIRROR

ADD requirements.txt .
RUN pip install -r requirements.txt

ADD llm_utils.py main.py ./

ENTRYPOINT ["python3", "main.py"]