github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-24 04:36:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: Fix build docker image error (#1325)

Browse Source
This commit is contained in:
Fangyin Cheng 2024-03-23 09:44:20 +08:00 committed by GitHub
parent 8a17099dd2
commit fa06be64c1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docker/base/Dockerfile
View File

@ -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"

7
setup.py
View File

@ -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",