Refactor ChatInput and ChatOutput to support returning messages as records

src/backend/langflow/components/io/ChatInput.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
 
 
@@ -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

src/backend/langflow/components/io/ChatOutput.py

@@ -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