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 <cristhian.lousa@gmail.com>
This commit is contained in:
Rodrigo Nader 2024-12-06 15:16:04 -03:00 committed by GitHub
commit dee1439c78
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 6 additions and 3 deletions

View file

@ -58,6 +58,7 @@ class MemoryComponent(Component):
value="Ascending",
info="Order of the messages.",
advanced=True,
tool_mode=True,
),
MultilineInput(
name="template",

View file

@ -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