diff --git a/Dockerfile.external b/Dockerfile.external
index f3afac45..8767654f 100644
--- a/Dockerfile.external
+++ b/Dockerfile.external
@@ -14,7 +14,8 @@ FROM base as dependencies
 WORKDIR /home/worker/app
 COPY pyproject.toml poetry.lock ./
 
-RUN poetry install --extras "ui vector-stores-qdrant llms-ollama embeddings-ollama"
+ARG POETRY_EXTRAS="ui vector-stores-qdrant llms-ollama embeddings-ollama"
+RUN poetry install --extras "${POETRY_EXTRAS}"
 
 FROM base as app
 
diff --git a/Dockerfile.local b/Dockerfile.local
index 8c5c0737..49e8d412 100644
--- a/Dockerfile.local
+++ b/Dockerfile.local
@@ -24,7 +24,8 @@ FROM base as dependencies
 WORKDIR /home/worker/app
 COPY pyproject.toml poetry.lock ./
 
-RUN poetry install --extras "ui embeddings-huggingface llms-llama-cpp vector-stores-qdrant"
+ARG POETRY_EXTRAS="ui embeddings-huggingface llms-llama-cpp vector-stores-qdrant"
+RUN poetry install --extras "${POETRY_EXTRAS}"
 
 FROM base as app