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

Add session_id parameter to ChatInput class

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-21 16:00:31 -03:00
commit e0e35b3245
1 changed files with 12 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/backend/langflow/components/io/ChatInput.py
View file

@ -21,6 +21,10 @@ class ChatInput(CustomComponent):
"display_name": "Sender Type", "display_name": "Sender Type",
}, },
"sender_name": {"display_name": "Sender Name"}, "sender_name": {"display_name": "Sender Name"},
"session_id": {
"display_name": "Session ID",
"info": "Session ID of the chat history.",
},
"as_record": { "as_record": {
"display_name": "As Record", "display_name": "As Record",
"info": "If true, the message will be returned as a Record.", "info": "If true, the message will be returned as a Record.",
@ -33,6 +37,7 @@ class ChatInput(CustomComponent):
sender_name: Optional[str] = "You", sender_name: Optional[str] = "You",
message: Optional[str] = None, message: Optional[str] = None,
as_record: Optional[bool] = False, as_record: Optional[bool] = False,
session_id: Optional[str] = None,
) -> Union[Text, Record]: ) -> Union[Text, Record]:
self.status = message self.status = message
if as_record: if as_record:
@ -40,9 +45,15 @@ class ChatInput(CustomComponent):
# Update the data of the record # Update the data of the record
message.data["sender"] = sender message.data["sender"] = sender
message.data["sender_name"] = sender_name message.data["sender_name"] = sender_name
message.data["session_id"] = session_id
return message return message
return Record( 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: if not message:
message = "" message = ""