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

Apply Ruff formatting

Browse source
This commit is contained in:
rodrigosnader 2024-06-26 14:13:47 +00:00 • committed by Gabriel Luiz Freitas Almeida
commit a892a5e126
1 changed files with 6 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/backend/base/langflow/components/helpers/StoreMessage.py
View file

@ -4,6 +4,7 @@ from langflow.schema.message import Message
from langflow.template import Output
from langflow.memory import get_messages, store_message
class StoreMessageComponent(Component):
display_name = "Store Message"
description = "Stores a chat message or text."
@ -15,7 +16,7 @@ class StoreMessageComponent(Component):
display_name="Message",
info="The chat message to be stored.",
input_types=["Message", "str"],
required=True
required=True,
),
StrInput(
name="sender",
@ -25,18 +26,13 @@ class StoreMessageComponent(Component):
advanced=True,
),
StrInput(
name="sender_name",
display_name="Sender Name",
info="The name of the sender.",
value="",
advanced=True
name="sender_name", display_name="Sender Name", info="The name of the sender.", value="", advanced=True
),
StrInput(
name="session_id",
display_name="Session ID",
info="The session ID of the chat.",
value="",
),
]
@ -48,7 +44,9 @@ class StoreMessageComponent(Component):
if isinstance(self.message, str):
if not self.session_id:
raise ValueError("If passing a text, Session ID cannot be empty.")
message = Message(text=self.message, sender=self.sender, sender_name=self.sender_name, session_id=self.session_id)
message = Message(
text=self.message, sender=self.sender, sender_name=self.sender_name, session_id=self.session_id
)
elif isinstance(self.message, Message):
message = self.message