From 8c90d68a03c9500a69b81d85ab825abee9d0f8fb Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 20 Jun 2024 09:20:51 -0300 Subject: [PATCH] refactor: Update Data class to simplify attribute retrieval --- src/backend/base/langflow/schema/data.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/backend/base/langflow/schema/data.py b/src/backend/base/langflow/schema/data.py index 62f343386..861f40878 100644 --- a/src/backend/base/langflow/schema/data.py +++ b/src/backend/base/langflow/schema/data.py @@ -154,8 +154,7 @@ class Data(BaseModel): return self.__getattribute__(key) if key in {"data", "text_key"} or key.startswith("_"): return super().__getattr__(key) - - return self.data.get(key, self.default_value) + return self.data[key] except KeyError: # Fallback to default behavior to raise AttributeError for undefined attributes raise AttributeError(f"'{type(self).__name__}' object has no attribute '{key}'")