rename Record to Data

tests/test_record.py

@@ -1,23 +1,22 @@
 from langchain_core.documents import Document
-
-from langflow.schema import Record
+from langflow.schema import Data
 
 
 def test_record_initialization():
-    record = Record(text_key="msg", data={"msg": "Hello, World!", "extra": "value"})
+    record = Data(text_key="msg", data={"msg": "Hello, World!", "extra": "value"})
     assert record.msg == "Hello, World!"
     assert record.extra == "value"
 
 
 def test_validate_data_with_extra_keys():
-    record = Record(dummy_key="dummy", data={"key": "value"})
+    record = Data(dummy_key="dummy", data={"key": "value"})
     assert record.data["dummy_key"] == "dummy"
     assert "dummy_key" in record.data
     assert record.key == "value"
 
 
 def test_conversion_to_document():
-    record = Record(data={"text": "Sample text", "meta": "data"})
+    record = Data(data={"text": "Sample text", "meta": "data"})
     document = record.to_lc_document()
     assert document.page_content == "Sample text"
     assert document.metadata == {"meta": "data"}
@@ -25,35 +24,35 @@ def test_conversion_to_document():
 
 def test_conversion_from_document():
     document = Document(page_content="Doc content", metadata={"meta": "info"})
-    record = Record.from_document(document)
+    record = Data.from_document(document)
     assert record.text == "Doc content"
     assert record.meta == "info"
 
 
 def test_add_method_for_strings():
-    record1 = Record(data={"text": "Hello"})
-    record2 = Record(data={"text": " World"})
+    record1 = Data(data={"text": "Hello"})
+    record2 = Data(data={"text": " World"})
     combined = record1 + record2
     assert combined.text == "Hello World"
 
 
 def test_add_method_for_integers():
-    record1 = Record(data={"number": 5})
-    record2 = Record(data={"number": 10})
+    record1 = Data(data={"number": 5})
+    record2 = Data(data={"number": 10})
     combined = record1 + record2
     assert combined.number == 15
 
 
 def test_add_method_with_non_overlapping_keys():
-    record1 = Record(data={"text": "Hello"})
-    record2 = Record(data={"number": 10})
+    record1 = Data(data={"text": "Hello"})
+    record2 = Data(data={"number": 10})
     combined = record1 + record2
     assert combined.text == "Hello"
     assert combined.number == 10
 
 
 def test_custom_attribute_get_set_del():
-    record = Record()
+    record = Data()
     record.custom_attr = "custom_value"
     assert record.custom_attr == "custom_value"
     del record.custom_attr
@@ -63,7 +62,7 @@ def test_custom_attribute_get_set_del():
 def test_deep_copy():
     import copy
 
-    record1 = Record(data={"text": "Hello", "number": 10})
+    record1 = Data(data={"text": "Hello", "number": 10})
     record2 = copy.deepcopy(record1)
     assert record2.text == "Hello"
     assert record2.number == 10
@@ -72,20 +71,20 @@ def test_deep_copy():
 
 
 def test_custom_attribute_setting_and_getting():
-    record = Record()
+    record = Data()
     record.dynamic_attribute = "Dynamic Value"
     assert record.dynamic_attribute == "Dynamic Value"
 
 
 def test_str_and_dir_methods():
-    record = Record(text_key="text", data={"text": "Test Text", "key": "value"})
+    record = Data(text_key="text", data={"text": "Test Text", "key": "value"})
     assert "Test Text" in str(record)
     assert "key" in dir(record)
     assert "data" in dir(record)
 
 
 def test_dir_includes_data_keys():
-    record = Record(data={"text": "Hello", "new_attr": "value"})
+    record = Data(data={"text": "Hello", "new_attr": "value"})
     dir_output = dir(record)
 
     # Check for standard attributes
@@ -103,7 +102,7 @@ def test_dir_includes_data_keys():
 
 
 def test_dir_reflects_attribute_deletion():
-    record = Record(data={"removable": "I can be removed"})
+    record = Data(data={"removable": "I can be removed"})
     assert "removable" in dir(record)
 
     # Delete the attribute and check again
@@ -113,27 +112,27 @@ def test_dir_reflects_attribute_deletion():
 
 def test_get_text_with_text_key():
     data = {"text": "Hello, World!"}
-    schema = Record(data=data, text_key="text", default_value="default")
+    schema = Data(data=data, text_key="text", default_value="default")
     result = schema.get_text()
     assert result == "Hello, World!"
 
 
 def test_get_text_without_text_key():
     data = {"other_key": "Hello, World!"}
-    schema = Record(data=data, text_key="text", default_value="default")
+    schema = Data(data=data, text_key="text", default_value="default")
     result = schema.get_text()
     assert result == "default"
 
 
 def test_get_text_with_empty_data():
     data = {}
-    schema = Record(data=data, text_key="text", default_value="default")
+    schema = Data(data=data, text_key="text", default_value="default")
     result = schema.get_text()
     assert result == "default"
 
 
 def test_get_text_with_none_data():
     data = None
-    schema = Record(data=data, text_key="text", default_value="default")
+    schema = Data(data=data, text_key="text", default_value="default")
     result = schema.get_text()
     assert result == "default"