Skip to content
10 changes: 3 additions & 7 deletions src/backend/base/langflow/components/processing/split_text.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,8 +63,7 @@ class SplitTextComponent(Component):
]

outputs = [
Output(display_name="Chunks", name="chunks", method="split_text"),
Output(display_name="DataFrame", name="dataframe", method="as_dataframe"),
Output(display_name="Chunks", name="dataframe", method="split_text"),
]

def _docs_to_data(self, docs) -> list[Data]:
Expand Down Expand Up @@ -132,8 +131,5 @@ def split_text_base(self):
msg = f"Error splitting text: {e}"
raise TypeError(msg) from e

def split_text(self) -> list[Data]:
return self._docs_to_data(self.split_text_base())

def as_dataframe(self) -> DataFrame:
return DataFrame(self.split_text())
def split_text(self) -> DataFrame:
return DataFrame(self._docs_to_data(self.split_text_base()))
Loading
Loading