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