From bd09ea626b973d1afa68cf56b10edac6da766b1a Mon Sep 17 00:00:00 2001
From: nv-sandipatil <122857913+spatil44@users.noreply.github.com>
Date: Wed, 4 Oct 2023 19:00:53 +0530
Subject: [PATCH] Changes related to
 https://github.com/nv-morpheus/Morpheus/issues/1234

---
 morpheus/stages/input/http_client_source_stage.py | 3 ++-
 morpheus/stages/input/http_server_source_stage.py | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/morpheus/stages/input/http_client_source_stage.py b/morpheus/stages/input/http_client_source_stage.py
index fa42f98172..c5222109a2 100644
--- a/morpheus/stages/input/http_client_source_stage.py
+++ b/morpheus/stages/input/http_client_source_stage.py
@@ -24,6 +24,7 @@
 import cudf
 
 from morpheus.cli.register_stage import register_stage
+from morpheus.config import PipelineModes
 from morpheus.config import Config
 from morpheus.messages import MessageMeta
 from morpheus.pipeline.preallocator_mixin import PreallocatorMixin
@@ -34,7 +35,7 @@
 logger = logging.getLogger(__name__)
 
 
-@register_stage("from-http-client", ignore_args=["query_params", "headers", "**request_kwargs"])
+@register_stage("from-http-client", modes=[PipelineModes.FIL, PipelineModes.NLP, PipelineModes.OTHER],ignore_args=["query_params", "headers", "**request_kwargs"])
 class HttpClientSourceStage(PreallocatorMixin, SingleOutputSource):
     """
     Source stage that polls a remote HTTP server for incoming data.
diff --git a/morpheus/stages/input/http_server_source_stage.py b/morpheus/stages/input/http_server_source_stage.py
index 48137b1bdc..8156ed908a 100644
--- a/morpheus/stages/input/http_server_source_stage.py
+++ b/morpheus/stages/input/http_server_source_stage.py
@@ -25,6 +25,7 @@
 import cudf
 
 from morpheus.cli.register_stage import register_stage
+from morpheus.config import PipelineModes
 from morpheus.config import Config
 from morpheus.messages import MessageMeta
 from morpheus.pipeline.preallocator_mixin import PreallocatorMixin
@@ -40,7 +41,7 @@
 SUPPORTED_METHODS = (HTTPMethod.POST, HTTPMethod.PUT)
 
 
-@register_stage("from-http")
+@register_stage("from-http", modes=[PipelineModes.FIL, PipelineModes.NLP, PipelineModes.OTHER])
 class HttpServerSourceStage(PreallocatorMixin, SingleOutputSource):
     """
     Source stage that starts an HTTP server and listens for incoming requests on a specified endpoint.