From 410b81ecbef17a4da09af45adfc13fddaf0357cc Mon Sep 17 00:00:00 2001 From: Eduardo Apolinario <653394+eapolinario@users.noreply.github.com> Date: Wed, 2 Oct 2024 08:51:38 -0700 Subject: [PATCH] Use uv in agent image (#2780) Signed-off-by: Eduardo Apolinario Co-authored-by: Eduardo Apolinario --- Dockerfile.agent | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) 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