update

Signed-off-by: Future-Outlier <[email protected]>

Dockerfile.agent

@@ -8,15 +8,13 @@ ARG VERSION
 RUN apt-get update && apt-get install build-essential -y \
     && pip install uv
 
-RUN uv pip install --system --no-cache-dir -U flytekit==$VERSION \
+RUN uv pip install --system --no-cache-dir -U flytekit[agent]==$VERSION \
   flytekitplugins-airflow==$VERSION \
   flytekitplugins-bigquery==$VERSION \
   flytekitplugins-k8sdataservice==$VERSION \
   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}/ \

flytekit/clis/sdk_in_container/serve.py

@@ -68,13 +68,6 @@ def agent(_: click.Context, port, prometheus_port, worker, timeout, modules):
     """
     import asyncio
 
-    # try:
-    #     from flytekit.extras.webhook import WebhookAgent  # noqa: F401 Webhook Agent Registration
-    # except ImportError:
-    #     from flytekit import logger
-
-    #     logger.debug("Flyte Agent Dependency is not installed. Please install it using `pip install flytekit[agent]`")
-
     working_dir = os.getcwd()
     if all(os.path.realpath(path) != working_dir for path in sys.path):
         sys.path.append(working_dir)
@@ -89,7 +82,9 @@ async def _start_grpc_server(port: int, prometheus_port: int, worker: int, timeo
         from flytekit.extend.backend.agent_service import AgentMetadataService, AsyncAgentService, SyncAgentService
         from flytekit.extras.webhook import WebhookAgent  # noqa: F401 Webhook Agent Registration
     except ImportError as e:
-        raise ImportError("Flyte agent dependencies are not installed. Please install it using `pip install flytekit[agent]`") from e
+        raise ImportError(
+            "Flyte agent dependencies are not installed. Please install it using `pip install flytekit[agent]`"
+        ) from e
 
     click.secho("🚀 Starting the agent service...")
     _start_http_server(prometheus_port)