wangtao
/
autogen
mirror of https://github.com/microsoft/autogen.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update RetrieveChat extra dependencies (#2449)

This commit is contained in:
Li Jiang 2024-04-21 02:32:57 +08:00 committed by GitHub
parent ded2d612c3
commit b7366b570f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
setup.py
View File

@ -35,6 +35,8 @@ jupyter_executor = [
"ipykernel>=6.29.0",
]
rag = ["sentence_transformers", "pypdf", "ipython", "beautifulsoup4", "markdownify"]
setuptools.setup(
name="pyautogen",
version=__version__,
@ -59,24 +61,9 @@ setuptools.setup(
],
"blendsearch": ["flaml[blendsearch]"],
"mathchat": ["sympy", "pydantic==1.10.9", "wolframalpha"],
"retrievechat": ["chromadb", "sentence_transformers", "pypdf", "ipython", "beautifulsoup4", "markdownify"],
"retrievechat-pgvector": [
"pgvector>=0.2.5",
"psycopg>=3.1.18",
"sentence_transformers",
"pypdf",
"ipython",
"beautifulsoup4",
"markdownify",
],
"retrievechat-qdrant": [
"qdrant_client[fastembed]",
"sentence_transformers",
"pypdf",
"ipython",
"beautifulsoup4",
"markdownify",
],
"retrievechat": ["chromadb"] + rag,
"retrievechat-pgvector": ["pgvector>=0.2.5", "psycopg>=3.1.18"] + rag,
"retrievechat-qdrant": ["qdrant_client[fastembed]"] + rag,
"autobuild": ["chromadb", "sentence-transformers", "huggingface-hub"],
"teachable": ["chromadb"],
"lmm": ["replicate", "pillow"],