diff --git a/src/backend/base/langflow/components/helpers/__init__.py b/src/backend/base/langflow/components/helpers/__init__.py index 7300242624c3..176282a9223d 100644 --- a/src/backend/base/langflow/components/helpers/__init__.py +++ b/src/backend/base/langflow/components/helpers/__init__.py @@ -3,7 +3,7 @@ from .id_generator import IDGeneratorComponent from .memory import MemoryComponent from .output_parser import OutputParserComponent -from .store_message import StoreMessageComponent +from .store_message import MessageStoreComponent from .structured_output import StructuredOutputComponent __all__ = [ @@ -11,7 +11,7 @@ "CurrentDateComponent", "IDGeneratorComponent", "MemoryComponent", + "MessageStoreComponent", "OutputParserComponent", - "StoreMessageComponent", "StructuredOutputComponent", ] diff --git a/src/backend/base/langflow/components/helpers/store_message.py b/src/backend/base/langflow/components/helpers/store_message.py index 544d2a09f2c8..ae779b399edd 100644 --- a/src/backend/base/langflow/components/helpers/store_message.py +++ b/src/backend/base/langflow/components/helpers/store_message.py @@ -7,10 +7,10 @@ from langflow.utils.constants import MESSAGE_SENDER_AI, MESSAGE_SENDER_NAME_AI -class StoreMessageComponent(Component): - display_name = "Store Message" +class MessageStoreComponent(Component): + display_name = "Message Store" description = "Stores a chat message or text into Langflow tables or an external memory." - icon = "save" + icon = "message-square-text" name = "StoreMessage" inputs = [ @@ -46,7 +46,7 @@ class StoreMessageComponent(Component): ] outputs = [ - Output(display_name="Stored Message", name="stored_message", method="store_message"), + Output(display_name="Stored Messages", name="stored_messages", method="store_message", hidden=True), ] async def store_message(self) -> Message: