# API requirements

langchain>=0.0.343
langchain-experimental>=0.0.42
fschat[model_worker]>=0.2.33
xformers>=0.0.22.post7
openai>=1.3.5
sentence_transformers
transformers>=4.35.2
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
fastapi>=0.104
nltk>=3.8.1
uvicorn~=0.23.1
starlette~=0.27.0
pydantic<2
unstructured[all-docs]>=0.11.0
python-magic-bin; sys_platform == 'win32'
SQLAlchemy==2.0.19
faiss-cpu
accelerate
spacy
PyMuPDF
rapidocr_onnxruntime

requests
pathlib
pytest
numexpr
strsimpy
markdownify
tiktoken
tqdm
websockets
numpy~=1.24.4
pandas~=2.0.3
einops
transformers_stream_generator==0.0.4

vllm==0.2.2; sys_platform == "linux"

# online api libs dependencies

# zhipuai>=1.0.7
# dashscope>=1.10.0
# qianfan>=0.2.0
# volcengine>=1.0.106

# uncomment libs if you want to use corresponding vector store
# pymilvus==2.1.3 # requires milvus==2.1.3
# psycopg2
# pgvector

# WebUI requirements

streamlit~=1.28.2 # # on win, make sure write its path in environment variable
streamlit-option-menu>=0.3.6
streamlit-antd-components>=0.2.3
streamlit-chatbox>=1.1.11
streamlit-modal==0.1.0
streamlit-aggrid>=0.3.4.post3
httpx[brotli,http2,socks]~=0.24.1
watchdog