From 6ab64e096343d50baf50f076736fb4397785525d Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 28 Feb 2024 20:32:17 -0300 Subject: [PATCH] Update display_name fallback in Vertex class --- src/backend/langflow/graph/vertex/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/backend/langflow/graph/vertex/base.py b/src/backend/langflow/graph/vertex/base.py index 3e9ffd64e..9b6cd2027 100644 --- a/src/backend/langflow/graph/vertex/base.py +++ b/src/backend/langflow/graph/vertex/base.py @@ -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