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

Update sender_type to sender in ChatInput class

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-20 22:26:39 -03:00
commit c9e879f494
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -16,7 +16,7 @@ class ChatInput(CustomComponent):
"display_name": "Message",
"multiline": True,
},
"sender_type": {
"sender": {
"options": ["Machine", "User"],
"display_name": "Sender Type",
},
@ -29,7 +29,7 @@ class ChatInput(CustomComponent):
def build(
self,
sender_type: Optional[str] = "User",
sender: Optional[str] = "User",
sender_name: Optional[str] = "You",
message: Optional[str] = None,
as_record: Optional[bool] = False,
@ -38,11 +38,11 @@ class ChatInput(CustomComponent):
if as_record:
if isinstance(message, Record):
# Update the data of the record
message.data["sender"] = sender_type
message.data["sender"] = sender
message.data["sender_name"] = sender_name
return message
return Record(
text=message, data={"sender": sender_type, "sender_name": sender_name}
text=message, data={"sender": sender, "sender_name": sender_name}
)
if not message:
message = ""