FROM python:3.11-slim

WORKDIR /app

# 配置pip使用阿里云镜像
ENV PIP_INDEX_URL=https://mirrors.aliyun.com/pypi/simple/
ENV PIP_TRUSTED_HOST=mirrors.aliyun.com
ENV PIP_NO_CACHE_DIR=1

# 复制requirements文件
COPY requirements-cpu-full.txt requirements.txt

# 不使用apt-get，直接用pip

# 升级pip并安装所有依赖
RUN pip install --upgrade pip setuptools wheel && \
    pip install torch --index-url https://download.pytorch.org/whl/cpu && \
    pip install transformers sentence-transformers && \
    pip install -r requirements.txt

# 复制项目文件
COPY . .

# 确保test_with_concept.pkl文件在正确位置
# 文件应该在项目根目录，如果不在，需要单独复制
RUN if [ -f /app/test_with_concept.pkl ]; then \
        echo "test_with_concept.pkl found in /app"; \
    else \
        echo "WARNING: test_with_concept.pkl not found"; \
    fi

# 创建必要目录
RUN mkdir -p /app/api_outputs /app/logs

# 设置环境变量
ENV PYTHONPATH=/app
ENV PYTHONDONTWRITEBYTECODE=1
ENV PYTHONUNBUFFERED=1
ENV RAG_CONFIG_PATH=rag_config_production.yaml

# API配置
ENV ONEAPI_KEY=sk-2c5045478a244022865e65c5c9b6adf7
ENV ONEAPI_MODEL=qwen2-7b-instruct
ENV ONEAPI_MODEL_GEN=qwen2-7b-instruct
ENV ONEAPI_MODEL_MAX=qwen2-7b-instruct
ENV ONEAPI_MODEL_EMBED=text-embedding-v3

# LangSmith配置
ENV LANGCHAIN_TRACING_V2=false
ENV LANGCHAIN_ENDPOINT=https://api.smith.langchain.com
ENV LANGCHAIN_API_KEY=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
ENV LANGCHAIN_PROJECT=hipporag-retriever

# Elasticsearch配置
ENV ELASTICSEARCH_HOST=http://101.200.154.78:9200
ENV ELASTICSEARCH_USERNAME=elastic
ENV ELASTICSEARCH_PASSWORD=Abcd123456

EXPOSE 8000

CMD ["uvicorn", "rag_api_server_production:app", "--host", "0.0.0.0", "--port", "8000"]