diff --git a/Dockerfile.agent b/Dockerfile.agent
index e2d106f7c2..7faa049336 100644
--- a/Dockerfile.agent
+++ b/Dockerfile.agent
@@ -5,10 +5,11 @@ LABEL org.opencontainers.image.source=https://github.com/flyteorg/flytekit
 
 ARG VERSION
 
-RUN apt-get update && apt-get install build-essential -y
+RUN apt-get update && apt-get install build-essential -y \
+    && pip install uv
 
-RUN pip install prometheus-client grpcio-health-checking
-RUN pip install --no-cache-dir -U flytekit==$VERSION \
+RUN uv pip install --system prometheus-client grpcio-health-checking
+RUN uv pip install --system --no-cache-dir -U flytekit==$VERSION \
   flytekitplugins-airflow==$VERSION \
   flytekitplugins-bigquery==$VERSION \
   flytekitplugins-openai==$VERSION \
@@ -24,6 +25,6 @@ CMD ["pyflyte", "serve", "agent", "--port", "8000"]
 FROM agent-slim AS agent-all
 ARG VERSION
 
-RUN pip install --no-cache-dir -U \
+RUN uv pip install --system --no-cache-dir -U \
   flytekitplugins-mmcloud==$VERSION \
   flytekitplugins-spark==$VERSION