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():