From 21bff66c1aa295fe94d53f63d4cd0c7c30dcb2e6 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Thu, 8 Feb 2024 18:32:08 -0300
Subject: [PATCH] Update ChatInput and ChatVertex classes

---
 src/backend/langflow/components/io/ChatInput.py | 16 +++++++++++-----
 src/backend/langflow/graph/vertex/types.py      |  3 ++-
 2 files changed, 13 insertions(+), 6 deletions(-)

diff --git a/src/backend/langflow/components/io/ChatInput.py b/src/backend/langflow/components/io/ChatInput.py
index 42adc0957..c6842c0b2 100644
--- a/src/backend/langflow/components/io/ChatInput.py
+++ b/src/backend/langflow/components/io/ChatInput.py
@@ -8,13 +8,19 @@ class ChatInput(CustomComponent):
     display_name = "Chat Input"
     description = "Used to get user input from the chat."
 
-    field_config = {
-        "code": {
-            "show": False,
+    def build_config(self):
+        return {
+            "message": {"input_types": ["Text"]},
+            "sender": {"options": ["Machine", "User"], "display_name": "Sender Type"},
+            "sender_name": {"display_name": "Sender Name"},
         }
-    }
 
-    def build(self, message: Optional[str] = None) -> Text:
+    def build(
+        self,
+        sender: Optional[str] = "User",
+        sender_name: Optional[str] = "You",
+        message: Optional[str] = None,
+    ) -> Text:
         self.repr_value = message
         if not message:
             message = ""
diff --git a/src/backend/langflow/graph/vertex/types.py b/src/backend/langflow/graph/vertex/types.py
index ffe35cc89..3be211746 100644
--- a/src/backend/langflow/graph/vertex/types.py
+++ b/src/backend/langflow/graph/vertex/types.py
@@ -2,6 +2,7 @@ import ast
 from typing import Callable, Dict, List, Optional, Union
 
 from langchain_core.messages import AIMessage
+
 from langflow.graph.utils import UnbuiltObject, flatten_list
 from langflow.graph.vertex.base import StatefulVertex, StatelessVertex
 from langflow.interface.utils import extract_input_variables_from_prompt
@@ -321,7 +322,7 @@ class ChatVertex(StatelessVertex):
 
     async def _run(self, *args, **kwargs):
         if self.is_interface_component:
-            if self.vertex_type == "ChatOutput":
+            if self.vertex_type in ["ChatOutput", "ChatInput"]:
                 artifacts = None
                 sender = self.params.get("sender", None)
                 sender_name = self.params.get("sender_name", None)