feat: add dataframe to Loop component, fixes loop input error

[rodrigosnader]

This pull request introduces several changes to the LoopComponent class in the langflow backend to support both Data and DataFrame types. The modifications include updates to the input handling, validation, and output methods, as well as adjustments to the initial setup templates.

Changes to LoopComponent class:

• Updated import statements to include DataFrame and changed DataInput to HandleInput. (src/backend/base/langflow/components/logic/loop.py)
• Modified the inputs attribute to accept both Data and DataFrame types and updated the display name and info accordingly. (src/backend/base/langflow/components/logic/loop.py)
• Enhanced the _validate_data method to handle DataFrame inputs and convert them to a list of Data objects. (src/backend/base/langflow/components/logic/loop.py)
• Changed the done_output method to return a DataFrame instead of a Data object when the iteration is complete. (src/backend/base/langflow/components/logic/loop.py)
• Updated the aggregated_output method to return a list of Data objects. (src/backend/base/langflow/components/logic/loop.py)

Changes to initial setup templates:

• Modified LoopTemplate.json to reflect the changes in input and output types, including the addition of DataFrame as a valid type. (src/backend/base/langflow/initial_setup/starter_projects/LoopTemplate.json) [1] [2] [3] [4] [5] [6]

These changes enhance the flexibility of the LoopComponent by allowing it to handle both Data and DataFrame inputs, making it more versatile for different data processing scenarios.

[Copilot]

PR Overview

This pull request adds support for DataFrame inputs and outputs in the LoopComponent, enhancing its data handling flexibility. Key changes include:

• Replacing DataInput with HandleInput and updating the input metadata to accept both Data and DataFrame.
• Modifying _validate_data to convert DataFrame inputs into a list of Data objects.
• Changing done_output to return a DataFrame when iteration is complete.

Reviewed Changes

File	Description
src/backend/base/langflow/components/logic/loop.py	Updates to inputs, validation, and output methods to incorporate DataFrame support

Copilot reviewed 2 out of 2 changed files in this pull request and generated 1 comment.

Comments suppressed due to low confidence (1)

src/backend/base/langflow/components/logic/loop.py:84

• [nitpick] The method name 'done_output' may not clearly indicate that it returns a DataFrame. Consider updating the name or adding documentation to clarify this behavior for improved code clarity.

def done_output(self) -> DataFrame:

[Copilot]

The method 'aggregated_output' returns a list of Data objects, while 'done_output' now returns a DataFrame. This discrepancy in return types may lead to inconsistencies when consuming these outputs; consider unifying the output types or clearly documenting the differences in behavior.

Suggested change

	def aggregated_output(self) -> list[Data]:
	"""Return the aggregated list once all items are processed."""
	def aggregated_output(self) -> DataFrame:
	"""Return the aggregated DataFrame once all items are processed."""

[codspeed-hq]

CodSpeed Performance Report

Merging #6996 will degrade performances by 35.52%

_{Comparing loop_component_dataframe_support (5a357b6) with main (58e0d42)}

Summary

❌ 2 regressions
✅ 17 untouched benchmarks

⚠️ Please fix the performance issues or acknowledge them on CodSpeed.

Benchmarks breakdown

	Benchmark	BASE	HEAD	Change
❌	test_build_flow_invalid_job_id	8 ms	12.4 ms	-35.52%
❌	test_cancel_nonexistent_build	7.7 ms	9.4 ms	-17.53%

[edwinjosechittilappilly]

Testing,

We might need to deplrecate the old one as well as update the template.

cc. @italojohnny May I know if you are already working on it if not I could pick it up!

[italojohnny]

Testing,

We might need to deplrecate the old one as well as update the template.

cc. @italojohnny May I know if you are already working on it if not I could pick it up!

I haven’t had time to check this PR yet. I’m finishing the tests for #7026

[edwinjosechittilappilly]

Working on Deprecating the old component and making this as a the new component.

[Copilot]

Pull Request Overview

This PR refactors the looping functionality by introducing a new BasicLoopComponent that supports both Data and DataFrame types for inputs and adjusts output behavior accordingly. Key changes include:

• Creating BasicLoopComponent with updated input validation and output methods.
• Updating related methods and tests to use BasicLoopComponent instead of the legacy LoopComponent.
• Adjusting import and type usages in test files and associated modules.

Reviewed Changes

Copilot reviewed 19 out of 19 changed files in this pull request and generated 1 comment.

Show a summary per file

File	Description
src/backend/base/langflow/components/logic/loop_basic.py	New BasicLoopComponent with updated input handling and output methods.
src/backend/base/langflow/graph/vertex/base.py	Enhanced get_value_from_output_names with improved documentation and edge handling.
src/backend/base/langflow/custom/custom_component/component.py	Added get_loop_output_value to support loop outputs.
src/backend/tests/unit/components/logic/test_loop.py	Updated to test using BasicLoopComponent.
Other test and module files	Minor updates to imports and dependencies supporting the new component changes.

[Copilot]

Consider using '>= data_length' instead of '> data_length' so that the loop terminates correctly when the current index equals the length of the data list.

Suggested change

	return current_index > data_length
	return current_index >= data_length