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