update

Signed-off-by: Future-Outlier <[email protected]>
flyteorg · Feb 27, 2025 · 5e2952b · 5e2952b
1 parent c3296c0
commit 5e2952b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flytekit/clis/sdk_in_container/serve.py b/flytekit/clis/sdk_in_container/serve.py
@@ -69,7 +69,7 @@ def agent(_: click.Context, port, prometheus_port, worker, timeout, modules):
     import asyncio
 
     try:
-        from flytekit.extras.webhook import WebhookTask  # noqa: F401
+        from flytekit.extras.webhook import WebhookAgent  # noqa: F401 Webhook Agent Registration
     except ImportError:
         from flytekit import logger