From e235bc415118e430e1f5dfa22af2fc8378d70187 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 20 Feb 2024 22:26:15 -0300 Subject: [PATCH] Update sender_type to sender in ChatOutput class --- src/backend/langflow/components/io/ChatOutput.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/components/io/ChatOutput.py b/src/backend/langflow/components/io/ChatOutput.py index db6650932..5896adb6b 100644 --- a/src/backend/langflow/components/io/ChatOutput.py +++ b/src/backend/langflow/components/io/ChatOutput.py @@ -18,7 +18,7 @@ class ChatOutput(CustomComponent): def build_config(self): return { "message": {"input_types": ["Text"], "display_name": "Message"}, - "sender_type": { + "sender": { "options": ["Machine", "User"], "display_name": "Sender Type", }, @@ -36,7 +36,7 @@ class ChatOutput(CustomComponent): def build( self, - sender_type: Optional[str] = "Machine", + sender: Optional[str] = "Machine", sender_name: Optional[str] = "AI", session_id: Optional[str] = None, message: Optional[str] = None, @@ -46,7 +46,7 @@ class ChatOutput(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 message.data["session_id"] = session_id @@ -54,7 +54,7 @@ class ChatOutput(CustomComponent): return Record( text=message, data={ - "sender": sender_type, + "sender": sender, "sender_name": sender_name, "session_id": session_id, },