From c9e879f494903720aad5b30beddfad7127e1d2c1 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 20 Feb 2024 22:26:39 -0300 Subject: [PATCH] Update sender_type to sender in ChatInput class --- src/backend/langflow/components/io/ChatInput.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/components/io/ChatInput.py b/src/backend/langflow/components/io/ChatInput.py index 461a0d353..b91001eff 100644 --- a/src/backend/langflow/components/io/ChatInput.py +++ b/src/backend/langflow/components/io/ChatInput.py @@ -16,7 +16,7 @@ class ChatInput(CustomComponent): "display_name": "Message", "multiline": True, }, - "sender_type": { + "sender": { "options": ["Machine", "User"], "display_name": "Sender Type", }, @@ -29,7 +29,7 @@ class ChatInput(CustomComponent): def build( self, - sender_type: Optional[str] = "User", + sender: Optional[str] = "User", sender_name: Optional[str] = "You", message: Optional[str] = None, as_record: Optional[bool] = False, @@ -38,11 +38,11 @@ class ChatInput(CustomComponent): if as_record: if isinstance(message, Record): # Update the data of the record - message.data["sender"] = sender_type + message.data["sender"] = sender message.data["sender_name"] = sender_name return message return Record( - text=message, data={"sender": sender_type, "sender_name": sender_name} + text=message, data={"sender": sender, "sender_name": sender_name} ) if not message: message = ""