Merge branch 'new-webhook' into add-httpx-in-flytekit-for-webhook

flyteorg · Feb 27, 2025 · 11cbaa3 · 11cbaa3
2 parents 2ad2f28 + 75999fc
commit 11cbaa3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/flytekit/clis/sdk_in_container/serve.py b/flytekit/clis/sdk_in_container/serve.py
@@ -68,7 +68,11 @@ def agent(_: click.Context, port, prometheus_port, worker, timeout, modules):
     """
     import asyncio
 
-    from flytekit.extras.webhook import WebhookTask  # noqa: F401
+    try:
+        from flytekit.extras.webhook import WebhookTask  # noqa: F401
+    except ImportError:
+        from flytekit import logger
+        logger.debug("Webhook 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):

diff --git a/pyproject.toml b/pyproject.toml
@@ -148,3 +148,8 @@ extend-exclude = ["tests/", "**/tests/**"]
 [tool.codespell]
 ignore-words-list = "ot,te,raison,fo,lits,assertIn"
 skip = "./docs/build,./.git,*.txt"
+
+[project.optional-dependencies]
+agent = [
+    "httpx",
+]