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

Refactor ChatInput and ChatOutput to support returning messages as records

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-26 18:53:10 -03:00
commit 84d3bd1ba6
2 changed files with 44 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,7 +1,6 @@
from typing import Optional, Union
from typing import Optional
from langflow import CustomComponent
from langflow.field_typing import Text
from langflow.schema import Record
@ -25,6 +24,10 @@ class ChatInput(CustomComponent):
"display_name": "Session ID",
"info": "Session ID of the chat history.",
},
"return_record": {
"display_name": "Return Record",
"info": "Return the message as a record containing the sender, sender_name, and session_id.",
},
}
def build(
@ -33,7 +36,9 @@ class ChatInput(CustomComponent):
sender_name: Optional[str] = "User",
message: Optional[str] = None,
session_id: Optional[str] = None,
return_record: Optional[bool] = False,
) -> Record:
if return_record:
if isinstance(message, Record):
# Update the data of the record
message.data["sender"] = sender

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

@ -1,6 +1,7 @@
from typing import Optional
from typing import Optional, Union
from langflow import CustomComponent
from langflow.field_typing import Text
from langflow.schema import Record
@ -27,6 +28,10 @@ class ChatOutput(CustomComponent):
"info": "Session ID of the chat history.",
"input_types": ["Text"],
},
"return_record": {
"display_name": "Return Record",
"info": "Return the message as a record containing the sender, sender_name, and session_id.",
},
}
def build(
@ -35,7 +40,9 @@ class ChatOutput(CustomComponent):
sender_name: Optional[str] = "AI",
session_id: Optional[str] = None,
message: Optional[str] = None,
) -> Record:
return_record: Optional[bool] = False,
) -> Union[Text, Record]:
if return_record:
if isinstance(message, Record):
# Update the data of the record
message.data["sender"] = sender