From 8caf385b9b61d412ff7dc5d232be4114e227811a Mon Sep 17 00:00:00 2001
From: Yashash H L <yashashhl25@gmail.com>
Date: Fri, 20 Dec 2024 15:45:19 +0530
Subject: [PATCH] fix server info check

Signed-off-by: Yashash H L <yashashhl25@gmail.com>
---
 numaflow/src/shared.rs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/numaflow/src/shared.rs b/numaflow/src/shared.rs
index 7d9ecab..a940f8a 100644
--- a/numaflow/src/shared.rs
+++ b/numaflow/src/shared.rs
@@ -78,7 +78,10 @@ pub(crate) struct ServerInfo {
 impl ServerInfo {
     pub fn new(container_type: ContainerType) -> Self {
         let mut metadata: HashMap<String, String> = HashMap::new();
-        if container_type == ContainerType::Map || container_type == ContainerType::BatchMap {
+        if container_type == ContainerType::Map
+            || container_type == ContainerType::BatchMap
+            || container_type == ContainerType::MapStream
+        {
             metadata.insert(
                 MAP_MODE_KEY.to_string(),
                 match container_type {