jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

feat: add tool mode to components that store and retrieve message history (#5110)

Browse source 
* 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
Download patch file Download diff file Expand all files Collapse all files

1
src/backend/base/langflow/components/helpers/memory.py
View file

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

8
src/backend/base/langflow/components/helpers/store_message.py
View file

@ -1,5 +1,5 @@
from langflow.custom import Component from langflow.custom import Component
from langflow.inputs import HandleInput, MessageInput from langflow.inputs import HandleInput
from langflow.inputs.inputs import MessageTextInput from langflow.inputs.inputs import MessageTextInput
from langflow.memory import aget_messages, astore_message from langflow.memory import aget_messages, astore_message
from langflow.schema.message import Message from langflow.schema.message import Message
@ -14,7 +14,9 @@ class StoreMessageComponent(Component):
name = "StoreMessage" name = "StoreMessage"
inputs = [ 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( HandleInput(
name="memory", name="memory",
display_name="External Memory", display_name="External Memory",
@ -48,7 +50,7 @@ class StoreMessageComponent(Component):
] ]
async def store_message(self) -> Message: 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.session_id = self.session_id or message.session_id
message.sender = self.sender or message.sender or MESSAGE_SENDER_AI message.sender = self.sender or message.sender or MESSAGE_SENDER_AI