From 911b9c527536455c30c7e0bcef56d63ea56e04cf Mon Sep 17 00:00:00 2001
From: Li Jiang <bnujli@gmail.com>
Date: Sun, 21 Apr 2024 02:32:57 +0800
Subject: [PATCH] Update RetrieveChat extra dependencies (#2449)

---
 setup.py | 23 +++++------------------
 1 file changed, 5 insertions(+), 18 deletions(-)

diff --git a/setup.py b/setup.py
index cea36b7052d..674df273883 100644
--- 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"],