diff --git a/docker/Dockerfile b/docker/Dockerfile
index 72ea5ce19..05932a58d 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -64,7 +64,7 @@ COPY . /tmp/aider
 
 # Install dependencies as root
 RUN /venv/bin/python -m pip install --upgrade --no-cache-dir pip && \
-    /venv/bin/python -m pip install --no-cache-dir /tmp/aider[playwright] boto3 \
+    /venv/bin/python -m pip install --no-cache-dir /tmp/aider[playwright] boto3 google-cloud-aiplatform \
        --extra-index-url https://download.pytorch.org/whl/cpu && \
     rm -rf /tmp/aider