From 1ee717ed4ead913e520f09b33907ff3305dcc510 Mon Sep 17 00:00:00 2001
From: Rodrigo <rodrigosilvanader@gmail.com>
Date: Wed, 12 Jun 2024 17:50:17 -0300
Subject: [PATCH] refactor: Update ChatOutput inputs to use specific input
 classes

---
 src/backend/base/langflow/components/outputs/ChatOutput.py | 2 +-
 src/frontend/src/utils/styleUtils.ts                       | 7 ++++---
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/backend/base/langflow/components/outputs/ChatOutput.py b/src/backend/base/langflow/components/outputs/ChatOutput.py
index 5b9683529..3b7c74927 100644
--- a/src/backend/base/langflow/components/outputs/ChatOutput.py
+++ b/src/backend/base/langflow/components/outputs/ChatOutput.py
@@ -26,7 +26,7 @@ class ChatOutput(ChatComponent):
             advanced=True,
             info="Type of sender.",
         ),
-        StrInput(name="sender_name", display_name="Sender Name", info="Name of the sender.", value="AI"),
+        StrInput(name="sender_name", display_name="Sender Name", info="Name of the sender.", value="AI", advanced=True),
         StrInput(name="session_id", display_name="Session ID", info="Session ID for the message.", advanced=True),
         BoolInput(
             name="record_template",
diff --git a/src/frontend/src/utils/styleUtils.ts b/src/frontend/src/utils/styleUtils.ts
index 43e92bfe2..518c23656 100644
--- a/src/frontend/src/utils/styleUtils.ts
+++ b/src/frontend/src/utils/styleUtils.ts
@@ -242,7 +242,7 @@ export const gradients = [
 export const nodeColors: { [char: string]: string } = {
   inputs: "#10B981",
   outputs: "#AA2411",
-  data: "#4367BF",
+  data: "#198BF6",
   prompts: "#4367BF",
   models: "#6344BE",
   model_specs: "#6344BE",
@@ -267,13 +267,14 @@ export const nodeColors: { [char: string]: string } = {
   experimental: "#E6277A",
   langchain_utilities: "#31A3CC",
   output_parsers: "#E6A627",
-  str: "#31a3cc",
-  Text: "#31a3cc",
+  str: "#4367BF",
+  Text: "#4367BF",
   retrievers: "#e6b25a",
   unknown: "#9CA3AF",
   custom_components: "#ab11ab",
   Records: "#31a3cc",
   Record: "#31a3cc",
+  Message: "#4367BF",
 };
 
 export const nodeNames: { [char: string]: string } = {