From bac1933a577d58c30e0d8b993b8dc72a54efdaea Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Fri, 8 Mar 2024 16:07:46 -0300
Subject: [PATCH] Add UpdateRecordComponent to helpers

---
 .../components/helpers/UpdateRecord.py        | 38 +++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 src/backend/langflow/components/helpers/UpdateRecord.py

diff --git a/src/backend/langflow/components/helpers/UpdateRecord.py b/src/backend/langflow/components/helpers/UpdateRecord.py
new file mode 100644
index 000000000..92c2bae54
--- /dev/null
+++ b/src/backend/langflow/components/helpers/UpdateRecord.py
@@ -0,0 +1,38 @@
+from langflow import CustomComponent
+from langflow.schema import Record
+
+
+class UpdateRecordComponent(CustomComponent):
+    display_name = "Update Record"
+    description = "Updates a record with new data."
+
+    def build_config(self):
+        return {
+            "record": {
+                "display_name": "Record",
+                "info": "The record to update.",
+            },
+            "new_data": {
+                "display_name": "New Data",
+                "info": "The new data to update the record with.",
+            },
+        }
+
+    def build(
+        self,
+        record: Record,
+        new_data: dict,
+    ) -> Record:
+        """
+        Updates a record with new data.
+
+        Args:
+            record (Record): The record to update.
+            new_data (dict): The new data to update the record with.
+
+        Returns:
+            Record: The updated record.
+        """
+        record.data.update(new_data)
+        self.status = record
+        return record