codefuse-chatbot/requirements.txt

60 lines
932 B
Plaintext

langchain==0.0.266
openai==0.28.1
sentence_transformers
fschat==0.2.33
transformers>=4.31.0
# torch~=2.0.0
fastapi~=0.99.1
nltk~=3.8.1
uvicorn~=0.23.1
starlette~=0.27.0
pydantic~=1.10.11
unstructured[all-docs]
python-magic-bin; sys_platform == 'win32'
SQLAlchemy==2.0.19
faiss-cpu
nltk
loguru
pypdf
duckduckgo-search==3.9.11
pysocks
accelerate
docker
jupyter
notebook
websockets
fake_useragent
selenium
jsonref
# uncomment libs if you want to use corresponding vector store
# pymilvus==2.1.3 # requires milvus==2.1.3
# psycopg2
# pgvector
numpy~=1.24.4
pandas~=2.0.3
streamlit>=1.25.0
streamlit-option-menu>=0.3.6
streamlit-antd-components>=0.1.11
streamlit-chatbox>=1.1.6
streamlit-aggrid>=0.3.4.post3
httpx
javalang==0.13.0
jsonref==1.1.0
chromadb==0.4.17
nebula3-python==3.1.0
# qwen model
protobuf==3.20.*
transformers_stream_generator
einops
auto-gptq
optimum
modelscope
# vllm model
vllm==0.2.2; sys_platform == "linux"