Fix DBCopilot environment Vulnerabilities

Azure · Aug 1, 2024 · cbf5c4e · cbf5c4e
1 parent c733ef6
commit cbf5c4e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 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
@@ -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
@@ -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 && \