From 9481f76b05b8b4818c75b342a00bb957a50f0e89 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 19 Feb 2024 12:30:14 -0300 Subject: [PATCH] Update ChatInput and ChatOutput to include sender information in Record data --- src/backend/langflow/components/io/ChatInput.py | 5 +++++ src/backend/langflow/components/io/ChatOutput.py | 7 +++++++ 2 files changed, 12 insertions(+) diff --git a/src/backend/langflow/components/io/ChatInput.py b/src/backend/langflow/components/io/ChatInput.py index ead455d42..d5204d19f 100644 --- a/src/backend/langflow/components/io/ChatInput.py +++ b/src/backend/langflow/components/io/ChatInput.py @@ -32,6 +32,11 @@ class ChatInput(CustomComponent): ) -> Union[Text, Record]: self.status = message if as_record: + if isinstance(message, Record): + # Update the data of the record + message.data["sender"] = sender_type + message.data["sender_name"] = sender_name + return message return Record( text=message, data={"sender": sender_type, "sender_name": sender_name} ) diff --git a/src/backend/langflow/components/io/ChatOutput.py b/src/backend/langflow/components/io/ChatOutput.py index 0c5c85664..699138f9f 100644 --- a/src/backend/langflow/components/io/ChatOutput.py +++ b/src/backend/langflow/components/io/ChatOutput.py @@ -40,6 +40,13 @@ class ChatOutput(CustomComponent): ) -> Text: self.status = message if as_record: + if isinstance(message, Record): + # Update the data of the record + message.data["sender"] = sender_type + message.data["sender_name"] = sender_name + message.data["session_id"] = session_id + + return message return Record( text=message, data={