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

Update display_name fallback in Vertex class

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-28 20:32:17 -03:00
commit 6ab64e0963
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/backend/langflow/graph/vertex/base.py
View file

@ -191,7 +191,7 @@ class Vertex:
def _parse_data(self) -> None:
self.data = self._data["data"]
self.output = self.data["node"]["base_classes"]
self.display_name = self.data["node"]["display_name"]
self.display_name = self.data["node"].get("display_name", self.id.split("-")[0])
self.pinned = self.data["node"].get("pinned", False)
self.selected_output_type = self.data["node"].get("selected_output_type")
self.is_input = self.data["node"].get("is_input") or self.is_input