diff --git a/src/backend/langflow/components/io/ChatInput.py b/src/backend/langflow/components/io/ChatInput.py
index ead455d42..d5204d19f 100644
--- a/src/backend/langflow/components/io/ChatInput.py
+++ b/src/backend/langflow/components/io/ChatInput.py
@@ -32,6 +32,11 @@ class ChatInput(CustomComponent):
     ) -> Union[Text, Record]:
         self.status = message
         if as_record:
+            if isinstance(message, Record):
+                # Update the data of the record
+                message.data["sender"] = sender_type
+                message.data["sender_name"] = sender_name
+                return message
             return Record(
                 text=message, data={"sender": sender_type, "sender_name": sender_name}
             )
diff --git a/src/backend/langflow/components/io/ChatOutput.py b/src/backend/langflow/components/io/ChatOutput.py
index 0c5c85664..699138f9f 100644
--- a/src/backend/langflow/components/io/ChatOutput.py
+++ b/src/backend/langflow/components/io/ChatOutput.py
@@ -40,6 +40,13 @@ class ChatOutput(CustomComponent):
     ) -> Text:
         self.status = message
         if as_record:
+            if isinstance(message, Record):
+                # Update the data of the record
+                message.data["sender"] = sender_type
+                message.data["sender_name"] = sender_name
+                message.data["session_id"] = session_id
+
+                return message
             return Record(
                 text=message,
                 data={