FROM python:3.10-slim

# 安装依赖
RUN apt-get update && apt-get install -y \
    git build-essential curl unzip && \
    rm -rf /var/lib/apt/lists/*

# 设置工作目录
WORKDIR /app

# 复制项目代码
COPY . /app

# 安装 Python 依赖
RUN pip install --upgrade pip && \
    pip install -r requirements.txt && \
    pip install openai

# 下载 OpenCompass 官方数据集（保留原始结构）
# RUN curl -L -o OpenCompassData.zip https://github.com/open-compass/opencompass/releases/download/0.2.2.rc1/OpenCompassData-core-20240207.zip && \
#    unzip OpenCompassData.zip && \
#    rm OpenCompassData.zip

# 设置 PYTHONPATH，确保 opencompass 模块可用
ENV PYTHONPATH=/app

# 默认运行评测任务（确保你已准备好 eval_myopenai_cmmlu.py）
CMD ["python", "run.py", "eval_myopenai_cmmlu.py", "--mode", "all", "-w", "results/myopenai_cmmlu"]
