2023-11-06 22:11:55 +08:00
|
|
|
# API requirements
|
|
|
|
|
|
2023-11-29 15:08:13 +08:00
|
|
|
langchain>=0.0.342
|
2023-11-24 22:47:55 +08:00
|
|
|
langchain-experimental>=0.0.42
|
|
|
|
|
fschat[model_worker]>=0.2.33
|
|
|
|
|
xformers>=0.0.22.post7
|
2023-11-29 15:08:13 +08:00
|
|
|
openai>=1.3.5
|
2023-07-27 23:22:07 +08:00
|
|
|
sentence_transformers
|
2023-11-24 22:47:55 +08:00
|
|
|
transformers>=4.35.2
|
2023-11-25 19:56:36 +08:00
|
|
|
torch==2.1.0 ##on win, install the cuda version manually if you want use gpu
|
|
|
|
|
torchvision #on win, install the cuda version manually if you want use gpu
|
|
|
|
|
torchaudio #on win, install the cuda version manually if you want use gpu
|
2023-10-20 18:13:55 +08:00
|
|
|
fastapi>=0.104
|
|
|
|
|
nltk>=3.8.1
|
2023-07-27 23:22:07 +08:00
|
|
|
uvicorn~=0.23.1
|
|
|
|
|
starlette~=0.27.0
|
2023-11-24 22:47:55 +08:00
|
|
|
pydantic<2
|
|
|
|
|
unstructured[all-docs]>=0.11.0
|
2023-08-08 12:03:33 +08:00
|
|
|
python-magic-bin; sys_platform == 'win32'
|
2023-08-10 15:51:22 +08:00
|
|
|
SQLAlchemy==2.0.19
|
2023-08-07 20:37:16 +08:00
|
|
|
faiss-cpu
|
2023-08-25 22:14:56 +08:00
|
|
|
accelerate
|
|
|
|
|
spacy
|
2023-10-07 22:00:56 +08:00
|
|
|
PyMuPDF
|
|
|
|
|
rapidocr_onnxruntime
|
2023-08-08 14:29:08 +08:00
|
|
|
|
2023-09-14 22:59:05 +08:00
|
|
|
requests
|
|
|
|
|
pathlib
|
|
|
|
|
pytest
|
|
|
|
|
numexpr
|
2023-10-25 08:58:31 +08:00
|
|
|
strsimpy
|
|
|
|
|
markdownify
|
2023-11-06 22:11:55 +08:00
|
|
|
tiktoken
|
|
|
|
|
tqdm
|
|
|
|
|
websockets
|
|
|
|
|
numpy~=1.24.4
|
|
|
|
|
pandas~=2.0.3
|
|
|
|
|
einops
|
|
|
|
|
transformers_stream_generator==0.0.4
|
2023-10-25 08:58:31 +08:00
|
|
|
|
2023-11-24 22:47:55 +08:00
|
|
|
vllm==0.2.2; sys_platform == "linux"
|
2023-10-25 08:58:31 +08:00
|
|
|
|
2023-11-24 22:47:55 +08:00
|
|
|
# online api libs dependencies
|
|
|
|
|
|
|
|
|
|
# zhipuai>=1.0.7
|
|
|
|
|
# dashscope>=1.10.0
|
|
|
|
|
# qianfan>=0.2.0
|
|
|
|
|
# volcengine>=1.0.106
|
2023-09-20 21:34:12 +08:00
|
|
|
|
2023-08-10 15:51:22 +08:00
|
|
|
# uncomment libs if you want to use corresponding vector store
|
|
|
|
|
# pymilvus==2.1.3 # requires milvus==2.1.3
|
2023-08-10 11:16:52 +08:00
|
|
|
# psycopg2
|
2023-08-10 15:51:22 +08:00
|
|
|
# pgvector
|
2023-08-12 10:44:04 +08:00
|
|
|
|
2023-11-06 22:11:55 +08:00
|
|
|
# WebUI requirements
|
|
|
|
|
|
2023-11-25 19:56:36 +08:00
|
|
|
streamlit~=1.28.2 # # on win, make sure write its path in environment variable
|
2023-08-12 10:44:04 +08:00
|
|
|
streamlit-option-menu>=0.3.6
|
2023-11-24 22:47:55 +08:00
|
|
|
streamlit-antd-components>=0.2.3
|
2023-11-03 18:56:27 +08:00
|
|
|
streamlit-chatbox>=1.1.11
|
2023-11-30 11:39:41 +08:00
|
|
|
streamlit-modal==0.1.0
|
2023-08-12 10:44:04 +08:00
|
|
|
streamlit-aggrid>=0.3.4.post3
|
2023-11-24 22:47:55 +08:00
|
|
|
httpx[brotli,http2,socks]~=0.24.1
|
2023-11-29 15:08:13 +08:00
|
|
|
watchdog
|