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

Update sender options in MessageHistoryComponent

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-28 13:40:28 -03:00
commit fdba9ebd99
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/langflow/components/io/MessageHistory.py
View file

@ -12,7 +12,7 @@ class MessageHistoryComponent(CustomComponent):
def build_config(self): def build_config(self):
return { return {
"sender": { "sender": {
"options": ["Machine", "User"], "options": ["Machine", "User", "Machine and User"],
"display_name": "Sender Type", "display_name": "Sender Type",
}, },
"sender_name": {"display_name": "Sender Name"}, "sender_name": {"display_name": "Sender Name"},
@ -38,6 +38,8 @@ class MessageHistoryComponent(CustomComponent):
session_id: Optional[str] = None, session_id: Optional[str] = None,
n_messages: int = 5, n_messages: int = 5,
) -> List[Record]: ) -> List[Record]:
if sender == "Machine and User":
sender = None
messages = get_messages( messages = get_messages(
sender=sender, sender=sender,
sender_name=sender_name, sender_name=sender_name,