fix: Fix build docker image error (#1325)

eosphoros-ai · Mar 23, 2024 · fa06be6 · fa06be6
1 parent 8a17099
commit fa06be6
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git 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
@@ -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",