Skip to content

Commit b7366b5

Browse files
authored
Update RetrieveChat extra dependencies (#2449)
1 parent ded2d61 commit b7366b5

File tree

1 file changed

+5
-18
lines changed

1 file changed

+5
-18
lines changed

setup.py

+5-18
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,8 @@
3535
"ipykernel>=6.29.0",
3636
]
3737

38+
rag = ["sentence_transformers", "pypdf", "ipython", "beautifulsoup4", "markdownify"]
39+
3840
setuptools.setup(
3941
name="pyautogen",
4042
version=__version__,
@@ -59,24 +61,9 @@
5961
],
6062
"blendsearch": ["flaml[blendsearch]"],
6163
"mathchat": ["sympy", "pydantic==1.10.9", "wolframalpha"],
62-
"retrievechat": ["chromadb", "sentence_transformers", "pypdf", "ipython", "beautifulsoup4", "markdownify"],
63-
"retrievechat-pgvector": [
64-
"pgvector>=0.2.5",
65-
"psycopg>=3.1.18",
66-
"sentence_transformers",
67-
"pypdf",
68-
"ipython",
69-
"beautifulsoup4",
70-
"markdownify",
71-
],
72-
"retrievechat-qdrant": [
73-
"qdrant_client[fastembed]",
74-
"sentence_transformers",
75-
"pypdf",
76-
"ipython",
77-
"beautifulsoup4",
78-
"markdownify",
79-
],
64+
"retrievechat": ["chromadb"] + rag,
65+
"retrievechat-pgvector": ["pgvector>=0.2.5", "psycopg>=3.1.18"] + rag,
66+
"retrievechat-qdrant": ["qdrant_client[fastembed]"] + rag,
8067
"autobuild": ["chromadb", "sentence-transformers", "huggingface-hub"],
8168
"teachable": ["chromadb"],
8269
"lmm": ["replicate", "pillow"],

0 commit comments

Comments
 (0)