2023-12-29 09:10:01 +08:00
|
|
|
torch~=2.1.2
|
|
|
|
|
torchvision~=0.16.2
|
|
|
|
|
torchaudio~=2.1.2
|
2023-12-29 09:31:16 +08:00
|
|
|
xformers==0.0.23.post1
|
|
|
|
|
transformers==4.36.2
|
|
|
|
|
sentence_transformers==2.2.2
|
2023-12-23 11:36:11 +08:00
|
|
|
|
2024-01-11 19:58:25 +08:00
|
|
|
langchain==0.0.354
|
2023-12-29 09:10:01 +08:00
|
|
|
langchain-experimental==0.0.47
|
2023-12-01 11:34:41 +08:00
|
|
|
pydantic==1.10.13
|
2023-12-29 09:10:01 +08:00
|
|
|
fschat==0.2.34
|
2024-01-11 19:58:25 +08:00
|
|
|
openai~=1.7.1
|
|
|
|
|
fastapi~=0.108.0
|
|
|
|
|
sse_starlette==1.8.2
|
2023-11-06 22:11:55 +08:00
|
|
|
nltk>=3.8.1
|
2023-12-01 11:34:41 +08:00
|
|
|
uvicorn>=0.24.0.post1
|
2024-01-11 19:58:25 +08:00
|
|
|
starlette~=0.32.0
|
2023-11-30 22:23:15 +08:00
|
|
|
unstructured[all-docs]==0.11.0
|
2023-08-12 10:44:04 +08:00
|
|
|
python-magic-bin; sys_platform == 'win32'
|
|
|
|
|
SQLAlchemy==2.0.19
|
2024-01-11 19:58:25 +08:00
|
|
|
faiss-cpu~=1.7.4 # `conda install faiss-gpu -c conda-forge` if you want to accelerate with gpus
|
|
|
|
|
accelerate~=0.24.1
|
2023-12-29 09:10:01 +08:00
|
|
|
spacy~=3.7.2
|
|
|
|
|
PyMuPDF~=1.23.8
|
2024-01-11 19:58:25 +08:00
|
|
|
rapidocr_onnxruntime==1.3.8
|
|
|
|
|
requests~=2.31.0
|
|
|
|
|
pathlib~=1.0.1
|
|
|
|
|
pytest~=7.4.3
|
|
|
|
|
numexpr~=2.8.6 # max version for py38
|
|
|
|
|
strsimpy~=0.2.1
|
|
|
|
|
markdownify~=0.11.6
|
2023-12-29 09:31:16 +08:00
|
|
|
tiktoken~=0.5.2
|
2023-12-01 11:34:41 +08:00
|
|
|
tqdm>=4.66.1
|
2023-12-23 11:36:11 +08:00
|
|
|
websockets>=12.0
|
2024-01-11 19:58:25 +08:00
|
|
|
numpy~=1.24.4
|
|
|
|
|
pandas~=2.0.3
|
2023-12-01 11:34:41 +08:00
|
|
|
einops>=0.7.0
|
|
|
|
|
transformers_stream_generator==0.0.4
|
2023-12-29 09:10:01 +08:00
|
|
|
vllm==0.2.6; sys_platform == "linux"
|
2024-01-11 19:58:25 +08:00
|
|
|
httpx[brotli,http2,socks]==0.25.2
|
|
|
|
|
llama-index
|
2023-10-04 12:05:46 +08:00
|
|
|
|
2023-12-04 09:39:56 +08:00
|
|
|
# optional document loaders
|
|
|
|
|
|
2024-01-11 19:58:25 +08:00
|
|
|
# rapidocr_paddle[gpu]>=1.3.0.post5 # gpu accelleration for ocr of pdf and image files
|
|
|
|
|
jq==1.6.0 # for .json and .jsonl files. suggest `conda install jq` on windows
|
2023-12-29 09:31:16 +08:00
|
|
|
beautifulsoup4~=4.12.2 # for .mhtml files
|
2023-12-29 09:10:01 +08:00
|
|
|
pysrt~=1.1.2
|
2023-09-20 21:34:12 +08:00
|
|
|
|
2023-12-23 11:36:11 +08:00
|
|
|
# Online api libs dependencies
|
2023-12-19 15:59:41 +08:00
|
|
|
|
2024-01-11 19:58:25 +08:00
|
|
|
zhipuai==1.0.7 # zhipu
|
|
|
|
|
dashscope==1.13.6 # qwen
|
2023-12-29 09:31:16 +08:00
|
|
|
# volcengine>=1.0.119 # fangzhou
|
|
|
|
|
|
|
|
|
|
# uncomment libs if you want to use corresponding vector store
|
|
|
|
|
# pymilvus>=2.3.4
|
2023-12-23 11:36:11 +08:00
|
|
|
# psycopg2==2.9.9
|
2023-12-01 11:34:41 +08:00
|
|
|
# pgvector>=0.2.4
|
|
|
|
|
|
|
|
|
|
# Agent and Search Tools
|
|
|
|
|
|
2024-01-11 19:58:25 +08:00
|
|
|
arxiv~=2.1.0
|
|
|
|
|
youtube-search~=2.1.2
|
|
|
|
|
duckduckgo-search~=3.9.9
|
|
|
|
|
metaphor-python~=0.1.23
|