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

Update ChatInput and ChatOutput to include sender information in Record data

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-19 12:30:14 -03:00
commit 9481f76b05
2 changed files with 12 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -32,6 +32,11 @@ class ChatInput(CustomComponent):
) -> Union[Text, Record]:
self.status = message
if as_record:
if isinstance(message, Record):
# Update the data of the record
message.data["sender"] = sender_type
message.data["sender_name"] = sender_name
return message
return Record(
text=message, data={"sender": sender_type, "sender_name": sender_name}
)

7
src/backend/langflow/components/io/ChatOutput.py
View file

@ -40,6 +40,13 @@ class ChatOutput(CustomComponent):
) -> Text:
self.status = message
if as_record:
if isinstance(message, Record):
# Update the data of the record
message.data["sender"] = sender_type
message.data["sender_name"] = sender_name
message.data["session_id"] = session_id
return message
return Record(
text=message,
data={