Refactor TextToRecordComponent build method

src/backend/langflow/components/helpers/TextToRecord.py

@@ -1,98 +1,25 @@
-from typing import Any, Optional
-
 from langflow import CustomComponent
 from langflow.schema import Record
-from langflow.template.field.base import TemplateField
 
 
 class TextToRecordComponent(CustomComponent):
     display_name = "Text to Record"
-    description = "A component to create a record from key-value pairs."
+    description = "A component to create a record from Text."
-    field_order = ["mode", "keys", "n_keys"]
     beta: bool = True
 
-    def set_key_template(self, build_config, field_value):
-        keys_template = TemplateField(
-            name="n_keys" if field_value == "Number" else "keys",
-            field_type="dict" if field_value == "Number" else "str",
-            is_list=False if field_value == "Number" else True,
-            display_name="Keys",
-            info=(
-                "The Number of keys to use for the record."
-                if field_value == "Number"
-                else "The keys to use for the record."
-            ),
-            input_types=["Text"],
-        )
-        build_config["keys"] = keys_template.to_dict()
-
-    def set_n_keys(self, build_config, field_name, field_value):
-        if int(field_value) == 0:
-            keep = ["n_keys", "code"]
-            for key in build_config.copy():
-                if key in keep:
-                    continue
-                del build_config[key]
-        build_config[field_name]["value"] = int(field_value)
-
-        # Add new fields depending on the field value
-        for i in range(int(field_value)):
-            field = TemplateField(
-                name=f"Key and Value {i}",
-                field_type="dict",
-                display_name="",
-                info="The key for the record.",
-                input_types=["Text"],
-            )
-            build_config[field.name] = field.to_dict()
-
-    def set_keys_template(self, build_config, field_value):
-        for key in build_config.copy():
-            if key == "keys":
-                continue
-            del build_config[key]
-        for i in range(int(field_value)):
-            field = TemplateField(
-                name=f"Key and Value {i}",
-                field_type="dict",
-                display_name="",
-                info="The key for the record.",
-                input_types=["Text"],
-            )
-            build_config[field.name] = field.to_dict()
-
-    def update_build_config(
-        self,
-        build_config: dict,
-        field_value: Any,
-        field_name: Optional[str] = None,
-    ):
-        if field_name == "mode":
-            build_config["mode"]["value"] = field_value
-            self.set_key_template(build_config, field_value)
-        if field_value is None:
-            return
-        if field_name == "n_keys":
-            self.set_n_keys(build_config, field_name, field_value)
-        elif field_name == "keys":
-            self.set_keys_template(build_config, field_value)
-        return build_config
-
     def build_config(self):
         return {
-            "mode": {
+            "data": {
-                "display_name": "Mode",
+                "display_name": "Data",
-                "options": ["Text", "Number"],
+                "info": "The data to convert to a record.",
-                "info": "The mode to use for creating the record.",
+                "input_types": ["Text"],
-                "real_time_refresh": True,
+            }
-                "input_types": [],
-            },
         }
 
-    def build(self, mode: str, **kwargs) -> Record:
+    def build(
-        if mode == "Text":
+        self,
-            data = kwargs
+        data: dict,
-        else:
+    ) -> Record:
-            data = {k: v for key, d in kwargs.items() for k, v in d.items() if key not in ["mode", "n_keys", "keys"]}
+        return_record = Record(data=data)
-        record = Record(data=data)
+        self.status = return_record
-        return record
+        return return_record