Update RetrieveChat extra dependencies (microsoft#2449)

jayralencar · Apr 20, 2024 · 911b9c5 · 911b9c5
1 parent bb351e3
commit 911b9c5
Showing 1 changed file with 5 additions and 18 deletions.
diff --git a/setup.py b/setup.py
@@ -35,6 +35,8 @@
     "ipykernel>=6.29.0",
 ]
 
+rag = ["sentence_transformers", "pypdf", "ipython", "beautifulsoup4", "markdownify"]
+
 setuptools.setup(
     name="pyautogen",
     version=__version__,
@@ -59,24 +61,9 @@
         ],
         "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"],