From fdba9ebd9956b991e644fce45daba8951b7e966c Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Wed, 28 Feb 2024 13:40:28 -0300
Subject: [PATCH] Update sender options in MessageHistoryComponent

---
 src/backend/langflow/components/io/MessageHistory.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/backend/langflow/components/io/MessageHistory.py b/src/backend/langflow/components/io/MessageHistory.py
index 345de4ba0..e9c3f1a87 100644
--- a/src/backend/langflow/components/io/MessageHistory.py
+++ b/src/backend/langflow/components/io/MessageHistory.py
@@ -12,7 +12,7 @@ class MessageHistoryComponent(CustomComponent):
     def build_config(self):
         return {
             "sender": {
-                "options": ["Machine", "User"],
+                "options": ["Machine", "User", "Machine and User"],
                 "display_name": "Sender Type",
             },
             "sender_name": {"display_name": "Sender Name"},
@@ -38,6 +38,8 @@ class MessageHistoryComponent(CustomComponent):
         session_id: Optional[str] = None,
         n_messages: int = 5,
     ) -> List[Record]:
+        if sender == "Machine and User":
+            sender = None
         messages = get_messages(
             sender=sender,
             sender_name=sender_name,