test vector store on docker

This commit is contained in:
binary-husky
2023-12-08 22:22:01 +08:00
parent 21bccf69d2
commit 0cb7dd5280
5 changed files with 43 additions and 7 deletions

View File

@@ -23,11 +23,13 @@ RUN pip3 install -r requirements.txt
# 装载项目文件,安装剩余依赖(必要)
RUN pip3 install torch --index-url https://download.pytorch.org/whl/cpu
RUN pip3 install langchain sentence-transformers unstructured[local-inference] faiss-cpu nltk beautifulsoup4 bitsandbytes tabulate icetk
COPY .cache /root/.cache
COPY . .
RUN pip3 install -r requirements.txt
# 非必要步骤,用于预热模块(可以删除)
RUN python3 -c 'from check_proxy import warm_up_modules; warm_up_modules()'
# 启动(必要)
CMD ["python3", "-u", "tests/test_vector_plugins.py"]
CMD ["python3", "-u", "main.py"]