Unify dependencies in agent image (flyteorg#3008)

* Install all flyteagent image dependencies in a single command

Signed-off-by: Eduardo Apolinario <eapolinario@users.noreply.github.com>

* test - Can I run docker commands in gh workflows?

Signed-off-by: Eduardo Apolinario <eapolinario@users.noreply.github.com>

* Remove `-it` flag from `docker run`

Signed-off-by: Eduardo Apolinario <eapolinario@users.noreply.github.com>

* Remove quick test after confirming that running `docker run --rm ghcr.io/flyteorg/flyteagent:1.14.0 pyflyte serve agent --port 8000 --timeout 1` in a gh workflow works

Signed-off-by: Eduardo Apolinario <eapolinario@users.noreply.github.com>

---------

Signed-off-by: Eduardo Apolinario <eapolinario@users.noreply.github.com>
Co-authored-by: Eduardo Apolinario <eapolinario@users.noreply.github.com>

.github/workflows/pythonpublish.yml

@@ -222,6 +222,9 @@ jobs:
           file: ./Dockerfile.agent
           cache-from: type=gha
           cache-to: type=gha,mode=max
+      - name: Confirm Agent can start
+        run: |
+          docker run --rm ghcr.io/${{ github.repository_owner }}/flyteagent:${{ github.sha }} pyflyte serve agent --port 8000 --timeout 1
       - name: Push flyteagent-all Image to GitHub Registry
         uses: docker/build-push-action@v2
         with:

Dockerfile.agent

@@ -8,14 +8,15 @@ ARG VERSION
 RUN apt-get update && apt-get install build-essential -y \
     && pip install uv
 
-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-k8sdataservice==flytekitplugins-k8sdataservice \
   flytekitplugins-openai==$VERSION \
   flytekitplugins-snowflake==$VERSION \
   flytekitplugins-awssagemaker==$VERSION \
+  prometheus-client \
+  grpcio-health-checking \
   && apt-get clean autoclean \
   && apt-get autoremove --yes \
   && rm -rf /var/lib/{apt,dpkg,cache,log}/ \