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:
parent
121b7c6e39
commit
dee1439c78
2 changed files with 6 additions and 3 deletions
|
|
@ -58,6 +58,7 @@ class MemoryComponent(Component):
|
|||
value="Ascending",
|
||||
info="Order of the messages.",
|
||||
advanced=True,
|
||||
tool_mode=True,
|
||||
),
|
||||
MultilineInput(
|
||||
name="template",
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue