From dee1439c781b55b6ebfad3c5c011fc90f7ffaed3 Mon Sep 17 00:00:00 2001 From: Rodrigo Nader Date: Fri, 6 Dec 2024 15:16:04 -0300 Subject: [PATCH] feat: add tool mode to components that store and retrieve message history (#5110) * add tool mode to components that store and retrieve message history * [autofix.ci] apply automated fixes * improve syntax --------- Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: Cristhian Zanforlin Lousa --- src/backend/base/langflow/components/helpers/memory.py | 1 + .../base/langflow/components/helpers/store_message.py | 8 +++++--- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/src/backend/base/langflow/components/helpers/memory.py b/src/backend/base/langflow/components/helpers/memory.py index 7dfebafd2..57b0f87f9 100644 --- a/src/backend/base/langflow/components/helpers/memory.py +++ b/src/backend/base/langflow/components/helpers/memory.py @@ -58,6 +58,7 @@ class MemoryComponent(Component): value="Ascending", info="Order of the messages.", advanced=True, + tool_mode=True, ), MultilineInput( name="template", diff --git a/src/backend/base/langflow/components/helpers/store_message.py b/src/backend/base/langflow/components/helpers/store_message.py index f9178e17e..20ef31c0b 100644 --- a/src/backend/base/langflow/components/helpers/store_message.py +++ b/src/backend/base/langflow/components/helpers/store_message.py @@ -1,5 +1,5 @@ from langflow.custom import Component -from langflow.inputs import HandleInput, MessageInput +from langflow.inputs import HandleInput from langflow.inputs.inputs import MessageTextInput from langflow.memory import aget_messages, astore_message from langflow.schema.message import Message @@ -14,7 +14,9 @@ class StoreMessageComponent(Component): name = "StoreMessage" inputs = [ - MessageInput(name="message", display_name="Message", info="The chat message to be stored.", required=True), + MessageTextInput( + name="message", display_name="Message", info="The chat message to be stored.", required=True, tool_mode=True + ), HandleInput( name="memory", display_name="External Memory", @@ -48,7 +50,7 @@ class StoreMessageComponent(Component): ] async def store_message(self) -> Message: - message = self.message + message = Message(text=self.message) if isinstance(self.message, str) else self.message message.session_id = self.session_id or message.session_id message.sender = self.sender or message.sender or MESSAGE_SENDER_AI