From ba79225b9c6a8981a19f1c6ed7bde72b000ddf45 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Fri, 9 Feb 2024 08:48:07 -0300
Subject: [PATCH] Add display_name for message input in ChatInput and
 ChatOutput

---
 src/backend/langflow/components/io/ChatInput.py  | 2 +-
 src/backend/langflow/components/io/ChatOutput.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/backend/langflow/components/io/ChatInput.py b/src/backend/langflow/components/io/ChatInput.py
index c6842c0b2..32cd1bf7f 100644
--- a/src/backend/langflow/components/io/ChatInput.py
+++ b/src/backend/langflow/components/io/ChatInput.py
@@ -10,7 +10,7 @@ class ChatInput(CustomComponent):
 
     def build_config(self):
         return {
-            "message": {"input_types": ["Text"]},
+            "message": {"input_types": ["Text"], "display_name": "Message"},
             "sender": {"options": ["Machine", "User"], "display_name": "Sender Type"},
             "sender_name": {"display_name": "Sender Name"},
         }
diff --git a/src/backend/langflow/components/io/ChatOutput.py b/src/backend/langflow/components/io/ChatOutput.py
index b37e754b4..7e24198c1 100644
--- a/src/backend/langflow/components/io/ChatOutput.py
+++ b/src/backend/langflow/components/io/ChatOutput.py
@@ -16,7 +16,7 @@ class ChatOutput(CustomComponent):
 
     def build_config(self):
         return {
-            "message": {"input_types": ["Text"]},
+            "message": {"input_types": ["Text"], "display_name": "Message"},
             "sender": {"options": ["Machine", "User"], "display_name": "Sender Type"},
             "sender_name": {"display_name": "Sender Name"},
             "session_id": {