From 2033fd65efcedc6eb4a201d269cce192b4809eb9 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Wed, 21 Feb 2024 16:00:46 -0300
Subject: [PATCH] Add StoreMessages component to store messages

---
 .../langflow/components/io/StoreMessages.py   | 26 +++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 src/backend/langflow/components/io/StoreMessages.py

diff --git a/src/backend/langflow/components/io/StoreMessages.py b/src/backend/langflow/components/io/StoreMessages.py
new file mode 100644
index 000000000..bc85c1028
--- /dev/null
+++ b/src/backend/langflow/components/io/StoreMessages.py
@@ -0,0 +1,26 @@
+from typing import List
+
+from langflow import CustomComponent
+from langflow.memory import add_messages
+from langflow.schema import Record
+
+
+class StoreMessages(CustomComponent):
+    display_name = "Store Messages"
+    description = "Used to store messages."
+
+    def build_config(self):
+        return {
+            "records": {
+                "display_name": "Records",
+                "info": "The list of records to store. Each record should contain the keys 'sender', 'sender_name', and 'session_id'.",
+            },
+        }
+
+    def build(
+        self,
+        records: List[Record],
+    ) -> List[Record]:
+        self.status = records
+        records = add_messages(records)
+        return records