upgrade setuptools version to fix CVE-2024-6345 (opea-project#806)

Signed-off-by: chensuyue <[email protected]>
wangkl2 · Oct 22, 2024 · 6518c0f · 6518c0f
1 parent 1987d39
commit 6518c0f
Show file tree

Hide file tree

Showing 50 changed files with 56 additions and 55 deletions.
diff --git a/comps/asr/whisper/Dockerfile b/comps/asr/whisper/Dockerfile
@@ -13,7 +13,7 @@ ARG ARCH=cpu
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     if [ "${ARCH}" = "cpu" ]; then \
         pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu ; \
         pip install --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cpu -r /home/user/comps/asr/whisper/requirements.txt ; \

diff --git a/comps/asr/whisper/dependency/Dockerfile b/comps/asr/whisper/dependency/Dockerfile
@@ -19,7 +19,7 @@ COPY --chown=user:user comps /home/user/comps
 
 USER user
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/user/comps/asr/whisper/requirements.txt && \
     if [ "${ARCH}" = "cpu" ]; then \
         pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu ; \

diff --git a/comps/chathistory/mongo/Dockerfile b/comps/chathistory/mongo/Dockerfile
@@ -19,7 +19,7 @@ USER user
 COPY comps /home/user/comps
 COPY requirements.txt /home/user/
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/user/comps/chathistory/mongo/requirements.txt && \
     pip install --no-cache-dir -r /home/user/requirements.txt
 

diff --git a/comps/embeddings/mosec/langchain/Dockerfile b/comps/embeddings/mosec/langchain/Dockerfile
@@ -15,7 +15,7 @@ USER user
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/user/comps/embeddings/mosec/langchain/requirements.txt
 
 ENV PYTHONPATH=$PYTHONPATH:/home/user

diff --git a/comps/embeddings/mosec/langchain/dependency/Dockerfile b/comps/embeddings/mosec/langchain/dependency/Dockerfile
@@ -1,21 +1,22 @@
 # Copyright (C) 2024 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
-From ubuntu:22.04
+FROM python:3.11-slim
+
 RUN useradd -m -s /bin/bash user && \
     mkdir -p /home/user && \
     chown -R user /home/user/
+
 ARG DEBIAN_FRONTEND=noninteractive
 
 ENV GLIBC_TUNABLES glibc.cpu.x86_shstk=permissive
-RUN apt update && apt install -y python3 python3-pip 
 
 COPY comps /home/user/comps
 
-RUN pip3 install torch==2.2.2 torchvision --index-url https://download.pytorch.org/whl/cpu 
-RUN pip3 install intel-extension-for-pytorch==2.2.0
-RUN pip3 install transformers
-RUN pip3 install llmspec mosec
+RUN pip install --no-cache-dir --upgrade pip setuptools
+RUN pip install --no-cache-dir torch==2.2.2 torchvision --index-url https://download.pytorch.org/whl/cpu
+RUN pip install --no-cache-dir intel-extension-for-pytorch==2.2.0
+RUN pip install --no-cache-dir transformers llmspec mosec
 
 RUN cd /home/user/ && export HF_ENDPOINT=https://hf-mirror.com && huggingface-cli download --resume-download BAAI/bge-large-zh-v1.5 --local-dir /home/user/bge-large-zh-v1.5
 USER user

diff --git a/comps/embeddings/multimodal/multimodal_langchain/Dockerfile b/comps/embeddings/multimodal/multimodal_langchain/Dockerfile
@@ -15,7 +15,7 @@ USER user
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/user/comps/embeddings/multimodal/multimodal_langchain/requirements.txt
 
 ENV PYTHONPATH=$PYTHONPATH:/home/user

diff --git a/comps/embeddings/multimodal_clip/Dockerfile b/comps/embeddings/multimodal_clip/Dockerfile
@@ -17,7 +17,7 @@ USER user
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     if [ ${ARCH} = "cpu" ]; then pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu; fi && \
     pip install --no-cache-dir -r /home/user/comps/embeddings/multimodal_clip/requirements.txt
 

diff --git a/comps/embeddings/predictionguard/Dockerfile b/comps/embeddings/predictionguard/Dockerfile
@@ -5,7 +5,7 @@ FROM python:3.11-slim
 
 COPY comps /home/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/comps/embeddings/predictionguard/requirements.txt
 
 ENV PYTHONPATH=$PYTHONPATH:/home

diff --git a/comps/embeddings/tei/langchain/Dockerfile b/comps/embeddings/tei/langchain/Dockerfile
@@ -17,7 +17,7 @@ USER user
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     if [ ${ARCH} = "cpu" ]; then pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu; fi && \
     pip install --no-cache-dir -r /home/user/comps/embeddings/tei/langchain/requirements.txt
 

diff --git a/comps/feedback_management/mongo/Dockerfile b/comps/feedback_management/mongo/Dockerfile
@@ -19,7 +19,7 @@ USER user
 COPY comps /home/user/comps
 COPY requirements.txt /home/user/
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/user/comps/feedback_management/mongo/requirements.txt && \
     pip install --no-cache-dir -r /home/user/requirements.txt
 

diff --git a/comps/guardrails/factuality/predictionguard/Dockerfile b/comps/guardrails/factuality/predictionguard/Dockerfile
@@ -5,7 +5,7 @@ FROM python:3.11-slim
 
 COPY comps /home/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/comps/guardrails/factuality/predictionguard/requirements.txt
 
 ENV PYTHONPATH=$PYTHONPATH:/home

diff --git a/comps/guardrails/llama_guard/langchain/Dockerfile b/comps/guardrails/llama_guard/langchain/Dockerfile
@@ -19,7 +19,7 @@ USER user
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     if [ ${ARCH} = "cpu" ]; then pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu; fi && \
     pip install --no-cache-dir -r /home/user/comps/guardrails/llama_guard/langchain/requirements.txt
 

diff --git a/comps/guardrails/pii_detection/predictionguard/Dockerfile b/comps/guardrails/pii_detection/predictionguard/Dockerfile
@@ -6,7 +6,7 @@ FROM python:3.11-slim
 
 COPY comps /home/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/comps/guardrails/pii_detection/predictionguard/requirements.txt
 
 ENV PYTHONPATH=$PYTHONPATH:/home

diff --git a/comps/guardrails/prompt_injection/predictionguard/Dockerfile b/comps/guardrails/prompt_injection/predictionguard/Dockerfile
@@ -6,7 +6,7 @@ FROM python:3.11-slim
 
 COPY comps /home/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/comps/guardrails/prompt_injection/predictionguard/requirements.txt
 
 ENV PYTHONPATH=$PYTHONPATH:/home

diff --git a/comps/guardrails/toxicity_detection/Dockerfile b/comps/guardrails/toxicity_detection/Dockerfile
@@ -20,7 +20,7 @@ USER user
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     if [ ${ARCH} = "cpu" ]; then pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu; fi && \
     pip install --no-cache-dir -r /home/user/comps/guardrails/toxicity_detection/requirements.txt
 

diff --git a/comps/guardrails/toxicity_detection/predictionguard/Dockerfile b/comps/guardrails/toxicity_detection/predictionguard/Dockerfile
@@ -6,7 +6,7 @@ FROM python:3.11-slim
 
 COPY comps /home/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/comps/guardrails/toxicity_detection/predictionguard/requirements.txt
 
 ENV PYTHONPATH=$PYTHONPATH:/home

diff --git a/comps/guardrails/wildguard/langchain/Dockerfile b/comps/guardrails/wildguard/langchain/Dockerfile
@@ -19,7 +19,7 @@ USER user
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     if [ ${ARCH} = "cpu" ]; then pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu; fi && \
     pip install --no-cache-dir -r /home/user/comps/guardrails/wildguard/langchain/requirements.txt
 

diff --git a/comps/image2video/Dockerfile b/comps/image2video/Dockerfile
@@ -8,7 +8,7 @@ ENV LANG=en_US.UTF-8
 
 COPY comps /home/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/comps/image2video/requirements.txt
 
 ENV PYTHONPATH=$PYTHONPATH:/home

diff --git a/comps/image2video/dependency/Dockerfile b/comps/image2video/dependency/Dockerfile
@@ -11,7 +11,7 @@ ARG ARCH="cpu"
 COPY comps /home/comps
 
 RUN apt-get update && apt-get install python3-opencv -y && \
-    pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir --upgrade pip setuptools && \
     if [ ${ARCH} = "cpu" ]; then pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu; fi && \
     pip install --no-cache-dir -r /home/comps/image2video/dependency/requirements.txt
 

diff --git a/comps/intent_detection/langchain/Dockerfile b/comps/intent_detection/langchain/Dockerfile
@@ -15,7 +15,7 @@ USER user
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/user/comps/intent_detection/langchain/requirements.txt
 
 ENV PYTHONPATH=$PYTHONPATH:/home/user

diff --git a/comps/llms/faq-generation/tgi/langchain/Dockerfile b/comps/llms/faq-generation/tgi/langchain/Dockerfile
@@ -15,7 +15,7 @@ USER user
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/user/comps/llms/faq-generation/tgi/langchain/requirements.txt
 
 ENV PYTHONPATH=$PYTHONPATH:/home/user

diff --git a/comps/llms/summarization/tgi/langchain/Dockerfile b/comps/llms/summarization/tgi/langchain/Dockerfile
@@ -17,7 +17,7 @@ USER user
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     if [ ${ARCH} = "cpu" ]; then pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu; fi && \
     pip install --no-cache-dir -r /home/user/comps/llms/summarization/tgi/langchain/requirements.txt
 

diff --git a/comps/llms/text-generation/ollama/langchain/Dockerfile b/comps/llms/text-generation/ollama/langchain/Dockerfile
@@ -16,7 +16,7 @@ USER user
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/user/comps/llms/text-generation/ollama/langchain/requirements.txt
 
 ENV PYTHONPATH=$PYTHONPATH:/home/user

diff --git a/comps/llms/text-generation/predictionguard/Dockerfile b/comps/llms/text-generation/predictionguard/Dockerfile
@@ -5,7 +5,7 @@ FROM python:3.11-slim
 
 COPY comps /home/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/comps/llms/text-generation/predictionguard/requirements.txt
 
 ENV PYTHONPATH=$PYTHONPATH:/home

diff --git a/comps/llms/text-generation/tgi/Dockerfile b/comps/llms/text-generation/tgi/Dockerfile
@@ -15,7 +15,7 @@ USER user
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/user/comps/llms/text-generation/tgi/requirements.txt
 
 ENV PYTHONPATH=$PYTHONPATH:/home/user

diff --git a/comps/llms/text-generation/vllm/langchain/Dockerfile b/comps/llms/text-generation/vllm/langchain/Dockerfile
@@ -17,7 +17,7 @@ USER user
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     if [ ${ARCH} = "cpu" ]; then \
       pip install --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cpu -r /home/user/comps/llms/text-generation/vllm/langchain/requirements.txt; \
     else \

diff --git a/comps/llms/text-generation/vllm/ray/Dockerfile b/comps/llms/text-generation/vllm/ray/Dockerfile
@@ -15,7 +15,7 @@ USER user
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/user/comps/llms/text-generation/vllm/ray/requirements.txt
 
 ENV PYTHONPATH=$PYTHONPATH:/home/user

diff --git a/comps/lvms/llava/Dockerfile b/comps/lvms/llava/Dockerfile
@@ -15,7 +15,7 @@ USER user
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     if [ ${ARCH} = "cpu" ]; then \
       pip install --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cpu -r /home/user/comps/lvms/llava/requirements.txt; \
     else \

diff --git a/comps/lvms/llava/dependency/Dockerfile b/comps/lvms/llava/dependency/Dockerfile
@@ -12,7 +12,7 @@ ENV PYTHONPATH=/home/user:/usr/lib/habanalabs/:/optimum-habana
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/user/comps/lvms/llava/requirements.txt
 
 ENV PYTHONPATH=$PYTHONPATH:/home/user

diff --git a/comps/lvms/predictionguard/Dockerfile b/comps/lvms/predictionguard/Dockerfile
@@ -8,7 +8,7 @@ ENV LANG=en_US.UTF-8
 
 COPY comps /home/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/comps/lvms/predictionguard/requirements.txt
 
 ENV PYTHONPATH=$PYTHONPATH:/home

diff --git a/comps/lvms/video-llama/Dockerfile b/comps/lvms/video-llama/Dockerfile
@@ -8,7 +8,7 @@ ENV LANG=en_US.UTF-8
 
 COPY comps /home/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/comps/lvms/video-llama/requirements.txt
 
 ENV PYTHONPATH=$PYTHONPATH:/home

diff --git a/comps/prompt_registry/mongo/Dockerfile b/comps/prompt_registry/mongo/Dockerfile
@@ -19,7 +19,7 @@ USER user
 COPY comps /home/user/comps
 COPY requirements.txt /home/user/
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/user/comps/prompt_registry/mongo/requirements.txt && \
     pip install --no-cache-dir -r /home/user/requirements.txt
 

diff --git a/comps/ragas/tgi/langchain/Dockerfile b/comps/ragas/tgi/langchain/Dockerfile
@@ -15,7 +15,7 @@ USER user
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/user/comps/ragas/tgi/langchain/requirements.txt
 
 ENV PYTHONPATH=$PYTHONPATH:/home/user

diff --git a/comps/reranks/mosec/langchain/Dockerfile b/comps/reranks/mosec/langchain/Dockerfile
@@ -15,7 +15,7 @@ USER user
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/user/comps/reranks/mosec/langchain/requirements.txt
 
 ENV PYTHONPATH=$PYTHONPATH:/home/user

diff --git a/comps/reranks/tei/Dockerfile b/comps/reranks/tei/Dockerfile
@@ -19,7 +19,7 @@ USER user
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     if [ ${ARCH} = "cpu" ]; then pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu; fi && \
     pip install --no-cache-dir -r /home/user/comps/reranks/tei/requirements.txt
 

diff --git a/comps/reranks/videoqna/Dockerfile b/comps/reranks/videoqna/Dockerfile
@@ -14,7 +14,7 @@ USER user
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     pip install --no-cache-dir -r /home/user/comps/reranks/videoqna/requirements.txt
 
 ENV PYTHONPATH=$PYTHONPATH:/home/user

diff --git a/comps/retrievers/milvus/langchain/Dockerfile b/comps/retrievers/milvus/langchain/Dockerfile
@@ -20,7 +20,7 @@ USER user
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     if [ ${ARCH} = "cpu" ]; then pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu; fi && \
     pip install --no-cache-dir -r /home/user/comps/retrievers/milvus/langchain/requirements.txt
 

diff --git a/comps/retrievers/multimodal/redis/langchain/Dockerfile b/comps/retrievers/multimodal/redis/langchain/Dockerfile
@@ -17,7 +17,7 @@ COPY comps /home/user/comps
 
 USER user
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     if [ ${ARCH} = "cpu" ]; then pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu; fi && \
     pip install --no-cache-dir -r /home/user/comps/retrievers/multimodal/redis/langchain/requirements.txt
 

diff --git a/comps/retrievers/neo4j/langchain/Dockerfile b/comps/retrievers/neo4j/langchain/Dockerfile
@@ -22,7 +22,7 @@ USER user
 
 COPY comps /home/user/comps
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     if [ ${ARCH} = "cpu" ]; then pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu; fi && \
     pip install --no-cache-dir -r /home/user/comps/retrievers/neo4j/langchain/requirements.txt
 

diff --git a/comps/retrievers/pathway/langchain/Dockerfile b/comps/retrievers/pathway/langchain/Dockerfile
@@ -17,7 +17,7 @@ COPY comps /home/user/comps
 
 USER user
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     if [ ${ARCH} = "cpu" ]; then pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu; fi && \
     pip install --no-cache-dir -r /home/user/comps/retrievers/pathway/langchain/requirements.txt
 

diff --git a/comps/retrievers/pgvector/langchain/Dockerfile b/comps/retrievers/pgvector/langchain/Dockerfile
@@ -17,7 +17,7 @@ COPY comps /home/user/comps
 
 USER user
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     if [ ${ARCH} = "cpu" ]; then pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu; fi && \
     pip install --no-cache-dir -r /home/user/comps/retrievers/pgvector/langchain/requirements.txt
 

diff --git a/comps/retrievers/pinecone/langchain/Dockerfile b/comps/retrievers/pinecone/langchain/Dockerfile
@@ -17,7 +17,7 @@ COPY comps /home/user/comps
 
 USER user
 
-RUN pip install --no-cache-dir --upgrade pip && \
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
     if [ ${ARCH} = "cpu" ]; then pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu; fi && \
     pip install --no-cache-dir -r /home/user/comps/retrievers/pinecone/langchain/requirements.txt