Use uv in agent image (#2780)

Signed-off-by: Eduardo Apolinario <[email protected]>
Co-authored-by: Eduardo Apolinario <[email protected]>

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