From e0e35b3245de8f4b32eb90263b3a6291071ec201 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Wed, 21 Feb 2024 16:00:31 -0300
Subject: [PATCH] Add session_id parameter to ChatInput class

---
 src/backend/langflow/components/io/ChatInput.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/backend/langflow/components/io/ChatInput.py b/src/backend/langflow/components/io/ChatInput.py
index b91001eff..3e03642b3 100644
--- a/src/backend/langflow/components/io/ChatInput.py
+++ b/src/backend/langflow/components/io/ChatInput.py
@@ -21,6 +21,10 @@ class ChatInput(CustomComponent):
                 "display_name": "Sender Type",
             },
             "sender_name": {"display_name": "Sender Name"},
+            "session_id": {
+                "display_name": "Session ID",
+                "info": "Session ID of the chat history.",
+            },
             "as_record": {
                 "display_name": "As Record",
                 "info": "If true, the message will be returned as a Record.",
@@ -33,6 +37,7 @@ class ChatInput(CustomComponent):
         sender_name: Optional[str] = "You",
         message: Optional[str] = None,
         as_record: Optional[bool] = False,
+        session_id: Optional[str] = None,
     ) -> Union[Text, Record]:
         self.status = message
         if as_record:
@@ -40,9 +45,15 @@ class ChatInput(CustomComponent):
                 # Update the data of the record
                 message.data["sender"] = sender
                 message.data["sender_name"] = sender_name
+                message.data["session_id"] = session_id
                 return message
             return Record(
-                text=message, data={"sender": sender, "sender_name": sender_name}
+                text=message,
+                data={
+                    "sender": sender,
+                    "sender_name": sender_name,
+                    "session_id": session_id,
+                },
             )
         if not message:
             message = ""