Refactor ChatComponent's add_message method

src/backend/langflow/io/schema.py

@@ -48,27 +48,28 @@ class ChatComponent(CustomComponent):
             raise ValueError(
                 "All of session_id, sender, and sender_name must be provided."
             )
-
+        if isinstance(message, Record):
-        if not record:
+            record = message
-            record = []
+            record.data.update(
-            if not session_id or not sender or not sender_name:
+                {
-                raise ValueError
+                    "session_id": session_id,
-            for text in text:
+                    "sender": sender,
-                record = Record(
+                    "sender_name": sender_name,
-                    text=text,
+                }
-                    data={
+            )
-                        "session_id": session_id,
+        else:
-                        "sender": sender,
+            record = Record(
-                        "sender_name": sender_name,
+                text=message,
-                    },
+                data={
-                )
+                    "session_id": session_id,
-                record.append(record)
+                    "sender": sender,
-        elif isinstance(record, Record):
+                    "sender_name": sender_name,
-            record = [record]
+                },
+            )
 
         self.status = record
-        record = add_messages(record)
+        records = add_messages([record])
-        return record
+        return records[0]
 
     def build(
         self,