From 8511c2ebad87aa6905924d1a155f10cdcf13430e Mon Sep 17 00:00:00 2001
From: ogabrielluiz <gabriel@langflow.org>
Date: Wed, 12 Jun 2024 16:14:48 -0300
Subject: [PATCH] refactor: Update ChatOutput inputs to use specific input
 classes

Update the inputs of the ChatOutput class in ChatOutput.py to use specific input classes such as StrInput, DropdownInput, and BoolInput. This change improves the organization and separation of concerns in the codebase, making it easier to understand and maintain.
---
 .../langflow/components/outputs/ChatOutput.py | 25 +++++++++++--------
 1 file changed, 14 insertions(+), 11 deletions(-)

diff --git a/src/backend/base/langflow/components/outputs/ChatOutput.py b/src/backend/base/langflow/components/outputs/ChatOutput.py
index 4bef18e8e..5b9683529 100644
--- a/src/backend/base/langflow/components/outputs/ChatOutput.py
+++ b/src/backend/base/langflow/components/outputs/ChatOutput.py
@@ -1,7 +1,8 @@
 from langflow.base.io.chat import ChatComponent
 from langflow.field_typing import Text
+from langflow.inputs import BoolInput, DropdownInput, StrInput
 from langflow.schema.message import Message
-from langflow.template import Input, Output
+from langflow.template import Output
 
 
 class ChatOutput(ChatComponent):
@@ -10,25 +11,25 @@ class ChatOutput(ChatComponent):
     icon = "ChatOutput"
 
     inputs = [
-        Input(
-            name="input_value", type=str, display_name="Text", multiline=True, info="Message to be passed as output."
+        StrInput(
+            name="input_value",
+            display_name="Text",
+            multiline=True,
+            info="Message to be passed as output.",
+            input_types=["Text", "Message"],
         ),
-        Input(
+        DropdownInput(
             name="sender",
-            type=str,
             display_name="Sender Type",
             options=["Machine", "User"],
             value="Machine",
             advanced=True,
             info="Type of sender.",
         ),
-        Input(name="sender_name", type=str, display_name="Sender Name", info="Name of the sender.", value="AI"),
-        Input(
-            name="session_id", type=str, display_name="Session ID", info="Session ID for the message.", advanced=True
-        ),
-        Input(
+        StrInput(name="sender_name", display_name="Sender Name", info="Name of the sender.", value="AI"),
+        StrInput(name="session_id", display_name="Session ID", info="Session ID for the message.", advanced=True),
+        BoolInput(
             name="record_template",
-            type=str,
             display_name="Record Template",
             value="{text}",
             advanced=True,
@@ -56,5 +57,7 @@ class ChatOutput(ChatComponent):
         )
         if self.session_id and isinstance(message, Message) and isinstance(message.text, str):
             self.store_message(message)
+            self.message.value = message
+
         self.status = message
         return message