Update ChatInput and ChatVertex classes

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 = ""

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)