From cbf5c4e29570c55a4be7ebafee78a5cbd695fbce Mon Sep 17 00:00:00 2001 From: Raoxiang Date: Thu, 1 Aug 2024 14:37:11 +0800 Subject: [PATCH] Fix DBCopilot environment Vulnerabilities --- .../environments/dbcopilot_embeddings/context/Dockerfile | 6 +++++- .../dbcopilot/environments/dbcopilot_mir/context/Dockerfile | 3 ++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/assets/large_language_models/dbcopilot/environments/dbcopilot_embeddings/context/Dockerfile b/assets/large_language_models/dbcopilot/environments/dbcopilot_embeddings/context/Dockerfile index 1ed5d07baf..a6a001730b 100644 --- a/assets/large_language_models/dbcopilot/environments/dbcopilot_embeddings/context/Dockerfile +++ b/assets/large_language_models/dbcopilot/environments/dbcopilot_embeddings/context/Dockerfile @@ -13,7 +13,11 @@ RUN apt-get update && apt-get install -y \ libc6=2.31-0ubuntu9.16 \ libc-dev-bin=2.31-0ubuntu9.16 \ git \ - git-man &&\ + git-man \ + libpython3.8-minimal=3.8.10-0ubuntu1~20.04.10 \ + libpython3.8-stdlib=3.8.10-0ubuntu1~20.04.10\ + python3.8-minimal=3.8.10-0ubuntu1~20.04.10 \ + python3.8=3.8.10-0ubuntu1~20.04.10 &&\ rm -rf /var/lib/apt/lists/* # Install MS SQL ODBC Driver diff --git a/assets/large_language_models/dbcopilot/environments/dbcopilot_mir/context/Dockerfile b/assets/large_language_models/dbcopilot/environments/dbcopilot_mir/context/Dockerfile index 1abddd9c2d..1462f12af0 100644 --- a/assets/large_language_models/dbcopilot/environments/dbcopilot_mir/context/Dockerfile +++ b/assets/large_language_models/dbcopilot/environments/dbcopilot_mir/context/Dockerfile @@ -27,7 +27,8 @@ RUN python -m pip install --upgrade pip && \ pip install promptflow-image-bundle[azure]==0.3.0 && \ pip install promptflow-vectordb==0.2.10 && \ ## Install dbcopilot - pip install db-copilot-tool==0.0.8 + pip install db-copilot-tool==0.0.8 \ + pip install setuptools==70.0.0 # Fix vunerabilities RUN /bin/bash -c "source /opt/miniconda/etc/profile.d/conda.sh && \