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

Update sender_type to sender in ChatOutput class

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-20 22:26:15 -03:00
commit e235bc4151
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/ChatOutput.py
View file

@ -18,7 +18,7 @@ class ChatOutput(CustomComponent):
def build_config(self): def build_config(self):
return { return {
"message": {"input_types": ["Text"], "display_name": "Message"}, "message": {"input_types": ["Text"], "display_name": "Message"},
"sender_type": { "sender": {
"options": ["Machine", "User"], "options": ["Machine", "User"],
"display_name": "Sender Type", "display_name": "Sender Type",
}, },
@ -36,7 +36,7 @@ class ChatOutput(CustomComponent):
def build( def build(
self, self,
sender_type: Optional[str] = "Machine", sender: Optional[str] = "Machine",
sender_name: Optional[str] = "AI", sender_name: Optional[str] = "AI",
session_id: Optional[str] = None, session_id: Optional[str] = None,
message: Optional[str] = None, message: Optional[str] = None,
@ -46,7 +46,7 @@ class ChatOutput(CustomComponent):
if as_record: if as_record:
if isinstance(message, Record): if isinstance(message, Record):
# Update the data of the record # Update the data of the record
message.data["sender"] = sender_type message.data["sender"] = sender
message.data["sender_name"] = sender_name message.data["sender_name"] = sender_name
message.data["session_id"] = session_id message.data["session_id"] = session_id
@ -54,7 +54,7 @@ class ChatOutput(CustomComponent):
return Record( return Record(
text=message, text=message,
data={ data={
"sender": sender_type, "sender": sender,
"sender_name": sender_name, "sender_name": sender_name,
"session_id": session_id, "session_id": session_id,
}, },