From 7d51a2e349238a7a0adf76581d428fd8f56c5a1e Mon Sep 17 00:00:00 2001
From: aries_ckt <916701291@qq.com>
Date: Tue, 26 Sep 2023 18:00:44 +0800
Subject: [PATCH] feat:spark conn

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 8fe47a614..ab9f33448 100644
--- a/setup.py
+++ b/setup.py
@@ -365,7 +365,7 @@ def all_datasource_requires():
     """
     pip install "db-gpt[datasource]"
     """
-    setup_spec.extras["datasource"] = ["pymssql", "pymysql"]
+    setup_spec.extras["datasource"] = ["pymssql", "pymysql", "pyspark"]
 
 
 def openai_requires():