diff --git a/src/backend/base/langflow/schema/dataframe.py b/src/backend/base/langflow/schema/dataframe.py index 193036dc2..4d7373690 100644 --- a/src/backend/base/langflow/schema/dataframe.py +++ b/src/backend/base/langflow/schema/dataframe.py @@ -72,6 +72,9 @@ class DataFrame(pandas_DataFrame): @text_key.setter def text_key(self, value: str) -> None: + if value not in self.columns: + msg = f"Text key '{value}' not found in DataFrame columns" + raise ValueError(msg) self._text_key = value @property