diff --git a/docker/base/Dockerfile b/docker/base/Dockerfile
index e4fad9697..b5be85c4f 100644
--- a/docker/base/Dockerfile
+++ b/docker/base/Dockerfile
@@ -3,7 +3,9 @@ ARG BASE_IMAGE="nvidia/cuda:11.8.0-runtime-ubuntu22.04"
 FROM ${BASE_IMAGE}
 ARG BASE_IMAGE
 
-RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y git python3 pip wget sqlite3 tzdata \
+# libpq-dev for PG
+RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y git python3 pip \
+    wget sqlite3 tzdata libpq-dev default-libmysqlclient-dev \
     && apt-get clean
 
 ARG BUILD_LOCAL_CODE="false"
diff --git a/setup.py b/setup.py
index df5a70c63..35da707d3 100644
--- a/setup.py
+++ b/setup.py
@@ -559,16 +559,17 @@ def all_datasource_requires():
     """
     pip install "dbgpt[datasource]"
     """
-
     setup_spec.extras["datasource"] = [
         # "sqlparse==0.4.4",
-        "pymssql",
         "pymysql",
-        "pyspark",
         "psycopg2",
         # for doris
         # mysqlclient 2.2.x have pkg-config issue on 3.10+
         "mysqlclient==2.1.0",
+    ]
+    setup_spec.extras["datasource_all"] = setup_spec.extras["datasource"] + [
+        "pyspark",
+        "pymssql",
         "pydoris>=1.0.2,<2.0.0",
         "clickhouse-connect",
         "pyhive",