From a892a5e126e3b08d3731f97ea589822a1a3c1ba9 Mon Sep 17 00:00:00 2001 From: rodrigosnader Date: Wed, 26 Jun 2024 14:13:47 +0000 Subject: [PATCH] Apply Ruff formatting --- .../langflow/components/helpers/StoreMessage.py | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) diff --git a/src/backend/base/langflow/components/helpers/StoreMessage.py b/src/backend/base/langflow/components/helpers/StoreMessage.py index 6e37573a8..4864bf4a7 100644 --- a/src/backend/base/langflow/components/helpers/StoreMessage.py +++ b/src/backend/base/langflow/components/helpers/StoreMessage.py @@ -4,6 +4,7 @@ from langflow.schema.message import Message from langflow.template import Output from langflow.memory import get_messages, store_message + class StoreMessageComponent(Component): display_name = "Store Message" description = "Stores a chat message or text." @@ -15,7 +16,7 @@ class StoreMessageComponent(Component): display_name="Message", info="The chat message to be stored.", input_types=["Message", "str"], - required=True + required=True, ), StrInput( name="sender", @@ -25,18 +26,13 @@ class StoreMessageComponent(Component): advanced=True, ), StrInput( - name="sender_name", - display_name="Sender Name", - info="The name of the sender.", - value="", - advanced=True + name="sender_name", display_name="Sender Name", info="The name of the sender.", value="", advanced=True ), StrInput( name="session_id", display_name="Session ID", info="The session ID of the chat.", value="", - ), ] @@ -48,7 +44,9 @@ class StoreMessageComponent(Component): if isinstance(self.message, str): if not self.session_id: raise ValueError("If passing a text, Session ID cannot be empty.") - message = Message(text=self.message, sender=self.sender, sender_name=self.sender_name, session_id=self.session_id) + message = Message( + text=self.message, sender=self.sender, sender_name=self.sender_name, session_id=self.session_id + ) elif isinstance(self.message, Message): message = self.message