first commit

This commit is contained in:
闫旭隆
2025-09-24 09:29:12 +08:00
parent 6339cdebb9
commit 2308536f66
360 changed files with 136381 additions and 0 deletions

61
Dockerfile.cpu-full Normal file
View File

@ -0,0 +1,61 @@
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"]