diff --git a/src/backend/langflow/components/io/ChatInput.py b/src/backend/langflow/components/io/ChatInput.py index 42f2715e6..704ab732b 100644 --- a/src/backend/langflow/components/io/ChatInput.py +++ b/src/backend/langflow/components/io/ChatInput.py @@ -25,10 +25,6 @@ class ChatInput(CustomComponent): "display_name": "Session ID", "info": "Session ID of the chat history.", }, - "as_record": { - "display_name": "As Record", - "info": "If true, the message will be returned as a Record.", - }, } def build( @@ -36,18 +32,15 @@ class ChatInput(CustomComponent): sender: Optional[str] = "User", sender_name: Optional[str] = "User", message: Optional[str] = None, - as_record: Optional[bool] = False, session_id: Optional[str] = None, - ) -> Union[Text, Record]: - self.status = message - if as_record: - if isinstance(message, Record): - # Update the data of the record - message.data["sender"] = sender - message.data["sender_name"] = sender_name - message.data["session_id"] = session_id - return message - return Record( + ) -> Record: + if isinstance(message, Record): + # Update the data of the record + message.data["sender"] = sender + message.data["sender_name"] = sender_name + message.data["session_id"] = session_id + else: + message = Record( text=message, data={ "sender": sender, diff --git a/src/backend/langflow/components/io/ChatOutput.py b/src/backend/langflow/components/io/ChatOutput.py index 5896adb6b..6812c6d34 100644 --- a/src/backend/langflow/components/io/ChatOutput.py +++ b/src/backend/langflow/components/io/ChatOutput.py @@ -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 @@ -28,10 +27,6 @@ class ChatOutput(CustomComponent): "info": "Session ID of the chat history.", "input_types": ["Text"], }, - "as_record": { - "display_name": "As Record", - "info": "If true, the message will be returned as a Record.", - }, } def build( @@ -40,18 +35,14 @@ class ChatOutput(CustomComponent): sender_name: Optional[str] = "AI", session_id: Optional[str] = None, message: Optional[str] = None, - as_record: Optional[bool] = False, - ) -> Union[Text, Record]: - self.status = message - if as_record: - if isinstance(message, Record): - # Update the data of the record - message.data["sender"] = sender - message.data["sender_name"] = sender_name - message.data["session_id"] = session_id - - return message - return Record( + ) -> Record: + if isinstance(message, Record): + # Update the data of the record + message.data["sender"] = sender + message.data["sender_name"] = sender_name + message.data["session_id"] = session_id + else: + message = Record( text=message, data={ "sender": sender,