65 lines
1.2 KiB
Plaintext
65 lines
1.2 KiB
Plaintext
# API requirements
|
||
|
||
# On Windows system, install the cuda version manually from https://pytorch.org/
|
||
torch~=2.1.2
|
||
torchvision~=0.16.2
|
||
torchaudio~=2.1.2
|
||
|
||
langchain==0.0.352
|
||
langchain-experimental==0.0.47
|
||
pydantic==1.10.13
|
||
fschat==0.2.34
|
||
xformers==0.0.23.post1
|
||
openai~=1.6.0
|
||
sentence_transformers==2.2.2
|
||
transformers==4.36.2
|
||
fastapi>=0.105
|
||
nltk>=3.8.1
|
||
uvicorn>=0.24.0.post1
|
||
starlette~=0.27.0
|
||
unstructured[all-docs]==0.11.0
|
||
python-magic-bin; sys_platform == 'win32'
|
||
SQLAlchemy==2.0.19
|
||
faiss-cpu~=1.7.4
|
||
accelerate==0.24.1
|
||
spacy~=3.7.2
|
||
PyMuPDF~=1.23.8
|
||
rapidocr_onnxruntime~=1.3.8
|
||
requests>=2.31.0
|
||
pathlib>=1.0.1
|
||
pytest>=7.4.3
|
||
numexpr>=2.8.8
|
||
strsimpy>=0.2.1
|
||
markdownify>=0.11.6
|
||
tiktoken>=0.5.2
|
||
tqdm>=4.66.1
|
||
websockets>=12.0
|
||
numpy~=1.26.2
|
||
pandas~=2.1.4
|
||
einops>=0.7.0
|
||
transformers_stream_generator==0.0.4
|
||
vllm==0.2.6; sys_platform == "linux"
|
||
|
||
# optional document loaders
|
||
|
||
rapidocr_paddle[gpu]>=1.3.0.post5
|
||
jq>=1.6.0
|
||
html2text
|
||
beautifulsoup4~=4.12.2
|
||
pysrt~=1.1.2
|
||
|
||
# Online api libs dependencies
|
||
|
||
zhipuai>=1.0.7,<=2.0.0
|
||
dashscope>=1.13.6
|
||
volcengine>=1.0.119
|
||
pymilvus>=2.3.4
|
||
# psycopg2==2.9.9
|
||
# pgvector>=0.2.4
|
||
|
||
# Agent and Search Tools
|
||
|
||
arxiv>=2.0.0
|
||
youtube-search>=2.1.2
|
||
duckduckgo-search>=3.9.9
|
||
metaphor-python>=0.1.23 |