jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

refactor: Update Data class to simplify attribute retrieval

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-20 09:20:51 -03:00
commit 8c90d68a03
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/backend/base/langflow/schema/data.py
View file

@ -154,8 +154,7 @@ class Data(BaseModel):
return self.__getattribute__(key) return self.__getattribute__(key)
if key in {"data", "text_key"} or key.startswith("_"): if key in {"data", "text_key"} or key.startswith("_"):
return super().__getattr__(key) return super().__getattr__(key)
return self.data[key]
return self.data.get(key, self.default_value)
except KeyError: except KeyError:
# Fallback to default behavior to raise AttributeError for undefined attributes # Fallback to default behavior to raise AttributeError for undefined attributes
raise AttributeError(f"'{type(self).__name__}' object has no attribute '{key}'") raise AttributeError(f"'{type(self).__name__}' object has no attribute '{key}'")