d85657f214* chore: update linting workflows to include dev branch in merge_group * Update README.md Add 1.0 banner * Update README.md * chore: update package versions in pyproject.toml files * Refactor "created_at" column type for consistency and fix cancel middleware (#2316) * chore: update linting workflows to include dev branch in merge_group * Update README.md Add 1.0 banner * Update README.md * chore: update package versions in pyproject.toml files * refactor: update "created_at" column type to use the "sa" module for consistency * Update README.md Add 1.0 banner * chore: Remove unused import in ToolCallingAgent.py * fix: adapt RequestCancelledMiddleware to handle cancelled requests * chore: Remove unused import in test_helper_components.py * refactor: Declare queue variable with explicit type in RequestCancelledMiddleware --------- Co-authored-by: Rodrigo Nader <rodrigosilvanader@gmail.com> * chore: Update AstraDB.py imports and method signature for search_documents * chore: Update package versions in pyproject.toml files * chore: Update run-name in release.yml for Langflow Release * fix: add call to _add_documents_to_vector_store in AstraDB component --------- Co-authored-by: Rodrigo Nader <rodrigosilvanader@gmail.com>
79 lines
2.7 KiB
Python
79 lines
2.7 KiB
Python
from langflow.components import helpers
|
|
from langflow.custom.utils import build_custom_component_template
|
|
from langflow.schema import Data
|
|
|
|
# def test_update_data_component():
|
|
# # Arrange
|
|
# update_data_component = helpers.UpdateDataComponent()
|
|
|
|
# # Act
|
|
# new_data = {"new_key": "new_value"}
|
|
# existing_data = Data(data={"existing_key": "existing_value"})
|
|
# result = update_data_component.build(existing_data, new_data)
|
|
# assert result.data == {"existing_key": "existing_value", "new_key": "new_value"}
|
|
# assert result.existing_key == "existing_value"
|
|
# assert result.new_key == "new_value"
|
|
|
|
|
|
# def test_document_to_data_component():
|
|
# # Arrange
|
|
# document_to_data_component = helpers.DocumentsToDataComponent()
|
|
|
|
# # Act
|
|
# # Replace with your actual test data
|
|
# document = Document(page_content="key: value", metadata={"url": "https://example.com"})
|
|
# result = document_to_data_component.build(document)
|
|
|
|
# # Assert
|
|
# # Replace with your actual expected result
|
|
# assert result == [Data(data={"text": "key: value", "url": "https://example.com"})]
|
|
|
|
|
|
def test_uuid_generator_component():
|
|
# Arrange
|
|
uuid_generator_component = helpers.UUIDGeneratorComponent()
|
|
uuid_generator_component.code = open(helpers.IDGenerator.__file__, "r").read()
|
|
|
|
frontend_node, _ = build_custom_component_template(uuid_generator_component)
|
|
|
|
# Act
|
|
build_config = frontend_node.get("template")
|
|
field_name = "unique_id"
|
|
build_config = uuid_generator_component.update_build_config(build_config, None, field_name)
|
|
unique_id = build_config["unique_id"]["value"]
|
|
result = uuid_generator_component.build(unique_id)
|
|
|
|
# Assert
|
|
# UUID should be a string of length 36
|
|
assert isinstance(result, str)
|
|
assert len(result) == 36
|
|
|
|
|
|
def test_data_as_text_component():
|
|
# Arrange
|
|
data_as_text_component = helpers.ParseDataComponent()
|
|
|
|
# Act
|
|
# Replace with your actual test data
|
|
data = [Data(data={"key": "value", "bacon": "eggs"})]
|
|
template = "Data:{data} -- Bacon:{bacon}"
|
|
data_as_text_component.set_attributes({"data": data, "template": template})
|
|
result = data_as_text_component.parse_data()
|
|
|
|
# Assert
|
|
# Replace with your actual expected result
|
|
assert result.text == "Data:{'key': 'value', 'bacon': 'eggs'} -- Bacon:eggs"
|
|
|
|
|
|
# def test_text_to_data_component():
|
|
# # Arrange
|
|
# text_to_data_component = helpers.CreateDataComponent()
|
|
|
|
# # Act
|
|
# # Replace with your actual test data
|
|
# dict_with_text = {"field_1": {"key": "value"}}
|
|
# result = text_to_data_component.build(number_of_fields=1, **dict_with_text)
|
|
|
|
# # Assert
|
|
# # Replace with your actual expected result
|
|
# assert result == Data(data={"key": "value"})
|